Mercurial Repositories > eric / file diff

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Plugins/VcsPlugins/vcsMercurial/TransplantExtension/ProjectHelper.py	Mon May 30 16:53:13 2011 +0200
@@ -0,0 +1,128 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2011 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing the transplant extension project helper.
+"""
+
+from PyQt4.QtCore import QObject
+from PyQt4.QtGui import QMenu
+
+from E5Gui.E5Action import E5Action
+from E5Gui import E5MessageBox
+
+
+class TransplantProjectHelper(QObject):
+    """
+    Class implementing the transplant extension project helper.
+    """
+    def __init__(self):
+        """
+        Constructor
+        """
+        QObject.__init__(self)
+        
+        self.actions = []
+        
+        self.initActions()
+    
+    def setObjects(self, vcsObject, projectObject):
+        """
+        Public method to set references to the vcs and project objects.
+        
+        @param vcsObject reference to the vcs object
+        @param projectObject reference to the project object
+        """
+        self.vcs = vcsObject
+        self.project = projectObject
+    
+    def getActions(self):
+        """
+        Public method to get a list of all actions.
+        
+        @return list of all actions (list of E5Action)
+        """
+        return self.actions[:]
+    
+    def initActions(self):
+        """
+        Public method to generate the action objects.
+        """
+        self.hgTransplantAct = E5Action(self.trUtf8('Transplant Changesets'),
+                self.trUtf8('Transplant Changesets'),
+                0, 0, self, 'mercurial_transplant')
+        self.hgTransplantAct.setStatusTip(self.trUtf8(
+            'Transplant changesets from another branch'
+        ))
+        self.hgTransplantAct.setWhatsThis(self.trUtf8(
+            """<b>Transplant Changesets</b>"""
+            """<p>This transplants changesets from another branch on top of the"""
+            """ current working directory with the log of the original changeset.</p>"""
+        ))
+        self.hgTransplantAct.triggered[()].connect(self.__hgTransplant)
+        self.actions.append(self.hgTransplantAct)
+        
+        self.hgTransplantContinueAct = E5Action(
+                self.trUtf8('Continue'),
+                self.trUtf8('Continue'),
+                0, 0, self, 'mercurial_transplant_continue')
+        self.hgTransplantContinueAct.setStatusTip(self.trUtf8(
+            'Continue the last transplant session after repair'
+        ))
+        self.hgTransplantContinueAct.setWhatsThis(self.trUtf8(
+            """<b>Continue</b>"""
+            """<p>This continues the last transplant session after repair.</p>"""
+        ))
+        self.hgTransplantContinueAct.triggered[()].connect(self.__hgTransplantContinue)
+        self.actions.append(self.hgTransplantContinueAct)
+    
+    def initMenu(self, mainMenu):
+        """
+        Public method to generate the extension menu.
+        
+        @param mainMenu reference to the main menu (QMenu)
+        @return populated menu (QMenu)
+        """
+        menu = QMenu(self.menuTitle(), mainMenu)
+        menu.setTearOffEnabled(True)
+        
+        menu.addAction(self.hgTransplantAct)
+        menu.addAction(self.hgTransplantContinueAct)
+        
+        return menu
+    
+    def menuTitle(self):
+        """
+        Public method to get the menu title.
+        """
+        return self.trUtf8("Transplant")
+    
+    def __hgTransplant(self):
+        """
+        Private slot used to transplant changesets from another branch.
+        """
+        shouldReopen = self.vcs.getExtensionObject("transplant")\
+            .hgTransplant(self.project.getProjectPath())
+        if shouldReopen:
+            res = E5MessageBox.yesNo(None,
+                self.trUtf8("Transplant Changesets"),
+                self.trUtf8("""The project should be reread. Do this now?"""),
+                yesDefault=True)
+            if res:
+                self.project.reopenProject()
+    
+    def __hgTransplantContinue(self):
+        """
+        Private slot used to continue the last transplant session after repair.
+        """
+        shouldReopen = self.vcs.getExtensionObject("transplant")\
+            .hgTransplantContinue(self.project.getProjectPath())
+        if shouldReopen:
+            res = E5MessageBox.yesNo(None,
+                self.trUtf8("Transplant Changesets (Continue)"),
+                self.trUtf8("""The project should be reread. Do this now?"""),
+                yesDefault=True)
+            if res:
+                self.project.reopenProject()