Mercurial Repositories > eric / file diff

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/eric6/MicroPython/ConnectionSelectionDialog.py	Wed Feb 24 20:17:35 2021 +0100
@@ -0,0 +1,98 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2021 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing a dialog to select the port to connect to and the type of
+the attached device.
+"""
+
+from PyQt5.QtCore import pyqtSlot, Qt
+from PyQt5.QtWidgets import QDialog, QDialogButtonBox
+
+from .Ui_ConnectionSelectionDialog import Ui_ConnectionSelectionDialog
+
+from . import MicroPythonDevices
+
+
+class ConnectionSelectionDialog(QDialog, Ui_ConnectionSelectionDialog):
+    """
+    Class implementing a dialog to select the port to connect to and the type
+    of the attached device.
+    """
+    PortNameRole = Qt.UserRole
+    VidPidRole = Qt.UserRole + 1
+    
+    def __init__(self, ports, currentPort, currentType, parent=None):
+        """
+        Constructor
+        
+        @param ports list of detected ports
+        @type list of str
+        @param currentPort port most recently selected
+        @type str
+        @param currentType device type most recently selected
+        @type str
+        @param parent reference to the parent widget (defaults to None)
+        @type QWidget (optional)
+        """
+        super(ConnectionSelectionDialog, self).__init__(parent)
+        self.setupUi(self)
+        
+        for index, (vid, pid, description, portName) in enumerate(
+            sorted(ports, key=lambda x: x[3])
+        ):
+            self.portNameComboBox.addItem(
+                self.tr("{0} - {1}", "description - port name")
+                .format(description, portName))
+            self.portNameComboBox.setItemData(
+                index, portName, self.PortNameRole)
+            self.portNameComboBox.setItemData(
+                index, (vid, pid), self.VidPidRole)
+        self.portNameComboBox.setCurrentText(currentPort)
+        
+        self.deviceTypeComboBox.addItem("", "")
+        for board, description in sorted(
+            MicroPythonDevices.getSupportedDevices(),
+            key=lambda x: x[1]
+        ):
+            self.deviceTypeComboBox.addItem(description, board)
+        index = self.deviceTypeComboBox.findData(currentType)
+        self.deviceTypeComboBox.setCurrentIndex(index)
+        
+        self.__updateOK()
+        
+        msh = self.minimumSizeHint()
+        self.resize(max(self.width(), msh.width()), msh.height())
+    
+    def __updateOK(self):
+        """
+        Private method to update the status of the OK button.
+        """
+        self.buttonBox.button(QDialogButtonBox.Ok).setEnabled(
+            bool(self.deviceTypeComboBox.currentData()))
+    
+    @pyqtSlot(str)
+    def on_deviceTypeComboBox_currentTextChanged(self, txt):
+        """
+        Private slot to handle the selection of a device type.
+        
+        @param txt selected device description
+        @type str
+        """
+        self.__updateOK()
+    
+    def getData(self):
+        """
+        Public method to get the entered data.
+        
+        @return tuple containing the VID, PID and name of the selected port
+            and the selected device type
+        @rtype tuple of (int, int, str, str)
+        """
+        return (
+            *self.portNameComboBox.currentData(self.VidPidRole),
+            self.portNameComboBox.currentData(self.PortNameRole),
+            self.deviceTypeComboBox.currentData(),
+        )