changelog.py
198 lines
| 6.4 KiB
| text/x-python
|
PythonLexer
/ mercurial / changelog.py
|
r1095 | # changelog.py - changelog class for mercurial | ||
|
r1089 | # | ||
|
r4635 | # Copyright 2005-2007 Matt Mackall <mpm@selenic.com> | ||
|
r1089 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
from revlog import * | ||||
|
r3891 | from i18n import _ | ||
|
r3877 | import os, time, util | ||
|
r1089 | |||
|
r3232 | def _string_escape(text): | ||
""" | ||||
>>> d = {'nl': chr(10), 'bs': chr(92), 'cr': chr(13), 'nul': chr(0)} | ||||
>>> s = "ab%(nl)scd%(bs)s%(bs)sn%(nul)sab%(cr)scd%(bs)s%(nl)s" % d | ||||
>>> s | ||||
'ab\\ncd\\\\\\\\n\\x00ab\\rcd\\\\\\n' | ||||
>>> res = _string_escape(s) | ||||
>>> s == _string_unescape(res) | ||||
True | ||||
""" | ||||
# subset of the string_escape codec | ||||
text = text.replace('\\', '\\\\').replace('\n', '\\n').replace('\r', '\\r') | ||||
return text.replace('\0', '\\0') | ||||
def _string_unescape(text): | ||||
return text.decode('string_escape') | ||||
|
r4261 | class appender: | ||
'''the changelog index must be update last on disk, so we use this class | ||||
to delay writes to it''' | ||||
def __init__(self, fp, buf): | ||||
self.data = buf | ||||
self.fp = fp | ||||
self.offset = fp.tell() | ||||
self.size = util.fstat(fp).st_size | ||||
def end(self): | ||||
return self.size + len("".join(self.data)) | ||||
def tell(self): | ||||
return self.offset | ||||
def flush(self): | ||||
pass | ||||
def close(self): | ||||
|
r4961 | self.fp.close() | ||
|
r4261 | |||
def seek(self, offset, whence=0): | ||||
'''virtual file offset spans real file and data''' | ||||
if whence == 0: | ||||
self.offset = offset | ||||
elif whence == 1: | ||||
self.offset += offset | ||||
elif whence == 2: | ||||
self.offset = self.end() + offset | ||||
if self.offset < self.size: | ||||
self.fp.seek(self.offset) | ||||
def read(self, count=-1): | ||||
'''only trick here is reads that span real file and data''' | ||||
ret = "" | ||||
if self.offset < self.size: | ||||
s = self.fp.read(count) | ||||
ret = s | ||||
self.offset += len(s) | ||||
if count > 0: | ||||
count -= len(s) | ||||
if count != 0: | ||||
doff = self.offset - self.size | ||||
self.data.insert(0, "".join(self.data)) | ||||
del self.data[1:] | ||||
s = self.data[0][doff:doff+count] | ||||
self.offset += len(s) | ||||
ret += s | ||||
return ret | ||||
def write(self, s): | ||||
self.data.append(s) | ||||
self.offset += len(s) | ||||
|
r1089 | class changelog(revlog): | ||
|
r4258 | def __init__(self, opener): | ||
revlog.__init__(self, opener, "00changelog.i") | ||||
|
r1089 | |||
|
r4261 | def delayupdate(self): | ||
"delay visibility of index updates to other readers" | ||||
self._realopener = self.opener | ||||
|
r4269 | self.opener = self._delayopener | ||
self._delaycount = self.count() | ||||
|
r4261 | self._delaybuf = [] | ||
|
r4269 | self._delayname = None | ||
|
r4261 | |||
def finalize(self, tr): | ||||
"finalize index updates" | ||||
self.opener = self._realopener | ||||
|
r4269 | # move redirected index data back into place | ||
if self._delayname: | ||||
util.rename(self._delayname + ".a", self._delayname) | ||||
elif self._delaybuf: | ||||
|
r4261 | fp = self.opener(self.indexfile, 'a') | ||
fp.write("".join(self._delaybuf)) | ||||
fp.close() | ||||
del self._delaybuf | ||||
|
r4269 | # split when we're done | ||
|
r4261 | self.checkinlinesize(tr) | ||
|
r4269 | def _delayopener(self, name, mode='r'): | ||
|
r4261 | fp = self._realopener(name, mode) | ||
|
r4269 | # only divert the index | ||
|
r4261 | if not name == self.indexfile: | ||
return fp | ||||
|
r4269 | # if we're doing an initial clone, divert to another file | ||
if self._delaycount == 0: | ||||
self._delayname = fp.name | ||||
return self._realopener(name + ".a", mode) | ||||
# otherwise, divert to memory | ||||
|
r4261 | return appender(fp, self._delaybuf) | ||
def checkinlinesize(self, tr, fp=None): | ||||
|
r4269 | if self.opener == self._delayopener: | ||
|
r4261 | return | ||
return revlog.checkinlinesize(self, tr, fp) | ||||
|
r3233 | def decode_extra(self, text): | ||
extra = {} | ||||
for l in text.split('\0'): | ||||
if not l: | ||||
continue | ||||
k, v = _string_unescape(l).split(':', 1) | ||||
extra[k] = v | ||||
return extra | ||||
def encode_extra(self, d): | ||||
|
r4847 | # keys must be sorted to produce a deterministic changelog entry | ||
|
r4848 | keys = d.keys() | ||
keys.sort() | ||||
items = [_string_escape('%s:%s' % (k, d[k])) for k in keys] | ||||
|
r3233 | return "\0".join(items) | ||
|
r1089 | def extract(self, text): | ||
|
r3077 | """ | ||
format used: | ||||
|
r3233 | nodeid\n : manifest node in ascii | ||
user\n : user, no \n or \r allowed | ||||
time tz extra\n : date (time is int or float, timezone is int) | ||||
: extra is metadatas, encoded and separated by '\0' | ||||
: older versions ignore it | ||||
files\n\n : files modified by the cset, no \n or \r allowed | ||||
(.*) : comment (free text, ideally utf-8) | ||||
changelog v0 doesn't use extra | ||||
|
r3077 | """ | ||
|
r1089 | if not text: | ||
|
r4176 | return (nullid, "", (0, 0), [], "", {'branch': 'default'}) | ||
|
r1089 | last = text.index("\n\n") | ||
|
r3771 | desc = util.tolocal(text[last + 2:]) | ||
|
r3233 | l = text[:last].split('\n') | ||
|
r1089 | manifest = bin(l[0]) | ||
|
r3771 | user = util.tolocal(l[1]) | ||
|
r3233 | |||
extra_data = l[2].split(' ', 2) | ||||
if len(extra_data) != 3: | ||||
time = float(extra_data.pop(0)) | ||||
try: | ||||
# various tools did silly things with the time zone field. | ||||
timezone = int(extra_data[0]) | ||||
except: | ||||
timezone = 0 | ||||
extra = {} | ||||
else: | ||||
time, timezone, extra = extra_data | ||||
time, timezone = float(time), int(timezone) | ||||
extra = self.decode_extra(extra) | ||||
|
r4176 | if not extra.get('branch'): | ||
extra['branch'] = 'default' | ||||
|
r1089 | files = l[3:] | ||
|
r3233 | return (manifest, user, (time, timezone), files, desc, extra) | ||
|
r1089 | |||
def read(self, node): | ||||
return self.extract(self.revision(node)) | ||||
def add(self, manifest, list, desc, transaction, p1=None, p2=None, | ||||
|
r3233 | user=None, date=None, extra={}): | ||
|
r3771 | user, desc = util.fromlocal(user), util.fromlocal(desc) | ||
|
r1195 | if date: | ||
|
r2523 | parseddate = "%d %d" % util.parsedate(date) | ||
|
r1195 | else: | ||
|
r2522 | parseddate = "%d %d" % util.makedate() | ||
|
r4176 | if extra and extra.get("branch") in ("default", ""): | ||
del extra["branch"] | ||||
|
r3233 | if extra: | ||
extra = self.encode_extra(extra) | ||||
parseddate = "%s %s" % (parseddate, extra) | ||||
|
r1089 | list.sort() | ||
|
r2522 | l = [hex(manifest), user, parseddate] + list + ["", desc] | ||
|
r1089 | text = "\n".join(l) | ||
return self.addrevision(text, transaction, self.count(), p1, p2) | ||||