largefiles: move lfstatus context manager to lfutil
Differential Revision: https://phab.mercurial-scm.org/D7139
--- a/hgext/largefiles/lfutil.py Fri Nov 08 14:35:53 2019 -0800
+++ b/hgext/largefiles/lfutil.py Fri Oct 18 17:46:37 2019 -0700
@@ -9,6 +9,7 @@
'''largefiles utility code: must not import other modules in this package.'''
from __future__ import absolute_import
+import contextlib
import copy
import hashlib
import os
@@ -39,6 +40,16 @@
# -- Private worker functions ------------------------------------------
+@contextlib.contextmanager
+def lfstatus(repo):
+ oldvalue = getattr(repo, 'lfstatus', False)
+ repo.lfstatus = True
+ try:
+ yield
+ finally:
+ repo.lfstatus = oldvalue
+
+
def getminsize(ui, assumelfiles, opt, default=10):
lfsize = opt
if not lfsize and assumelfiles:
--- a/hgext/largefiles/overrides.py Fri Nov 08 14:35:53 2019 -0800
+++ b/hgext/largefiles/overrides.py Fri Oct 18 17:46:37 2019 -0700
@@ -9,7 +9,6 @@
'''Overridden Mercurial commands and functions for the largefiles extension'''
from __future__ import absolute_import
-import contextlib
import copy
import os
@@ -50,6 +49,8 @@
eh = exthelper.exthelper()
+lfstatus = lfutil.lfstatus
+
# -- Utility functions: commonly/repeatedly needed functionality ---------------
@@ -158,16 +159,6 @@
return added, bad
-@contextlib.contextmanager
-def lfstatus(repo):
- oldvalue = getattr(repo, 'lfstatus', False)
- repo.lfstatus = True
- try:
- yield
- finally:
- repo.lfstatus = oldvalue
-
-
def removelargefiles(ui, repo, isaddremove, matcher, uipathfn, dryrun, **opts):
after = opts.get('after')
m = composelargefilematcher(matcher, repo[None].manifest())