store.py
127 lines
| 4.1 KiB
| text/x-python
|
PythonLexer
/ mercurial / store.py
|
r6839 | # store.py - repository store handling for Mercurial | ||
# | ||||
# Copyright 2008 Matt Mackall <mpm@selenic.com> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r6840 | import os, stat, osutil, util | ||
|
r6839 | def _buildencodefun(): | ||
e = '_' | ||||
win_reserved = [ord(x) for x in '\\:*?"<>|'] | ||||
cmap = dict([ (chr(x), chr(x)) for x in xrange(127) ]) | ||||
for x in (range(32) + range(126, 256) + win_reserved): | ||||
cmap[chr(x)] = "~%02x" % x | ||||
for x in range(ord("A"), ord("Z")+1) + [ord(e)]: | ||||
cmap[chr(x)] = e + chr(x).lower() | ||||
dmap = {} | ||||
for k, v in cmap.iteritems(): | ||||
dmap[v] = k | ||||
def decode(s): | ||||
i = 0 | ||||
while i < len(s): | ||||
for l in xrange(1, 4): | ||||
try: | ||||
yield dmap[s[i:i+l]] | ||||
i += l | ||||
break | ||||
except KeyError: | ||||
pass | ||||
else: | ||||
raise KeyError | ||||
return (lambda s: "".join([cmap[c] for c in s]), | ||||
lambda s: "".join(list(decode(s)))) | ||||
encodefilename, decodefilename = _buildencodefun() | ||||
|
r6898 | def _calcmode(path): | ||
try: | ||||
# files in .hg/ will be created using this mode | ||||
mode = os.stat(path).st_mode | ||||
# avoid some useless chmods | ||||
if (0777 & ~util._umask) == (0777 & mode): | ||||
mode = None | ||||
except OSError: | ||||
mode = None | ||||
return mode | ||||
|
r6903 | _data = 'data 00manifest.d 00manifest.i 00changelog.d 00changelog.i' | ||
|
r6898 | class basicstore: | ||
|
r6840 | '''base class for local repository stores''' | ||
|
r6988 | def __init__(self, path, opener, pathjoiner): | ||
self.pathjoiner = pathjoiner | ||||
|
r6840 | self.path = path | ||
|
r6898 | self.createmode = _calcmode(path) | ||
self.opener = opener(self.path) | ||||
self.opener.createmode = self.createmode | ||||
|
r6840 | |||
def join(self, f): | ||||
|
r6988 | return self.pathjoiner(self.path, f) | ||
|
r6840 | |||
|
r6899 | def _walk(self, relpath, recurse): | ||
|
r6900 | '''yields (unencoded, encoded, size)''' | ||
|
r6988 | path = self.pathjoiner(self.path, relpath) | ||
|
r6840 | striplen = len(self.path) + len(os.sep) | ||
|
r6899 | prefix = path[striplen:] | ||
l = [] | ||||
if os.path.isdir(path): | ||||
visit = [path] | ||||
while visit: | ||||
p = visit.pop() | ||||
for f, kind, st in osutil.listdir(p, stat=True): | ||||
|
r6988 | fp = self.pathjoiner(p, f) | ||
|
r6899 | if kind == stat.S_IFREG and f[-2:] in ('.d', '.i'): | ||
|
r6900 | n = util.pconvert(fp[striplen:]) | ||
l.append((n, n, st.st_size)) | ||||
|
r6899 | elif kind == stat.S_IFDIR and recurse: | ||
visit.append(fp) | ||||
return util.sort(l) | ||||
|
r6840 | |||
|
r6900 | def datafiles(self): | ||
|
r6899 | return self._walk('data', True) | ||
|
r6840 | |||
def walk(self): | ||||
|
r6900 | '''yields (unencoded, encoded, size)''' | ||
|
r6840 | # yield data files first | ||
|
r6892 | for x in self.datafiles(): | ||
|
r6840 | yield x | ||
# yield manifest before changelog | ||||
|
r6899 | meta = self._walk('', False) | ||
|
r6840 | meta.reverse() | ||
for x in meta: | ||||
yield x | ||||
|
r6903 | def copylist(self): | ||
return ['requires'] + _data.split() | ||||
|
r6898 | class encodedstore(basicstore): | ||
|
r6988 | def __init__(self, path, opener, pathjoiner): | ||
self.pathjoiner = pathjoiner | ||||
self.path = self.pathjoiner(path, 'store') | ||||
|
r6898 | self.createmode = _calcmode(self.path) | ||
|
r6896 | op = opener(self.path) | ||
|
r6840 | op.createmode = self.createmode | ||
|
r6902 | self.opener = lambda f, *args, **kw: op(encodefilename(f), *args, **kw) | ||
|
r6840 | |||
|
r6900 | def datafiles(self): | ||
for a, b, size in self._walk('data', True): | ||||
|
r6892 | try: | ||
|
r6900 | a = decodefilename(a) | ||
|
r6892 | except KeyError: | ||
|
r6900 | a = None | ||
yield a, b, size | ||||
|
r6840 | |||
def join(self, f): | ||||
|
r6988 | return self.pathjoiner(self.path, encodefilename(f)) | ||
|
r6840 | |||
|
r6903 | def copylist(self): | ||
return (['requires', '00changelog.i'] + | ||||
|
r6988 | [self.pathjoiner('store', f) for f in _data.split()]) | ||
|
r6903 | |||
|
r6989 | def store(requirements, path, opener, pathjoiner=None): | ||
pathjoiner = pathjoiner or os.path.join | ||||
|
r6898 | if 'store' in requirements: | ||
|
r6988 | return encodedstore(path, opener, pathjoiner) | ||
return basicstore(path, opener, pathjoiner) | ||||