AaronFiore/electrum
1
0
Fork 0
Code Activity

qt network dialog: use intenum for columns

Browse Source
This commit is contained in:
SomberNight
2019-02-03 20:04:33 +01:00
parent 905e271db9
commit 43487910c7
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
electrum/gui/qt/network_dialog.py
View File

@@ -24,6 +24,7 @@
# SOFTWARE. # SOFTWARE.
import socket import socket
from enum import IntEnum
from PyQt5.QtGui import * from PyQt5.QtGui import *
from PyQt5.QtCore import * from PyQt5.QtCore import *
@@ -132,6 +133,11 @@ class NodesListWidget(QTreeWidget):
class ServerListWidget(QTreeWidget): class ServerListWidget(QTreeWidget):
class Columns(IntEnum):
HOST = 0
PORT = 1
SERVER_STR_ROLE = Qt.UserRole + 100
def __init__(self, parent): def __init__(self, parent):
QTreeWidget.__init__(self) QTreeWidget.__init__(self)
@@ -145,7 +151,7 @@ class ServerListWidget(QTreeWidget):
if not item: if not item:
return return
menu = QMenu() menu = QMenu()
server = item.data(1, Qt.UserRole) server = item.data(self.Columns.HOST, self.SERVER_STR_ROLE)
menu.addAction(_("Use as server"), lambda: self.set_server(server)) menu.addAction(_("Use as server"), lambda: self.set_server(server))
menu.exec_(self.viewport().mapToGlobal(position)) menu.exec_(self.viewport().mapToGlobal(position))
@@ -176,13 +182,13 @@ class ServerListWidget(QTreeWidget):
if port: if port:
x = QTreeWidgetItem([_host, port]) x = QTreeWidgetItem([_host, port])
server = serialize_server(_host, port, protocol) server = serialize_server(_host, port, protocol)
x.setData(1, Qt.UserRole, server) x.setData(self.Columns.HOST, self.SERVER_STR_ROLE, server)
self.addTopLevelItem(x) self.addTopLevelItem(x)
h = self.header() h = self.header()
h.setStretchLastSection(False) h.setStretchLastSection(False)
h.setSectionResizeMode(0, QHeaderView.Stretch) h.setSectionResizeMode(self.Columns.HOST, QHeaderView.Stretch)
h.setSectionResizeMode(1, QHeaderView.ResizeToContents) h.setSectionResizeMode(self.Columns.PORT, QHeaderView.ResizeToContents)
super().update() super().update()