This is an archive of the discontinued Mercurial Phabricator instance.

Differential D5866

largefiles: use wrappedfunction() for util.copyfile() override
ClosedPublic

Authored by martinvonz on Feb 5 2019, 6:54 PM.

Details

Reviewers
pulkit
Group Reviewers
hg-reviewers
Commits
rHG028bb170e74d: largefiles: use wrappedfunction() for util.copyfile() override

Diff Detail

Repository
rHG Mercurial
Lint
Automatic diff as part of commit; lint not applicable.
Unit
Automatic diff as part of commit; unit tests not applicable.

Revision Contents
Changeset List

PathPackages
Mhgext/largefiles/overrides.py (27 lines)

Diff 13846

hgext/largefiles/overrides.py

oldmatch = installmatchfn(overridematch) oldmatch = installmatchfn(overridematch)
listpats = [] listpats = []
for pat in pats: for pat in pats:
if matchmod.patkind(pat) is not None: if matchmod.patkind(pat) is not None:
listpats.append(pat) listpats.append(pat)
else: else:
listpats.append(makestandin(pat)) listpats.append(makestandin(pat))
try:
origcopyfile = util.copyfile
copiedfiles = [] copiedfiles = []
def overridecopyfile(src, dest, *args, **kwargs): def overridecopyfile(orig, src, dest, *args, **kwargs):
if (lfutil.shortname in src and if (lfutil.shortname in src and
dest.startswith(repo.wjoin(lfutil.shortname))): dest.startswith(repo.wjoin(lfutil.shortname))):
destlfile = dest.replace(lfutil.shortname, '') destlfile = dest.replace(lfutil.shortname, '')
if not opts['force'] and os.path.exists(destlfile): if not opts['force'] and os.path.exists(destlfile):
raise IOError('', raise IOError('',
_('destination largefile already exists')) _('destination largefile already exists'))
copiedfiles.append((src, dest)) copiedfiles.append((src, dest))
origcopyfile(src, dest, *args, **kwargs) orig(src, dest, *args, **kwargs)
with extensions.wrappedfunction(util, 'copyfile', overridecopyfile):
util.copyfile = overridecopyfile
result += orig(ui, repo, listpats, opts, rename) result += orig(ui, repo, listpats, opts, rename)
finally:
util.copyfile = origcopyfile
lfdirstate = lfutil.openlfdirstate(ui, repo) lfdirstate = lfutil.openlfdirstate(ui, repo)
for (src, dest) in copiedfiles: for (src, dest) in copiedfiles:
if (lfutil.shortname in src and if (lfutil.shortname in src and
dest.startswith(repo.wjoin(lfutil.shortname))): dest.startswith(repo.wjoin(lfutil.shortname))):
srclfile = src.replace(repo.wjoin(lfutil.standin('')), '') srclfile = src.replace(repo.wjoin(lfutil.standin('')), '')
destlfile = dest.replace(repo.wjoin(lfutil.standin('')), '') destlfile = dest.replace(repo.wjoin(lfutil.standin('')), '')
destlfiledir = repo.wvfs.dirname(repo.wjoin(destlfile)) or '.' destlfiledir = repo.wvfs.dirname(repo.wjoin(destlfile)) or '.'