hg.py
230 lines
| 7.0 KiB
| text/x-python
|
PythonLexer
/ mercurial / hg.py
|
r0 | # hg.py - repository classes for mercurial | ||
# | ||||
# Copyright 2005 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. | ||||
|
r1089 | from node import * | ||
from repo import * | ||||
|
r262 | from demandload import * | ||
|
r2431 | from i18n import gettext as _ | ||
|
r1945 | demandload(globals(), "localrepo bundlerepo httprepo sshrepo statichttprepo") | ||
|
r2809 | demandload(globals(), "errno lock os shutil util merge@_merge verify@_verify") | ||
|
r0 | |||
|
r2740 | def _local(path): | ||
|
r2768 | return (os.path.isfile(path and util.drop_scheme('file', path)) and | ||
bundlerepo or localrepo) | ||||
|
r2469 | |||
|
r2472 | schemes = { | ||
|
r2740 | 'bundle': bundlerepo, | ||
'file': _local, | ||||
'hg': httprepo, | ||||
'http': httprepo, | ||||
'https': httprepo, | ||||
'old-http': statichttprepo, | ||||
'ssh': sshrepo, | ||||
'static-http': statichttprepo, | ||||
|
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() | ||
|
r2740 | def repository(ui, path=None, create=False): | ||
|
r2774 | """return a repository object for the specified path""" | ||
|
r2740 | return _lookup(path).instance(ui, path, create) | ||
|
r2597 | |||
|
r2719 | def defaultdest(source): | ||
'''return default destination of clone if none is given''' | ||||
return os.path.basename(os.path.normpath(source)) | ||||
|
r2774 | |||
|
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 | ||||
destination is local repository | ||||
""" | ||||
|
r2719 | if isinstance(source, str): | ||
src_repo = repository(ui, source) | ||||
else: | ||||
src_repo = source | ||||
source = src_repo.url() | ||||
|
r2597 | if dest is None: | ||
|
r2719 | dest = defaultdest(source) | ||
def localpath(path): | ||||
if path.startswith('file://'): | ||||
return path[7:] | ||||
if path.startswith('file:'): | ||||
return path[5:] | ||||
return path | ||||
dest = localpath(dest) | ||||
source = localpath(source) | ||||
|
r2597 | |||
if os.path.exists(dest): | ||||
raise util.Abort(_("destination '%s' already exists"), dest) | ||||
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) | ||||
dest_repo = None | ||||
try: | ||||
dest_repo = repository(ui, dest) | ||||
raise util.Abort(_("destination '%s' already exists." % dest)) | ||||
except RepoError: | ||||
dest_repo = repository(ui, dest, create=True) | ||||
dest_path = None | ||||
dir_cleanup = None | ||||
if dest_repo.local(): | ||||
|
r2719 | dest_path = os.path.realpath(dest_repo.root) | ||
|
r2597 | dir_cleanup = DirCleanup(dest_path) | ||
abspath = source | ||||
copy = False | ||||
if src_repo.local() and dest_repo.local(): | ||||
abspath = os.path.abspath(source) | ||||
copy = not pull and not rev | ||||
src_lock, dest_lock = None, None | ||||
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 | ||||
if copy: | ||||
# we lock here to avoid premature writing to the target | ||||
dest_lock = lock.lock(os.path.join(dest_path, ".hg", "lock")) | ||||
|
r2631 | # we need to remove the (empty) data dir in dest so copyfiles | ||
# can do its work | ||||
os.rmdir(os.path.join(dest_path, ".hg", "data")) | ||||
|
r2597 | files = "data 00manifest.d 00manifest.i 00changelog.d 00changelog.i" | ||
for f in files.split(): | ||||
src = os.path.join(source, ".hg", f) | ||||
dst = os.path.join(dest_path, ".hg", f) | ||||
try: | ||||
util.copyfiles(src, dst) | ||||
except OSError, inst: | ||||
if inst.errno != errno.ENOENT: | ||||
raise | ||||
|
r2631 | # we need to re-init the repo after manually copying the data | ||
# into it | ||||
|
r2597 | dest_repo = repository(ui, dest) | ||
else: | ||||
revs = None | ||||
if rev: | ||||
if not src_repo.local(): | ||||
raise util.Abort(_("clone by revision not supported yet " | ||||
"for remote repositories")) | ||||
revs = [src_repo.lookup(r) for r in rev] | ||||
if dest_repo.local(): | ||||
|
r2613 | dest_repo.clone(src_repo, heads=revs, stream=stream) | ||
|
r2597 | elif src_repo.local(): | ||
src_repo.push(dest_repo, revs=revs) | ||||
else: | ||||
raise util.Abort(_("clone from remote to remote not supported")) | ||||
if src_lock: | ||||
src_lock.release() | ||||
if dest_repo.local(): | ||||
fp = dest_repo.opener("hgrc", "w", text=True) | ||||
fp.write("[paths]\n") | ||||
fp.write("default = %s\n" % abspath) | ||||
fp.close() | ||||
if dest_lock: | ||||
dest_lock.release() | ||||
if update: | ||||
|
r2808 | _merge.update(dest_repo, dest_repo.changelog.tip()) | ||
|
r2597 | if dir_cleanup: | ||
dir_cleanup.close() | ||||
return src_repo, dest_repo | ||||
|
r2775 | |||
|
r2808 | def update(repo, node): | ||
"""update the working directory to node, merging linear changes""" | ||||
return _merge.update(repo, node) | ||||
|
r2775 | |||
|
r2808 | def clean(repo, node, wlock=None, show_stats=True): | ||
"""forcibly switch the working directory to node, clobbering changes""" | ||||
return _merge.update(repo, node, force=True, wlock=wlock, | ||||
show_stats=show_stats) | ||||
|
r2775 | |||
|
r2808 | def merge(repo, node, force=None, remind=True, wlock=None): | ||
"""branch merge with node, resolving changes""" | ||||
|
r2815 | return _merge.update(repo, node, branchmerge=True, force=force, | ||
|
r2808 | remind=remind, wlock=wlock) | ||
|
r2812 | def revert(repo, node, choose, wlock): | ||
|
r2808 | """revert changes to revision in node without updating dirstate""" | ||
|
r2811 | return _merge.update(repo, node, force=True, partial=choose, | ||
|
r2812 | show_stats=False, wlock=wlock) | ||
|
r2778 | |||
def verify(repo): | ||||
"""verify the consistency of a repository""" | ||||
return _verify.verify(repo) | ||||