##// END OF EJS Templates
fix bogus close spotted by pychecker (no close() in global scope)
Benoit Boissinot -
r4961:3fdd09ad default
parent child Browse files
Show More
@@ -1,196 +1,196 b''
1 1 # changelog.py - changelog class for mercurial
2 2 #
3 3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms
6 6 # of the GNU General Public License, incorporated herein by reference.
7 7
8 8 from revlog import *
9 9 from i18n import _
10 10 import os, time, util
11 11
12 12 def _string_escape(text):
13 13 """
14 14 >>> d = {'nl': chr(10), 'bs': chr(92), 'cr': chr(13), 'nul': chr(0)}
15 15 >>> s = "ab%(nl)scd%(bs)s%(bs)sn%(nul)sab%(cr)scd%(bs)s%(nl)s" % d
16 16 >>> s
17 17 'ab\\ncd\\\\\\\\n\\x00ab\\rcd\\\\\\n'
18 18 >>> res = _string_escape(s)
19 19 >>> s == _string_unescape(res)
20 20 True
21 21 """
22 22 # subset of the string_escape codec
23 23 text = text.replace('\\', '\\\\').replace('\n', '\\n').replace('\r', '\\r')
24 24 return text.replace('\0', '\\0')
25 25
26 26 def _string_unescape(text):
27 27 return text.decode('string_escape')
28 28
29 29 class appender:
30 30 '''the changelog index must be update last on disk, so we use this class
31 31 to delay writes to it'''
32 32 def __init__(self, fp, buf):
33 33 self.data = buf
34 34 self.fp = fp
35 35 self.offset = fp.tell()
36 36 self.size = util.fstat(fp).st_size
37 37
38 38 def end(self):
39 39 return self.size + len("".join(self.data))
40 40 def tell(self):
41 41 return self.offset
42 42 def flush(self):
43 43 pass
44 44 def close(self):
45 close(self.fp)
45 self.fp.close()
46 46
47 47 def seek(self, offset, whence=0):
48 48 '''virtual file offset spans real file and data'''
49 49 if whence == 0:
50 50 self.offset = offset
51 51 elif whence == 1:
52 52 self.offset += offset
53 53 elif whence == 2:
54 54 self.offset = self.end() + offset
55 55 if self.offset < self.size:
56 56 self.fp.seek(self.offset)
57 57
58 58 def read(self, count=-1):
59 59 '''only trick here is reads that span real file and data'''
60 60 ret = ""
61 61 old_offset = self.offset
62 62 if self.offset < self.size:
63 63 s = self.fp.read(count)
64 64 ret = s
65 65 self.offset += len(s)
66 66 if count > 0:
67 67 count -= len(s)
68 68 if count != 0:
69 69 doff = self.offset - self.size
70 70 self.data.insert(0, "".join(self.data))
71 71 del self.data[1:]
72 72 s = self.data[0][doff:doff+count]
73 73 self.offset += len(s)
74 74 ret += s
75 75 return ret
76 76
77 77 def write(self, s):
78 78 self.data.append(s)
79 79 self.offset += len(s)
80 80
81 81 class changelog(revlog):
82 82 def __init__(self, opener):
83 83 revlog.__init__(self, opener, "00changelog.i")
84 84
85 85 def delayupdate(self):
86 86 "delay visibility of index updates to other readers"
87 87 self._realopener = self.opener
88 88 self.opener = self._delayopener
89 89 self._delaycount = self.count()
90 90 self._delaybuf = []
91 91 self._delayname = None
92 92
93 93 def finalize(self, tr):
94 94 "finalize index updates"
95 95 self.opener = self._realopener
96 96 # move redirected index data back into place
97 97 if self._delayname:
98 98 util.rename(self._delayname + ".a", self._delayname)
99 99 elif self._delaybuf:
100 100 fp = self.opener(self.indexfile, 'a')
101 101 fp.write("".join(self._delaybuf))
102 102 fp.close()
103 103 del self._delaybuf
104 104 # split when we're done
105 105 self.checkinlinesize(tr)
106 106
107 107 def _delayopener(self, name, mode='r'):
108 108 fp = self._realopener(name, mode)
109 109 # only divert the index
110 110 if not name == self.indexfile:
111 111 return fp
112 112 # if we're doing an initial clone, divert to another file
113 113 if self._delaycount == 0:
114 114 self._delayname = fp.name
115 115 return self._realopener(name + ".a", mode)
116 116 # otherwise, divert to memory
117 117 return appender(fp, self._delaybuf)
118 118
119 119 def checkinlinesize(self, tr, fp=None):
120 120 if self.opener == self._delayopener:
121 121 return
122 122 return revlog.checkinlinesize(self, tr, fp)
123 123
124 124 def decode_extra(self, text):
125 125 extra = {}
126 126 for l in text.split('\0'):
127 127 if not l:
128 128 continue
129 129 k, v = _string_unescape(l).split(':', 1)
130 130 extra[k] = v
131 131 return extra
132 132
133 133 def encode_extra(self, d):
134 134 items = [_string_escape(":".join(t)) for t in d.iteritems()]
135 135 return "\0".join(items)
136 136
137 137 def extract(self, text):
138 138 """
139 139 format used:
140 140 nodeid\n : manifest node in ascii
141 141 user\n : user, no \n or \r allowed
142 142 time tz extra\n : date (time is int or float, timezone is int)
143 143 : extra is metadatas, encoded and separated by '\0'
144 144 : older versions ignore it
145 145 files\n\n : files modified by the cset, no \n or \r allowed
146 146 (.*) : comment (free text, ideally utf-8)
147 147
148 148 changelog v0 doesn't use extra
149 149 """
150 150 if not text:
151 151 return (nullid, "", (0, 0), [], "", {'branch': 'default'})
152 152 last = text.index("\n\n")
153 153 desc = util.tolocal(text[last + 2:])
154 154 l = text[:last].split('\n')
155 155 manifest = bin(l[0])
156 156 user = util.tolocal(l[1])
157 157
158 158 extra_data = l[2].split(' ', 2)
159 159 if len(extra_data) != 3:
160 160 time = float(extra_data.pop(0))
161 161 try:
162 162 # various tools did silly things with the time zone field.
163 163 timezone = int(extra_data[0])
164 164 except:
165 165 timezone = 0
166 166 extra = {}
167 167 else:
168 168 time, timezone, extra = extra_data
169 169 time, timezone = float(time), int(timezone)
170 170 extra = self.decode_extra(extra)
171 171 if not extra.get('branch'):
172 172 extra['branch'] = 'default'
173 173 files = l[3:]
174 174 return (manifest, user, (time, timezone), files, desc, extra)
175 175
176 176 def read(self, node):
177 177 return self.extract(self.revision(node))
178 178
179 179 def add(self, manifest, list, desc, transaction, p1=None, p2=None,
180 180 user=None, date=None, extra={}):
181 181
182 182 user, desc = util.fromlocal(user), util.fromlocal(desc)
183 183
184 184 if date:
185 185 parseddate = "%d %d" % util.parsedate(date)
186 186 else:
187 187 parseddate = "%d %d" % util.makedate()
188 188 if extra and extra.get("branch") in ("default", ""):
189 189 del extra["branch"]
190 190 if extra:
191 191 extra = self.encode_extra(extra)
192 192 parseddate = "%s %s" % (parseddate, extra)
193 193 list.sort()
194 194 l = [hex(manifest), user, parseddate] + list + ["", desc]
195 195 text = "\n".join(l)
196 196 return self.addrevision(text, transaction, self.count(), p1, p2)
General Comments 0
You need to be logged in to leave comments. Login now