| pulkit |
| hg-reviewers |
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | hgext/largefiles/overrides.py (37 lines) |
| Status | Author | Revision | |
|---|---|---|---|
| Closed | martinvonz | ||
| Closed | martinvonz | ||
| Closed | martinvonz | ||
| Closed | martinvonz |
| manifest or f in excluded) | manifest or f in excluded) | ||||
| m._files = [lf for lf in m._files if notlfile(lf)] | m._files = [lf for lf in m._files if notlfile(lf)] | ||||
| m._fileset = set(m._files) | m._fileset = set(m._files) | ||||
| m.always = lambda: False | m.always = lambda: False | ||||
| origmatchfn = m.matchfn | origmatchfn = m.matchfn | ||||
| m.matchfn = lambda f: notlfile(f) and origmatchfn(f) | m.matchfn = lambda f: notlfile(f) and origmatchfn(f) | ||||
| return m | return m | ||||
| def installnormalfilesmatchfn(manifest): | |||||
| '''installmatchfn with a matchfn that ignores all largefiles''' | |||||
| def overridematch(ctx, pats=(), opts=None, globbed=False, | |||||
| default='relpath', badfn=None): | |||||
| if opts is None: | |||||
| opts = {} | |||||
| match = oldmatch(ctx, pats, opts, globbed, default, badfn=badfn) | |||||
| return composenormalfilematcher(match, manifest) | |||||
| oldmatch = installmatchfn(overridematch) | |||||
| def installmatchfn(f): | def installmatchfn(f): | ||||
| '''monkey patch the scmutil module with a custom match function. | '''monkey patch the scmutil module with a custom match function. | ||||
| Warning: it is monkey patching the _module_ on runtime! Not thread safe!''' | Warning: it is monkey patching the _module_ on runtime! Not thread safe!''' | ||||
| oldmatch = scmutil.match | oldmatch = scmutil.match | ||||
| setattr(f, 'oldmatch', oldmatch) | setattr(f, 'oldmatch', oldmatch) | ||||
| scmutil.match = f | scmutil.match = f | ||||
| return oldmatch | return oldmatch | ||||
| return orig(ui, repo, pats, opts, rename) | return orig(ui, repo, pats, opts, rename) | ||||
| # This could copy both lfiles and normal files in one command, | # This could copy both lfiles and normal files in one command, | ||||
| # but we don't want to do that. First replace their matcher to | # but we don't want to do that. First replace their matcher to | ||||
| # only match normal files and run it, then replace it to just | # only match normal files and run it, then replace it to just | ||||
| # match largefiles and run it again. | # match largefiles and run it again. | ||||
| nonormalfiles = False | nonormalfiles = False | ||||
| nolfiles = False | nolfiles = False | ||||
| installnormalfilesmatchfn(repo[None].manifest()) | manifest = repo[None].manifest() | ||||
| def normalfilesmatchfn(orig, ctx, pats=(), opts=None, globbed=False, | |||||
| default='relpath', badfn=None): | |||||
| if opts is None: | |||||
| opts = {} | |||||
| match = orig(ctx, pats, opts, globbed, default, badfn=badfn) | |||||
| return composenormalfilematcher(match, manifest) | |||||
| with extensions.wrappedfunction(scmutil, 'match', normalfilesmatchfn): | |||||
| try: | try: | ||||
| result = orig(ui, repo, pats, opts, rename) | result = orig(ui, repo, pats, opts, rename) | ||||
| except error.Abort as e: | except error.Abort as e: | ||||
| if pycompat.bytestr(e) != _('no files to copy'): | if pycompat.bytestr(e) != _('no files to copy'): | ||||
| raise e | raise e | ||||
| else: | else: | ||||
| nonormalfiles = True | nonormalfiles = True | ||||
| result = 0 | result = 0 | ||||
| finally: | |||||
| restorematchfn() | |||||
| # The first rename can cause our current working directory to be removed. | # The first rename can cause our current working directory to be removed. | ||||
| # In that case there is nothing left to copy/rename so just quit. | # In that case there is nothing left to copy/rename so just quit. | ||||
| try: | try: | ||||
| repo.getcwd() | repo.getcwd() | ||||
| except OSError: | except OSError: | ||||
| return result | return result | ||||