diff hgext/largefiles/overrides.py @ 28878:a75c9665ef06

largefiles: introduce push --lfrev to control which revisions are pushed The default of pushing all largefiles referenced in outgoing revisions is safe, but also expensive and sometimes not what is needed. We thus introduce a --lfrev option, similar to what pull already has. By specifying an empty set of revisions (or null), it is possible to get lazy (and insecure!) pushes of revisions without referenced largefiles, similar to how pull works.
author Mads Kiilerich <madski@unity3d.com>
date Sun, 27 Mar 2016 13:00:28 -0700
parents a7f7b7acf489
children 814076f4ace3
line wrap: on
line diff
--- a/hgext/largefiles/overrides.py	Wed Apr 13 01:45:45 2016 +0200
+++ b/hgext/largefiles/overrides.py	Sun Mar 27 13:00:28 2016 -0700
@@ -801,6 +801,21 @@
         ui.status(_("%d largefiles cached\n") % numcached)
     return result
 
+def overridepush(orig, ui, repo, *args, **kwargs):
+    """Override push command and store --lfrev parameters in opargs"""
+    lfrevs = kwargs.pop('lfrev', None)
+    if lfrevs:
+        opargs = kwargs.setdefault('opargs', {})
+        opargs['lfrevs'] = scmutil.revrange(repo, lfrevs)
+    return orig(ui, repo, *args, **kwargs)
+
+def exchangepushoperation(orig, *args, **kwargs):
+    """Override pushoperation constructor and store lfrevs parameter"""
+    lfrevs = kwargs.pop('lfrevs', None)
+    pushop = orig(*args, **kwargs)
+    pushop.lfrevs = lfrevs
+    return pushop
+
 revsetpredicate = registrar.revsetpredicate()
 
 @revsetpredicate('pulled()')