context.py
480 lines
| 15.9 KiB
| text/x-python
|
PythonLexer
/ mercurial / context.py
|
r2563 | # context.py - changeset and file context objects for mercurial | ||
# | ||||
|
r2859 | # Copyright 2006 Matt Mackall <mpm@selenic.com> | ||
|
r2563 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r3172 | from node import * | ||
|
r3225 | from i18n import gettext as _ | ||
|
r3126 | from demandload import demandload | ||
|
r3302 | demandload(globals(), "ancestor bdiff repo revlog util os") | ||
|
r3122 | |||
|
r2563 | class changectx(object): | ||
"""A changecontext object makes access to data related to a particular | ||||
changeset convenient.""" | ||||
|
r3132 | def __init__(self, repo, changeid=None): | ||
|
r2563 | """changeid is a revision number, node, or tag""" | ||
self._repo = repo | ||||
|
r3143 | if not changeid and changeid != 0: | ||
|
r3132 | p1, p2 = self._repo.dirstate.parents() | ||
self._rev = self._repo.changelog.rev(p1) | ||||
if self._rev == -1: | ||||
changeid = 'tip' | ||||
else: | ||||
self._node = p1 | ||||
return | ||||
|
r2643 | self._node = self._repo.lookup(changeid) | ||
|
r2563 | self._rev = self._repo.changelog.rev(self._node) | ||
|
r3166 | def __str__(self): | ||
return short(self.node()) | ||||
|
r3151 | def __repr__(self): | ||
|
r3216 | return "<changectx %s>" % str(self) | ||
|
r3151 | |||
|
r3165 | def __eq__(self, other): | ||
return self._rev == other._rev | ||||
|
r3168 | def __nonzero__(self): | ||
return self._rev != -1 | ||||
|
r3215 | def __getattr__(self, name): | ||
if name == '_changeset': | ||||
|
r2563 | self._changeset = self._repo.changelog.read(self.node()) | ||
return self._changeset | ||||
|
r3215 | elif name == '_manifest': | ||
self._manifest = self._repo.manifest.read(self._changeset[0]) | ||||
|
r2563 | return self._manifest | ||
|
r3337 | elif name == '_manifestdelta': | ||
md = self._repo.manifest.readdelta(self._changeset[0]) | ||||
self._manifestdelta = md | ||||
return self._manifestdelta | ||||
|
r3215 | else: | ||
raise AttributeError, name | ||||
def changeset(self): return self._changeset | ||||
def manifest(self): return self._manifest | ||||
|
r2563 | |||
def rev(self): return self._rev | ||||
def node(self): return self._node | ||||
|
r3215 | def user(self): return self._changeset[1] | ||
def date(self): return self._changeset[2] | ||||
def files(self): return self._changeset[3] | ||||
def description(self): return self._changeset[4] | ||||
|
r2563 | |||
def parents(self): | ||||
"""return contexts for each parent changeset""" | ||||
|
r2627 | p = self._repo.changelog.parents(self._node) | ||
|
r2563 | return [ changectx(self._repo, x) for x in p ] | ||
def children(self): | ||||
"""return contexts for each child changeset""" | ||||
|
r2627 | c = self._repo.changelog.children(self._node) | ||
|
r2563 | return [ changectx(self._repo, x) for x in c ] | ||
def filenode(self, path): | ||||
|
r3336 | if '_manifest' in self.__dict__: | ||
|
r3242 | try: | ||
return self._manifest[path] | ||||
except KeyError: | ||||
raise repo.LookupError(_("'%s' not found in manifest") % path) | ||||
|
r3337 | if '_manifestdelta' in self.__dict__ or path in self.files(): | ||
if path in self._manifestdelta: | ||||
return self._manifestdelta[path] | ||||
|
r3215 | node, flag = self._repo.manifest.find(self._changeset[0], path) | ||
|
r3242 | if not node: | ||
raise repo.LookupError(_("'%s' not found in manifest") % path) | ||||
|
r2563 | return node | ||
|
r2628 | def filectx(self, path, fileid=None): | ||
|
r2563 | """get a file context from this changeset""" | ||
|
r2628 | if fileid is None: | ||
fileid = self.filenode(path) | ||||
|
r3214 | return filectx(self._repo, path, fileid=fileid, changectx=self) | ||
|
r2563 | |||
def filectxs(self): | ||||
"""generate a file context for each file in this changeset's | ||||
manifest""" | ||||
mf = self.manifest() | ||||
m = mf.keys() | ||||
m.sort() | ||||
for f in m: | ||||
yield self.filectx(f, fileid=mf[f]) | ||||
|
r3125 | def ancestor(self, c2): | ||
""" | ||||
return the ancestor context of self and c2 | ||||
""" | ||||
n = self._repo.changelog.ancestor(self._node, c2._node) | ||||
return changectx(self._repo, n) | ||||
|
r2563 | class filectx(object): | ||
"""A filecontext object makes access to data related to a particular | ||||
filerevision convenient.""" | ||||
|
r3214 | def __init__(self, repo, path, changeid=None, fileid=None, | ||
filelog=None, changectx=None): | ||||
|
r2563 | """changeid can be a changeset revision, node, or tag. | ||
fileid can be a file revision or node.""" | ||||
self._repo = repo | ||||
self._path = path | ||||
|
r3143 | assert changeid is not None or fileid is not None | ||
|
r2643 | |||
|
r3124 | if filelog: | ||
self._filelog = filelog | ||||
|
r3214 | if changectx: | ||
self._changectx = changectx | ||||
self._changeid = changectx.node() | ||||
|
r3124 | |||
|
r3143 | if fileid is None: | ||
|
r2643 | self._changeid = changeid | ||
else: | ||||
|
r3213 | self._fileid = fileid | ||
|
r2563 | |||
|
r3144 | def __getattr__(self, name): | ||
if name == '_changectx': | ||||
|
r2643 | self._changectx = changectx(self._repo, self._changeid) | ||
return self._changectx | ||||
|
r3213 | elif name == '_filelog': | ||
self._filelog = self._repo.file(self._path) | ||||
return self._filelog | ||||
elif name == '_changeid': | ||||
self._changeid = self._filelog.linkrev(self._filenode) | ||||
return self._changeid | ||||
|
r3144 | elif name == '_filenode': | ||
|
r3241 | try: | ||
|
r3336 | if '_fileid' in self.__dict__: | ||
|
r3241 | self._filenode = self._filelog.lookup(self._fileid) | ||
else: | ||||
self._filenode = self._changectx.filenode(self._path) | ||||
except revlog.RevlogError, inst: | ||||
raise repo.LookupError(str(inst)) | ||||
|
r3144 | return self._filenode | ||
elif name == '_filerev': | ||||
self._filerev = self._filelog.rev(self._filenode) | ||||
return self._filerev | ||||
else: | ||||
raise AttributeError, name | ||||
|
r2563 | |||
|
r3168 | def __nonzero__(self): | ||
return self._filerev != nullid | ||||
|
r3166 | def __str__(self): | ||
return "%s@%s" % (self.path(), short(self.node())) | ||||
|
r3151 | def __repr__(self): | ||
|
r3216 | return "<filectx %s>" % str(self) | ||
|
r3151 | |||
|
r3165 | def __eq__(self, other): | ||
return self._path == other._path and self._changeid == other._changeid | ||||
|
r3207 | def filectx(self, fileid): | ||
'''opens an arbitrary revision of the file without | ||||
opening a new filelog''' | ||||
return filectx(self._repo, self._path, fileid=fileid, | ||||
filelog=self._filelog) | ||||
|
r2563 | def filerev(self): return self._filerev | ||
def filenode(self): return self._filenode | ||||
def filelog(self): return self._filelog | ||||
|
r3150 | def rev(self): | ||
|
r3336 | if '_changectx' in self.__dict__: | ||
|
r3150 | return self._changectx.rev() | ||
return self._filelog.linkrev(self._filenode) | ||||
|
r3144 | def node(self): return self._changectx.node() | ||
def user(self): return self._changectx.user() | ||||
def date(self): return self._changectx.date() | ||||
def files(self): return self._changectx.files() | ||||
def description(self): return self._changectx.description() | ||||
def manifest(self): return self._changectx.manifest() | ||||
|
r3149 | def changectx(self): return self._changectx | ||
|
r2563 | |||
def data(self): return self._filelog.read(self._filenode) | ||||
def renamed(self): return self._filelog.renamed(self._filenode) | ||||
|
r3122 | def path(self): return self._path | ||
|
r3302 | def size(self): return self._filelog.size(self._filerev) | ||
|
r2563 | |||
|
r3310 | def cmp(self, text): return self._filelog.cmp(self._filenode, text) | ||
|
r2563 | def parents(self): | ||
|
r3124 | p = self._path | ||
fl = self._filelog | ||||
pl = [ (p, n, fl) for n in self._filelog.parents(self._filenode) ] | ||||
|
r3123 | |||
|
r3124 | r = self.renamed() | ||
|
r3122 | if r: | ||
|
r3124 | pl[0] = (r[0], r[1], None) | ||
return [ filectx(self._repo, p, fileid=n, filelog=l) | ||||
for p,n,l in pl if n != nullid ] | ||||
|
r2563 | |||
def children(self): | ||||
# hard for renames | ||||
c = self._filelog.children(self._filenode) | ||||
|
r3124 | return [ filectx(self._repo, self._path, fileid=x, | ||
filelog=self._filelog) for x in c ] | ||||
|
r2566 | |||
|
r3172 | def annotate(self, follow=False): | ||
'''returns a list of tuples of (ctx, line) for each line | ||||
in the file, where ctx is the filectx of the node where | ||||
that line was last changed''' | ||||
def decorate(text, rev): | ||||
return ([rev] * len(text.splitlines()), text) | ||||
def pair(parent, child): | ||||
for a1, a2, b1, b2 in bdiff.blocks(parent[1], child[1]): | ||||
child[0][b1:b2] = parent[0][a1:a2] | ||||
return child | ||||
getlog = util.cachefunc(lambda x: self._repo.file(x)) | ||||
def getctx(path, fileid): | ||||
log = path == self._path and self._filelog or getlog(path) | ||||
return filectx(self._repo, path, fileid=fileid, filelog=log) | ||||
getctx = util.cachefunc(getctx) | ||||
def parents(f): | ||||
# we want to reuse filectx objects as much as possible | ||||
p = f._path | ||||
|
r3217 | if f._filerev is None: # working dir | ||
pl = [ (n.path(), n.filerev()) for n in f.parents() ] | ||||
else: | ||||
pl = [ (p, n) for n in f._filelog.parentrevs(f._filerev) ] | ||||
|
r3172 | |||
if follow: | ||||
r = f.renamed() | ||||
if r: | ||||
pl[0] = (r[0], getlog(r[0]).rev(r[1])) | ||||
|
r3146 | |||
|
r3172 | return [ getctx(p, n) for p, n in pl if n != -1 ] | ||
|
r3217 | |||
|
r3172 | # find all ancestors | ||
needed = {self: 1} | ||||
visit = [self] | ||||
files = [self._path] | ||||
while visit: | ||||
f = visit.pop(0) | ||||
for p in parents(f): | ||||
if p not in needed: | ||||
needed[p] = 1 | ||||
visit.append(p) | ||||
if p._path not in files: | ||||
files.append(p._path) | ||||
else: | ||||
# count how many times we'll use this | ||||
needed[p] += 1 | ||||
# sort by revision (per file) which is a topological order | ||||
visit = [] | ||||
files.reverse() | ||||
for f in files: | ||||
fn = [(n._filerev, n) for n in needed.keys() if n._path == f] | ||||
fn.sort() | ||||
visit.extend(fn) | ||||
hist = {} | ||||
for r, f in visit: | ||||
curr = decorate(f.data(), f) | ||||
for p in parents(f): | ||||
if p != nullid: | ||||
curr = pair(hist[p], curr) | ||||
# trim the history of unneeded revs | ||||
needed[p] -= 1 | ||||
if not needed[p]: | ||||
del hist[p] | ||||
hist[f] = curr | ||||
return zip(hist[f][0], hist[f][1].splitlines(1)) | ||||
|
r3124 | |||
|
r3126 | def ancestor(self, fc2): | ||
""" | ||||
find the common ancestor file context, if any, of self, and fc2 | ||||
""" | ||||
|
r3135 | acache = {} | ||
|
r3217 | |||
# prime the ancestor cache for the working directory | ||||
for c in (self, fc2): | ||||
if c._filerev == None: | ||||
pl = [ (n.path(), n.filenode()) for n in c.parents() ] | ||||
acache[(c._path, None)] = pl | ||||
|
r3126 | flcache = {self._path:self._filelog, fc2._path:fc2._filelog} | ||
|
r3135 | def parents(vertex): | ||
if vertex in acache: | ||||
return acache[vertex] | ||||
f, n = vertex | ||||
if f not in flcache: | ||||
|
r3126 | flcache[f] = self._repo.file(f) | ||
|
r3135 | fl = flcache[f] | ||
pl = [ (f,p) for p in fl.parents(n) if p != nullid ] | ||||
re = fl.renamed(n) | ||||
if re: | ||||
pl.append(re) | ||||
acache[vertex]=pl | ||||
return pl | ||||
|
r3126 | |||
|
r3135 | a, b = (self._path, self._filenode), (fc2._path, fc2._filenode) | ||
v = ancestor.ancestor(a, b, parents) | ||||
if v: | ||||
f,n = v | ||||
return filectx(self._repo, f, fileid=n, filelog=flcache[f]) | ||||
|
r3126 | |||
|
r3135 | return None | ||
|
r3217 | |||
class workingctx(changectx): | ||||
"""A workingctx object makes access to data related to | ||||
the current working directory convenient.""" | ||||
def __init__(self, repo): | ||||
self._repo = repo | ||||
self._rev = None | ||||
self._node = None | ||||
def __str__(self): | ||||
|
r3313 | return str(self._parents[0]) + "+" | ||
|
r3217 | |||
def __nonzero__(self): | ||||
return True | ||||
def __getattr__(self, name): | ||||
if name == '_parents': | ||||
self._parents = self._repo.parents() | ||||
return self._parents | ||||
if name == '_status': | ||||
self._status = self._repo.status() | ||||
return self._status | ||||
if name == '_manifest': | ||||
self._buildmanifest() | ||||
return self._manifest | ||||
else: | ||||
raise AttributeError, name | ||||
def _buildmanifest(self): | ||||
"""generate a manifest corresponding to the working directory""" | ||||
|
r3218 | man = self._parents[0].manifest().copy() | ||
|
r3217 | copied = self._repo.dirstate.copies() | ||
modified, added, removed, deleted, unknown = self._status[:5] | ||||
for i,l in (("a", added), ("m", modified), ("u", unknown)): | ||||
for f in l: | ||||
man[f] = man.get(copied.get(f, f), nullid) + i | ||||
man.set(f, util.is_exec(self._repo.wjoin(f), man.execf(f))) | ||||
for f in deleted + removed: | ||||
del man[f] | ||||
self._manifest = man | ||||
def manifest(self): return self._manifest | ||||
def user(self): return self._repo.ui.username() | ||||
def date(self): return util.makedate() | ||||
def description(self): return "" | ||||
def files(self): | ||||
f = self.modified() + self.added() + self.removed() | ||||
f.sort() | ||||
return f | ||||
def modified(self): return self._status[0] | ||||
def added(self): return self._status[1] | ||||
def removed(self): return self._status[2] | ||||
def deleted(self): return self._status[3] | ||||
def unknown(self): return self._status[4] | ||||
def clean(self): return self._status[5] | ||||
def parents(self): | ||||
"""return contexts for each parent changeset""" | ||||
return self._parents | ||||
def children(self): | ||||
return [] | ||||
def filectx(self, path): | ||||
"""get a file context from the working directory""" | ||||
return workingfilectx(self._repo, path, workingctx=self) | ||||
def ancestor(self, c2): | ||||
"""return the ancestor context of self and c2""" | ||||
return self._parents[0].ancestor(c2) # punt on two parents for now | ||||
class workingfilectx(filectx): | ||||
"""A workingfilectx object makes access to data related to a particular | ||||
file in the working directory convenient.""" | ||||
def __init__(self, repo, path, filelog=None, workingctx=None): | ||||
"""changeid can be a changeset revision, node, or tag. | ||||
fileid can be a file revision or node.""" | ||||
self._repo = repo | ||||
self._path = path | ||||
self._changeid = None | ||||
self._filerev = self._filenode = None | ||||
if filelog: | ||||
self._filelog = filelog | ||||
if workingctx: | ||||
self._changectx = workingctx | ||||
def __getattr__(self, name): | ||||
if name == '_changectx': | ||||
self._changectx = workingctx(repo) | ||||
return self._changectx | ||||
elif name == '_repopath': | ||||
|
r3298 | self._repopath = (self._repo.dirstate.copied(self._path) | ||
or self._path) | ||||
return self._repopath | ||||
|
r3217 | elif name == '_filelog': | ||
self._filelog = self._repo.file(self._repopath) | ||||
return self._filelog | ||||
else: | ||||
raise AttributeError, name | ||||
def __nonzero__(self): | ||||
return True | ||||
def __str__(self): | ||||
|
r3313 | return "%s@%s" % (self.path(), self._changectx) | ||
|
r3217 | |||
def filectx(self, fileid): | ||||
'''opens an arbitrary revision of the file without | ||||
opening a new filelog''' | ||||
return filectx(self._repo, self._repopath, fileid=fileid, | ||||
filelog=self._filelog) | ||||
def rev(self): | ||||
|
r3336 | if '_changectx' in self.__dict__: | ||
|
r3217 | return self._changectx.rev() | ||
return self._filelog.linkrev(self._filenode) | ||||
def data(self): return self._repo.wread(self._path) | ||||
def renamed(self): | ||||
rp = self._repopath | ||||
if rp == self._path: | ||||
return None | ||||
return rp, self._workingctx._parents._manifest.get(rp, nullid) | ||||
def parents(self): | ||||
'''return parent filectxs, following copies if necessary''' | ||||
p = self._path | ||||
rp = self._repopath | ||||
|
r3298 | pcl = self._changectx._parents | ||
|
r3217 | fl = self._filelog | ||
pl = [ (rp, pcl[0]._manifest.get(rp, nullid), fl) ] | ||||
if len(pcl) > 1: | ||||
if rp != p: | ||||
fl = None | ||||
pl.append((p, pcl[1]._manifest.get(p, nullid), fl)) | ||||
return [ filectx(self._repo, p, fileid=n, filelog=l) | ||||
for p,n,l in pl if n != nullid ] | ||||
def children(self): | ||||
return [] | ||||
|
r3302 | def size(self): return os.stat(self._repo.wjoin(self._path)).st_size | ||
|
r3310 | |||
def cmp(self, text): return self._repo.wread(self._path) == text | ||||