darcs.py
163 lines
| 6.0 KiB
| text/x-python
|
PythonLexer
|
r8250 | # darcs.py - darcs support for the convert extension | ||
# | ||||
# Copyright 2007-2009 Matt Mackall <mpm@selenic.com> and others | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2, incorporated herein by reference. | ||||
|
r5359 | |||
|
r5512 | from common import NoRepo, checktool, commandline, commit, converter_source | ||
|
r5359 | from mercurial.i18n import _ | ||
from mercurial import util | ||||
import os, shutil, tempfile | ||||
# The naming drift of ElementTree is fun! | ||||
try: from xml.etree.cElementTree import ElementTree | ||||
except ImportError: | ||||
try: from xml.etree.ElementTree import ElementTree | ||||
except ImportError: | ||||
try: from elementtree.cElementTree import ElementTree | ||||
except ImportError: | ||||
try: from elementtree.ElementTree import ElementTree | ||||
except ImportError: ElementTree = None | ||||
|
r5512 | class darcs_source(converter_source, commandline): | ||
|
r5359 | def __init__(self, ui, path, rev=None): | ||
|
r5512 | converter_source.__init__(self, ui, path, rev=rev) | ||
commandline.__init__(self, ui, 'darcs') | ||||
|
r5359 | |||
|
r5520 | # check for _darcs, ElementTree, _darcs/inventory so that we can | ||
# easily skip test-convert-darcs if ElementTree is not around | ||||
|
r7973 | if not os.path.exists(os.path.join(path, '_darcs', 'inventories')): | ||
raise NoRepo("%s does not look like a darcs repo" % path) | ||||
|
r5520 | if not os.path.exists(os.path.join(path, '_darcs')): | ||
|
r5521 | raise NoRepo("%s does not look like a darcs repo" % path) | ||
|
r5359 | |||
|
r5497 | checktool('darcs') | ||
|
r9242 | version = self.run0('--version').splitlines()[0].strip() | ||
if version < '2.1': | ||||
raise util.Abort(_('darcs version 2.1 or newer needed (found %r)') % | ||||
version) | ||||
|
r5497 | |||
|
r5359 | if ElementTree is None: | ||
raise util.Abort(_("Python ElementTree module is not available")) | ||||
self.path = os.path.realpath(path) | ||||
self.lastrev = None | ||||
self.changes = {} | ||||
self.parents = {} | ||||
self.tags = {} | ||||
def before(self): | ||||
self.tmppath = tempfile.mkdtemp( | ||||
prefix='convert-' + os.path.basename(self.path) + '-') | ||||
output, status = self.run('init', repodir=self.tmppath) | ||||
self.checkexit(status) | ||||
|
r5512 | tree = self.xml('changes', xml_output=True, summary=True, | ||
repodir=self.path) | ||||
|
r5359 | tagname = None | ||
child = None | ||||
for elt in tree.findall('patch'): | ||||
node = elt.get('hash') | ||||
name = elt.findtext('name', '') | ||||
if name.startswith('TAG '): | ||||
tagname = name[4:].strip() | ||||
elif tagname is not None: | ||||
self.tags[tagname] = node | ||||
tagname = None | ||||
self.changes[node] = elt | ||||
self.parents[child] = [node] | ||||
child = node | ||||
self.parents[child] = [] | ||||
def after(self): | ||||
|
r9467 | self.ui.debug('cleaning up %s\n' % self.tmppath) | ||
|
r5362 | shutil.rmtree(self.tmppath, ignore_errors=True) | ||
|
r5359 | |||
|
r5512 | def xml(self, cmd, **kwargs): | ||
|
r5359 | etree = ElementTree() | ||
|
r5512 | fp = self._run(cmd, **kwargs) | ||
|
r5359 | etree.parse(fp) | ||
self.checkexit(fp.close()) | ||||
return etree.getroot() | ||||
|
r9527 | def manifest(self): | ||
man = [] | ||||
output, status = self.run('show', 'files', no_directories=True, | ||||
repodir=self.tmppath) | ||||
self.checkexit(status) | ||||
for line in output.split('\n'): | ||||
path = line[2:] | ||||
if path: | ||||
man.append(path) | ||||
return man | ||||
|
r5359 | def getheads(self): | ||
return self.parents[None] | ||||
def getcommit(self, rev): | ||||
elt = self.changes[rev] | ||||
date = util.strdate(elt.get('local_date'), '%a %b %d %H:%M:%S %Z %Y') | ||||
desc = elt.findtext('name') + '\n' + elt.findtext('comment', '') | ||||
return commit(author=elt.get('author'), date=util.datestr(date), | ||||
desc=desc.strip(), parents=self.parents[rev]) | ||||
def pull(self, rev): | ||||
|
r5512 | output, status = self.run('pull', self.path, all=True, | ||
match='hash %s' % rev, | ||||
no_test=True, no_posthook=True, | ||||
external_merge='/bin/false', | ||||
|
r5359 | repodir=self.tmppath) | ||
if status: | ||||
if output.find('We have conflicts in') == -1: | ||||
self.checkexit(status, output) | ||||
|
r5512 | output, status = self.run('revert', all=True, repodir=self.tmppath) | ||
|
r5359 | self.checkexit(status, output) | ||
|
r9527 | def getchanges(self, rev): | ||
|
r5359 | copies = {} | ||
changes = [] | ||||
|
r9527 | man = None | ||
|
r5359 | for elt in self.changes[rev].find('summary').getchildren(): | ||
if elt.tag in ('add_directory', 'remove_directory'): | ||||
continue | ||||
if elt.tag == 'move': | ||||
|
r9527 | if man is None: | ||
man = self.manifest() | ||||
source, dest = elt.get('from'), elt.get('to') | ||||
if source in man: | ||||
# File move | ||||
changes.append((source, rev)) | ||||
changes.append((dest, rev)) | ||||
copies[dest] = source | ||||
else: | ||||
# Directory move, deduce file moves from manifest | ||||
source = source + '/' | ||||
for f in man: | ||||
if not f.startswith(source): | ||||
continue | ||||
fdest = dest + '/' + f[len(source):] | ||||
changes.append((f, rev)) | ||||
changes.append((fdest, rev)) | ||||
copies[fdest] = f | ||||
|
r5359 | else: | ||
changes.append((elt.text.strip(), rev)) | ||||
|
r9527 | self.pull(rev) | ||
|
r5359 | self.lastrev = rev | ||
|
r8209 | return sorted(changes), copies | ||
|
r5359 | |||
def getfile(self, name, rev): | ||||
if rev != self.lastrev: | ||||
raise util.Abort(_('internal calling inconsistency')) | ||||
return open(os.path.join(self.tmppath, name), 'rb').read() | ||||
def getmode(self, name, rev): | ||||
mode = os.lstat(os.path.join(self.tmppath, name)).st_mode | ||||
return (mode & 0111) and 'x' or '' | ||||
def gettags(self): | ||||
return self.tags | ||||