diff -r 4e8b98454baa -r 800c432b34c8 eric7/Plugins/VcsPlugins/vcsMercurial/CloseheadExtension/HgCloseHeadSelectionDialog.py
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/eric7/Plugins/VcsPlugins/vcsMercurial/CloseheadExtension/HgCloseHeadSelectionDialog.py Sat May 15 18:45:04 2021 +0200
@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2019 - 2021 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing a dialog to select the heads to be closed.
+"""
+
+from PyQt5.QtCore import pyqtSlot
+from PyQt5.QtWidgets import QDialog, QDialogButtonBox, QTreeWidgetItem
+
+from .Ui_HgCloseHeadSelectionDialog import Ui_HgCloseHeadSelectionDialog
+
+
+class HgCloseHeadSelectionDialog(QDialog, Ui_HgCloseHeadSelectionDialog):
+ """
+ Class implementing a dialog to select the heads to be closed.
+ """
+ def __init__(self, vcs, parent=None):
+ """
+ Constructor
+
+ @param vcs reference to the VCS object
+ @type Hg
+ @param parent reference to the parent widget
+ @type QWidget
+ """
+ super().__init__(parent)
+ self.setupUi(self)
+
+ self.buttonBox.button(
+ QDialogButtonBox.StandardButton.Cancel).setDefault(True)
+ self.buttonBox.button(
+ QDialogButtonBox.StandardButton.Ok).setEnabled(False)
+
+ heads = self.__getHeads(vcs)
+ for revision, branch in heads:
+ QTreeWidgetItem(self.headsList, [revision, branch])
+
+ def __getHeads(self, vcs):
+ """
+ Private method to get the open heads.
+
+ @param vcs reference to the VCS object
+ @type Hg
+ @return list of tuples containing the revision and the corresponding
+ branch name
+ @rtype list of tuples of (str, str)
+ """
+ args = vcs.initCommand("heads")
+ args.append('--template')
+ args.append('{node|short}@@@{branches}\n')
+
+ output = ""
+ client = vcs.getClient()
+ output, error = client.runcommand(args)
+
+ heads = []
+ if output:
+ for line in output.splitlines():
+ line = line.strip()
+ if line:
+ revision, branch = line.split("@@@")
+ heads.append((revision, branch))
+
+ return heads
+
+ @pyqtSlot()
+ def on_headsList_itemSelectionChanged(self):
+ """
+ Private slot handling changes of the selection.
+ """
+ self.buttonBox.button(QDialogButtonBox.StandardButton.Ok).setEnabled(
+ len(self.headsList.selectedItems()) > 0
+ )
+
+ def getData(self):
+ """
+ Public method to retrieve the entered data.
+
+ @return tuple containing a list of selected revisions and the commit
+ message
+ @rtype tuple of (list of str, str)
+ """
+ revisions = [itm.text(0) for itm in self.headsList.selectedItems()]
+ message = self.logEdit.toPlainText().strip()
+
+ return revisions, message
