hg.py
286 lines
| 9.3 KiB
| text/x-python
|
PythonLexer
/ mercurial / hg.py
|
r0 | # hg.py - repository classes for mercurial | ||
# | ||||
|
r4635 | # Copyright 2005-2007 Matt Mackall <mpm@selenic.com> | ||
|
r2859 | # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com> | ||
|
r0 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r3891 | from i18n import _ | ||
|
r3877 | import localrepo, bundlerepo, httprepo, sshrepo, statichttprepo | ||
|
r5177 | import errno, lock, os, shutil, util, extensions | ||
|
r3877 | import merge as _merge | ||
import verify as _verify | ||||
|
r0 | |||
|
r2740 | def _local(path): | ||
|
r3195 | return (os.path.isfile(util.drop_scheme('file', path)) and | ||
|
r2768 | bundlerepo or localrepo) | ||
|
r2469 | |||
|
r6525 | def parseurl(url, revs=[]): | ||
|
r5177 | '''parse url#branch, returning url, branch + revs''' | ||
if '#' not in url: | ||||
|
r5248 | return url, (revs or None), None | ||
|
r5177 | |||
url, rev = url.split('#', 1) | ||||
|
r5248 | return url, revs + [rev], rev | ||
|
r5177 | |||
|
r2472 | schemes = { | ||
|
r2740 | 'bundle': bundlerepo, | ||
'file': _local, | ||||
'http': httprepo, | ||||
'https': httprepo, | ||||
'ssh': sshrepo, | ||||
'static-http': statichttprepo, | ||||
|
r4853 | } | ||
|
r2469 | |||
|
r2740 | def _lookup(path): | ||
scheme = 'file' | ||||
if path: | ||||
c = path.find(':') | ||||
if c > 0: | ||||
scheme = path[:c] | ||||
thing = schemes.get(scheme) or schemes['file'] | ||||
try: | ||||
return thing(path) | ||||
except TypeError: | ||||
return thing | ||||
|
r2775 | |||
|
r2719 | def islocal(repo): | ||
'''return true if repo or path is local''' | ||||
if isinstance(repo, str): | ||||
|
r2740 | try: | ||
return _lookup(repo).islocal(repo) | ||||
except AttributeError: | ||||
return False | ||||
|
r2719 | return repo.local() | ||
|
r3195 | def repository(ui, path='', create=False): | ||
|
r2774 | """return a repository object for the specified path""" | ||
|
r2847 | repo = _lookup(path).instance(ui, path, create) | ||
|
r4074 | ui = getattr(repo, "ui", ui) | ||
|
r5192 | for name, module in extensions.extensions(): | ||
hook = getattr(module, 'reposetup', None) | ||||
if hook: | ||||
hook(ui, repo) | ||||
|
r2847 | return repo | ||
|
r2597 | |||
|
r2719 | def defaultdest(source): | ||
'''return default destination of clone if none is given''' | ||||
return os.path.basename(os.path.normpath(source)) | ||||
|
r2774 | |||
|
r6524 | def localpath(path): | ||
if path.startswith('file://localhost/'): | ||||
return path[16:] | ||||
if path.startswith('file://'): | ||||
return path[7:] | ||||
if path.startswith('file:'): | ||||
return path[5:] | ||||
return path | ||||
|
r2613 | def clone(ui, source, dest=None, pull=False, rev=None, update=True, | ||
stream=False): | ||||
|
r2597 | """Make a copy of an existing repository. | ||
Create a copy of an existing repository in a new directory. The | ||||
source and destination are URLs, as passed to the repository | ||||
function. Returns a pair of repository objects, the source and | ||||
newly created destination. | ||||
The location of the source is added to the new repository's | ||||
.hg/hgrc file, as the default to be used for future pulls and | ||||
pushes. | ||||
If an exception is raised, the partly cloned/updated destination | ||||
repository will be deleted. | ||||
|
r2600 | |||
|
r2719 | Arguments: | ||
source: repository object or URL | ||||
|
r2597 | |||
dest: URL of destination repository to create (defaults to base | ||||
name of source repository) | ||||
pull: always pull from source repository, even in local case | ||||
|
r2621 | stream: stream raw data uncompressed from repository (fast over | ||
LAN, slow over WAN) | ||||
|
r2613 | |||
|
r2597 | rev: revision to clone up to (implies pull=True) | ||
update: update working directory after clone completes, if | ||||
|
r6526 | destination is local repository (True means update to default rev, | ||
anything else is treated as a revision) | ||||
|
r2597 | """ | ||
|
r4478 | |||
|
r2719 | if isinstance(source, str): | ||
|
r6089 | origsource = ui.expandpath(source) | ||
source, rev, checkout = parseurl(origsource, rev) | ||||
|
r2719 | src_repo = repository(ui, source) | ||
else: | ||||
src_repo = source | ||||
|
r6089 | origsource = source = src_repo.url() | ||
checkout = None | ||||
|
r2719 | |||
|
r2597 | if dest is None: | ||
|
r2719 | dest = defaultdest(source) | ||
|
r3841 | ui.status(_("destination directory: %s\n") % dest) | ||
|
r2719 | |||
dest = localpath(dest) | ||||
source = localpath(source) | ||||
|
r2597 | |||
if os.path.exists(dest): | ||||
|
r3072 | raise util.Abort(_("destination '%s' already exists") % dest) | ||
|
r2597 | |||
class DirCleanup(object): | ||||
def __init__(self, dir_): | ||||
self.rmtree = shutil.rmtree | ||||
self.dir_ = dir_ | ||||
def close(self): | ||||
self.dir_ = None | ||||
def __del__(self): | ||||
if self.dir_: | ||||
self.rmtree(self.dir_, True) | ||||
|
r4915 | src_lock = dest_lock = dir_cleanup = None | ||
try: | ||||
if islocal(dest): | ||||
dir_cleanup = DirCleanup(dest) | ||||
|
r2597 | |||
|
r4915 | abspath = origsource | ||
copy = False | ||||
|
r6315 | if src_repo.cancopy() and islocal(dest): | ||
|
r5248 | abspath = os.path.abspath(util.drop_scheme('file', origsource)) | ||
|
r4915 | copy = not pull and not rev | ||
|
r2597 | |||
|
r4915 | if copy: | ||
try: | ||||
# we use a lock here because if we race with commit, we | ||||
# can end up with extra data in the cloned revlogs that's | ||||
# not pointed to by changesets, thus causing verify to | ||||
# fail | ||||
src_lock = src_repo.lock() | ||||
except lock.LockException: | ||||
copy = False | ||||
|
r2597 | |||
|
r4915 | if copy: | ||
if not os.path.exists(dest): | ||||
os.mkdir(dest) | ||||
|
r5569 | try: | ||
dest_path = os.path.realpath(os.path.join(dest, ".hg")) | ||||
os.mkdir(dest_path) | ||||
except OSError, inst: | ||||
if inst.errno == errno.EEXIST: | ||||
dir_cleanup.close() | ||||
raise util.Abort(_("destination '%s' already exists") | ||||
% dest) | ||||
raise | ||||
|
r2597 | |||
|
r6903 | for f in src_repo.store.copylist(): | ||
src = os.path.join(src_repo.path, f) | ||||
|
r6944 | dst = os.path.join(dest_path, f) | ||
dstbase = os.path.dirname(dst) | ||||
if dstbase and not os.path.exists(dstbase): | ||||
os.mkdir(dstbase) | ||||
|
r6903 | if os.path.exists(src): | ||
if dst.endswith('data'): | ||||
# lock to avoid premature writing to the target | ||||
dest_lock = lock.lock(os.path.join(dstbase, "lock")) | ||||
util.copyfiles(src, dst) | ||||
|
r2597 | |||
|
r4915 | # we need to re-init the repo after manually copying the data | ||
# into it | ||||
dest_repo = repository(ui, dest) | ||||
else: | ||||
|
r5569 | try: | ||
dest_repo = repository(ui, dest, create=True) | ||||
except OSError, inst: | ||||
if inst.errno == errno.EEXIST: | ||||
dir_cleanup.close() | ||||
raise util.Abort(_("destination '%s' already exists") | ||||
% dest) | ||||
raise | ||||
|
r2597 | |||
|
r4915 | revs = None | ||
if rev: | ||||
if 'lookup' not in src_repo.capabilities: | ||||
raise util.Abort(_("src repository does not support revision " | ||||
"lookup and so doesn't support clone by " | ||||
"revision")) | ||||
revs = [src_repo.lookup(r) for r in rev] | ||||
|
r3849 | |||
|
r4915 | if dest_repo.local(): | ||
dest_repo.clone(src_repo, heads=revs, stream=stream) | ||||
elif src_repo.local(): | ||||
src_repo.push(dest_repo, revs=revs) | ||||
else: | ||||
raise util.Abort(_("clone from remote to remote not supported")) | ||||
|
r2597 | |||
|
r5186 | if dir_cleanup: | ||
dir_cleanup.close() | ||||
|
r2597 | |||
if dest_repo.local(): | ||||
|
r4915 | fp = dest_repo.opener("hgrc", "w", text=True) | ||
fp.write("[paths]\n") | ||||
fp.write("default = %s\n" % abspath) | ||||
fp.close() | ||||
|
r5185 | |||
|
r4915 | if update: | ||
|
r6338 | dest_repo.ui.status(_("updating working directory\n")) | ||
|
r6526 | if update is not True: | ||
checkout = update | ||||
elif not checkout: | ||||
|
r5248 | try: | ||
checkout = dest_repo.lookup("default") | ||||
except: | ||||
checkout = dest_repo.changelog.tip() | ||||
|
r4915 | _update(dest_repo, checkout) | ||
|
r2597 | |||
|
r4915 | return src_repo, dest_repo | ||
finally: | ||||
del src_lock, dest_lock, dir_cleanup | ||||
|
r2775 | |||
|
r3316 | def _showstats(repo, stats): | ||
stats = ((stats[0], _("updated")), | ||||
(stats[1], _("merged")), | ||||
(stats[2], _("removed")), | ||||
(stats[3], _("unresolved"))) | ||||
note = ", ".join([_("%d files %s") % s for s in stats]) | ||||
repo.ui.status("%s\n" % note) | ||||
def _update(repo, node): return update(repo, node) | ||||
|
r2808 | def update(repo, node): | ||
"""update the working directory to node, merging linear changes""" | ||||
|
r3869 | pl = repo.parents() | ||
|
r4917 | stats = _merge.update(repo, node, False, False, None) | ||
|
r3316 | _showstats(repo, stats) | ||
if stats[3]: | ||||
|
r6518 | repo.ui.status(_("use 'hg resolve' to retry unresolved file merges\n")) | ||
|
r5635 | return stats[3] > 0 | ||
|
r2775 | |||
|
r4917 | def clean(repo, node, show_stats=True): | ||
|
r2808 | """forcibly switch the working directory to node, clobbering changes""" | ||
|
r4917 | stats = _merge.update(repo, node, False, True, None) | ||
|
r3316 | if show_stats: _showstats(repo, stats) | ||
|
r5635 | return stats[3] > 0 | ||
|
r2775 | |||
|
r4917 | def merge(repo, node, force=None, remind=True): | ||
|
r2808 | """branch merge with node, resolving changes""" | ||
|
r4917 | stats = _merge.update(repo, node, True, force, False) | ||
|
r3316 | _showstats(repo, stats) | ||
if stats[3]: | ||||
pl = repo.parents() | ||||
|
r6518 | repo.ui.status(_("use 'hg resolve' to retry unresolved file merges\n")) | ||
|
r3316 | elif remind: | ||
repo.ui.status(_("(branch merge, don't forget to commit)\n")) | ||||
|
r5635 | return stats[3] > 0 | ||
|
r2808 | |||
|
r4917 | def revert(repo, node, choose): | ||
|
r2808 | """revert changes to revision in node without updating dirstate""" | ||
|
r5635 | return _merge.update(repo, node, False, True, choose)[3] > 0 | ||
|
r2778 | |||
def verify(repo): | ||||
"""verify the consistency of a repository""" | ||||
return _verify.verify(repo) | ||||