hgext/largefiles/lfutil.py
changeset 43583 73e6d3346e4f
parent 43085 eef9a2d67051
child 43584 a02e4c12ae60
--- 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: