hg.py
209 lines
| 6.4 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") | ||
|
r2597 | demandload(globals(), "errno lock os shutil util") | ||
|
r0 | |||
|
r2469 | def bundle(ui, path): | ||
if path.startswith('bundle://'): | ||||
path = path[9:] | ||||
else: | ||||
path = path[7:] | ||||
s = path.split("+", 1) | ||||
if len(s) == 1: | ||||
repopath, bundlename = "", s[0] | ||||
else: | ||||
repopath, bundlename = s | ||||
return bundlerepo.bundlerepository(ui, repopath, bundlename) | ||||
def hg(ui, path): | ||||
ui.warn(_("hg:// syntax is deprecated, please use http:// instead\n")) | ||||
return httprepo.httprepository(ui, path.replace("hg://", "http://")) | ||||
def local_(ui, path, create=0): | ||||
|
r2472 | if path.startswith('file:'): | ||
path = path[5:] | ||||
|
r2469 | return localrepo.localrepository(ui, path, create) | ||
|
r2549 | def ssh_(ui, path, create=0): | ||
return sshrepo.sshrepository(ui, path, create) | ||||
|
r2469 | def old_http(ui, path): | ||
ui.warn(_("old-http:// syntax is deprecated, " | ||||
"please use static-http:// instead\n")) | ||||
return statichttprepo.statichttprepository( | ||||
ui, path.replace("old-http://", "http://")) | ||||
def static_http(ui, path): | ||||
return statichttprepo.statichttprepository( | ||||
ui, path.replace("static-http://", "http://")) | ||||
|
r2472 | schemes = { | ||
|
r2469 | 'bundle': bundle, | ||
'file': local_, | ||||
'hg': hg, | ||||
'http': lambda ui, path: httprepo.httprepository(ui, path), | ||||
'https': lambda ui, path: httprepo.httpsrepository(ui, path), | ||||
'old-http': old_http, | ||||
|
r2549 | 'ssh': ssh_, | ||
|
r2469 | 'static-http': static_http, | ||
} | ||||
|
r60 | def repository(ui, path=None, create=0): | ||
|
r2595 | scheme = None | ||
if path: | ||||
c = path.find(':') | ||||
if c > 0: | ||||
scheme = schemes.get(path[:c]) | ||||
else: | ||||
path = '' | ||||
ctor = scheme or schemes['file'] | ||||
|
r2555 | if create: | ||
try: | ||||
|
r2469 | return ctor(ui, path, create) | ||
|
r2555 | except TypeError: | ||
raise util.Abort(_('cannot create new repository over "%s" protocol') % | ||||
scheme) | ||||
return ctor(ui, path) | ||||
|
r2597 | |||
|
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 | |||
|
r2597 | Keyword arguments: | ||
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 | ||||
""" | ||||
if dest is None: | ||||
dest = os.path.basename(os.path.normpath(source)) | ||||
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) | ||||
src_repo = repository(ui, source) | ||||
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(): | ||||
dest_path = os.path.realpath(dest) | ||||
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")) | ||||
# 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")) | ||||
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 | ||||
# we need to re-init the repo after manually copying the data | ||||
# into it | ||||
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: | ||||
dest_repo.update(dest_repo.changelog.tip()) | ||||
if dir_cleanup: | ||||
dir_cleanup.close() | ||||
return src_repo, dest_repo | ||||