diff hgext/mq.py @ 2836:307439d6fede

mq: do not allow to push from repo with patches applied
author Vadim Gelfer <vadim.gelfer@gmail.com>
date Thu, 10 Aug 2006 16:10:59 -0700
parents addd03c7fbfa
children 851b07ec450c
line wrap: on
line diff
--- a/hgext/mq.py
+++ b/hgext/mq.py
@@ -1832,18 +1832,28 @@ def select(ui, repo, *args, **opts):
 
 def reposetup(ui, repo):
     class mqrepo(repo.__class__):
+        def abort_if_wdir_patched(self, errmsg, force=False):
+            if self.mq.applied and not force:
+                parent = revlog.hex(self.dirstate.parents()[0])
+                if parent in [s.rev for s in self.mq.applied]:
+                    raise util.Abort(errmsg)
+            
         def commit(self, *args, **opts):
             if len(args) >= 6:
                 force = args[5]
             else:
                 force = opts.get('force')
-            if self.mq.applied and not force:
-                parent = revlog.hex(self.dirstate.parents()[0])
-                if parent in [s.rev for s in self.mq.applied]:
-                    raise util.Abort(_('cannot commit over an applied mq patch'))
+            self.abort_if_wdir_patched(
+                _('cannot commit over an applied mq patch'),
+                force)
 
             return super(mqrepo, self).commit(*args, **opts)
 
+        def push(self, remote, force=False, revs=None):
+            if self.mq.applied and not force:
+                raise util.Abort(_('source has mq patches applied'))
+            return super(mqrepo, self).push(remote, force, revs)
+            
         def tags(self):
             if self.tagscache:
                 return self.tagscache