diff hgext/largefiles/uisetup.py @ 35303:67b7e39b441b

largefiles: allow to run 'debugupgraderepo' on repo with largefiles The extensions wrap the necessary function to ensure the 'largefiles' requirements won't be dropped. It is now possible to run `hg debugupgraderepo` on a repository with largefiles.
author Boris Feld <boris.feld@octobus.net>
date Thu, 07 Dec 2017 01:53:14 +0100
parents a9d8caf95941
children 95a9be56c3bb
line wrap: on
line diff
--- a/hgext/largefiles/uisetup.py	Thu Dec 07 01:51:54 2017 +0100
+++ b/hgext/largefiles/uisetup.py	Thu Dec 07 01:53:14 2017 +0100
@@ -30,6 +30,7 @@
     scmutil,
     sshpeer,
     subrepo,
+    upgrade,
     wireproto,
 )
 
@@ -60,6 +61,12 @@
 
     extensions.wrapfunction(copies, 'pathcopies', overrides.copiespathcopies)
 
+    extensions.wrapfunction(upgrade, 'preservedrequirements',
+                            overrides.upgraderequirements)
+
+    extensions.wrapfunction(upgrade, 'supporteddestrequirements',
+                            overrides.upgraderequirements)
+
     # Subrepos call status function
     entry = extensions.wrapcommand(commands.table, 'status',
                                    overrides.overridestatus)