def mouse_down(self, e):
p = e.position
row_num = self.find_row(p)
col_id = self.find_column(p)
print "Mouse down at", p, "in cell", (row_num, col_id)
if row_num >= 0 and col_id:
print "Cell rect =", self.cell_rect(row_num, col_id)
TableView.mouse_down(self, e)
def __init__(self, parent, *args, **kwargs):
super(DevicesListWidget, self).__init__(*args, **kwargs)
self.setWindowTitle("Devices list")
self.setWindowState(Qt.WindowMaximized)
self.setLayout(VLayout(margin=0, spacing=0))
self.mqtt = parent.mqtt
self.mdi = parent.mdi
self.idx = None
self.nam = QNetworkAccessManager()
self.backup = bytes()
self.settings = QSettings("{}/TDM/tdm.cfg".format(QDir.homePath()),
QSettings.IniFormat)
self.hidden_columns = self.settings.value("hidden_columns", [1, 2])
self.tb = Toolbar(Qt.Horizontal, 16, Qt.ToolButtonTextBesideIcon)
self.tb.addAction(QIcon("GUI/icons/add.png"), "Add", self.device_add)
self.layout().addWidget(self.tb)
self.device_list = TableView()
self.model = parent.device_model
self.telemetry_model = parent.telemetry_model
self.sorted_device_model = QSortFilterProxyModel()
self.sorted_device_model.setSourceModel(parent.device_model)
self.device_list.setModel(self.sorted_device_model)
self.device_list.setupColumns(columns, self.hidden_columns)
self.device_list.setSortingEnabled(True)
self.device_list.setWordWrap(True)
self.device_list.setItemDelegate(DeviceDelegate())
self.device_list.sortByColumn(DevMdl.TOPIC, Qt.AscendingOrder)
self.device_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.layout().addWidget(self.device_list)
self.device_list.clicked.connect(self.select_device)
self.device_list.doubleClicked.connect(self.device_config)
self.device_list.customContextMenuRequested.connect(
self.show_list_ctx_menu)
self.device_list.horizontalHeader().setContextMenuPolicy(
Qt.CustomContextMenu)
self.device_list.horizontalHeader().customContextMenuRequested.connect(
self.show_header_ctx_menu)
self.ctx_menu = QMenu()
self.ctx_menu_relays = None
self.create_actions()
self.build_header_ctx_menu()
class DevicesListWidget(QWidget):
def __init__(self, parent, *args, **kwargs):
super(DevicesListWidget, self).__init__(*args, **kwargs)
self.setWindowTitle("Devices list")
self.setWindowState(Qt.WindowMaximized)
self.setLayout(VLayout(margin=0, spacing=0))
self.mqtt = parent.mqtt
self.mdi = parent.mdi
self.idx = None
self.nam = QNetworkAccessManager()
self.backup = bytes()
self.settings = QSettings("{}/TDM/tdm.cfg".format(QDir.homePath()),
QSettings.IniFormat)
self.hidden_columns = self.settings.value("hidden_columns", [1, 2])
self.tb = Toolbar(Qt.Horizontal, 16, Qt.ToolButtonTextBesideIcon)
self.tb.addAction(QIcon("GUI/icons/add.png"), "Add", self.device_add)
self.layout().addWidget(self.tb)
self.device_list = TableView()
self.model = parent.device_model
self.telemetry_model = parent.telemetry_model
self.sorted_device_model = QSortFilterProxyModel()
self.sorted_device_model.setSourceModel(parent.device_model)
self.device_list.setModel(self.sorted_device_model)
self.device_list.setupColumns(columns, self.hidden_columns)
self.device_list.setSortingEnabled(True)
self.device_list.setWordWrap(True)
self.device_list.setItemDelegate(DeviceDelegate())
self.device_list.sortByColumn(DevMdl.TOPIC, Qt.AscendingOrder)
self.device_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.layout().addWidget(self.device_list)
self.device_list.clicked.connect(self.select_device)
self.device_list.doubleClicked.connect(self.device_config)
self.device_list.customContextMenuRequested.connect(
self.show_list_ctx_menu)
self.device_list.horizontalHeader().setContextMenuPolicy(
Qt.CustomContextMenu)
self.device_list.horizontalHeader().customContextMenuRequested.connect(
self.show_header_ctx_menu)
self.ctx_menu = QMenu()
self.ctx_menu_relays = None
self.create_actions()
self.build_header_ctx_menu()
def create_actions(self):
self.ctx_menu.addAction(QIcon("GUI/icons/configure.png"), "Configure",
self.device_config)
self.ctx_menu.addAction(QIcon("GUI/icons/delete.png"), "Remove",
self.device_delete)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/refresh.png"), "Refresh",
self.ctx_menu_refresh)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/on.png"), "Power ON",
lambda: self.ctx_menu_power(state="ON"))
self.ctx_menu.addAction(QIcon("GUI/icons/off.png"), "Power OFF",
lambda: self.ctx_menu_power(state="OFF"))
self.ctx_menu_relays = QMenu("Relays")
self.ctx_menu_relays.setIcon(QIcon("GUI/icons/switch.png"))
relays_btn = self.ctx_menu.addMenu(self.ctx_menu_relays)
self.ctx_menu_relays.setEnabled(False)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/clear.png"), "Clear retained",
self.ctx_menu_clean_retained)
self.ctx_menu.addSeparator()
self.ctx_menu_copy = QMenu("Copy")
self.ctx_menu_copy.setIcon(QIcon("GUI/icons/copy.png"))
copy_btn = self.ctx_menu.addMenu(self.ctx_menu_copy)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction("Set teleperiod", self.ctx_menu_teleperiod)
self.ctx_menu.addAction(QIcon("GUI/icons/restart.png"), "Restart",
self.ctx_menu_restart)
self.ctx_menu.addAction(QIcon("GUI/icons/web.png"), "Open WebUI",
self.ctx_menu_webui)
self.ctx_menu.addSeparator()
self.ctx_menu_ota = QMenu("OTA upgrade")
self.ctx_menu_ota.addAction("Set OTA URL", self.ctx_menu_ota_set_url)
self.ctx_menu_ota.addAction("Upgrade", self.ctx_menu_ota_set_upgrade)
ota_btn = self.ctx_menu.addMenu(self.ctx_menu_ota)
self.ctx_menu.addAction("Config backup", self.ctx_menu_config_backup)
self.ctx_menu_copy.addAction(
"IP", lambda: self.ctx_menu_copy_value(DevMdl.IP))
self.ctx_menu_copy.addAction(
"MAC", lambda: self.ctx_menu_copy_value(DevMdl.MAC))
self.ctx_menu_copy.addAction("BSSID", self.ctx_menu_copy_bssid)
self.ctx_menu_copy.addSeparator()
self.ctx_menu_copy.addAction(
"Topic", lambda: self.ctx_menu_copy_value(DevMdl.TOPIC))
self.ctx_menu_copy.addAction(
"FullTopic", lambda: self.ctx_menu_copy_value(DevMdl.FULL_TOPIC))
self.ctx_menu_copy.addAction(
"STAT topic", lambda: self.ctx_menu_copy_prefix_topic("STAT"))
self.ctx_menu_copy.addAction(
"CMND topic", lambda: self.ctx_menu_copy_prefix_topic("CMND"))
self.ctx_menu_copy.addAction(
"TELE topic", lambda: self.ctx_menu_copy_prefix_topic("TELE"))
self.tb.addActions(self.ctx_menu.actions())
self.tb.widgetForAction(ota_btn).setPopupMode(QToolButton.InstantPopup)
self.tb.widgetForAction(relays_btn).setPopupMode(
QToolButton.InstantPopup)
self.tb.widgetForAction(copy_btn).setPopupMode(
QToolButton.InstantPopup)
def ctx_menu_copy_value(self, column):
if self.idx:
row = self.idx.row()
value = self.model.data(self.model.index(row, column))
QApplication.clipboard().setText(value)
def ctx_menu_copy_bssid(self):
if self.idx:
QApplication.clipboard().setText(self.model.bssid(self.idx))
def ctx_menu_copy_prefix_topic(self, prefix):
if self.idx:
if prefix == "STAT":
topic = self.model.statTopic(self.idx)
elif prefix == "CMND":
topic = self.model.commandTopic(self.idx)
elif prefix == "TELE":
topic = self.model.teleTopic(self.idx)
QApplication.clipboard().setText(topic)
def ctx_menu_clean_retained(self):
if self.idx:
relays = self.model.data(
self.model.index(self.idx.row(), DevMdl.POWER))
if relays and len(relays.keys()) > 0:
cmnd_topic = self.model.commandTopic(self.idx)
for r in relays.keys():
self.mqtt.publish(cmnd_topic + r, retain=True)
QMessageBox.information(self, "Clear retained",
"Cleared retained messages.")
def ctx_menu_power(self, relay=None, state=None):
if self.idx:
relays = self.model.data(
self.model.index(self.idx.row(), DevMdl.POWER))
cmnd_topic = self.model.commandTopic(self.idx)
if relay:
self.mqtt.publish(cmnd_topic + relay, payload=state)
elif relays:
for r in relays.keys():
self.mqtt.publish(cmnd_topic + r, payload=state)
def ctx_menu_restart(self):
if self.idx:
self.mqtt.publish("{}/restart".format(
self.model.commandTopic(self.idx)),
payload="1")
def ctx_menu_refresh(self):
if self.idx:
for q in initial_queries:
self.mqtt.publish("{}/status".format(
self.model.commandTopic(self.idx)),
payload=q)
def ctx_menu_teleperiod(self):
if self.idx:
teleperiod, ok = QInputDialog.getInt(
self, "Set telemetry period",
"Input 1 to reset to default\n[Min: 10, Max: 3600]",
int(
self.model.data(
self.model.index(self.idx.row(),
DevMdl.TELEPERIOD))), 1, 3600)
if ok:
if teleperiod != 1 and teleperiod < 10:
teleperiod = 10
self.mqtt.publish("{}/teleperiod".format(
self.model.commandTopic(self.idx)),
payload=teleperiod)
def ctx_menu_telemetry(self):
if self.idx:
self.mqtt.publish("{}/status".format(
self.model.commandTopic(self.idx)),
payload=8)
def ctx_menu_webui(self):
if self.idx:
QDesktopServices.openUrl(
QUrl("http://{}".format(self.model.ip(self.idx))))
def ctx_menu_config_backup(self):
if self.idx:
self.backup = bytes()
ip = self.model.data(self.model.index(self.idx.row(), DevMdl.IP))
self.dl = self.nam.get(
QNetworkRequest(QUrl("http://{}/dl".format(ip))))
self.dl.readyRead.connect(self.get_dump)
self.dl.finished.connect(self.save_dump)
def ctx_menu_ota_set_url(self):
if self.idx:
current_url = self.model.data(
self.model.index(self.idx.row(), DevMdl.OTA_URL))
url, ok = QInputDialog.getText(
self,
"Set OTA URL",
'100 chars max. Set to "1" to reset to default.',
text=current_url)
if ok:
self.mqtt.publish("{}/otaurl".format(
self.model.commandTopic(self.idx)),
payload=url)
def ctx_menu_ota_set_upgrade(self):
if self.idx:
current_url = self.model.data(
self.model.index(self.idx.row(), DevMdl.OTA_URL))
if QMessageBox.question(
self, "OTA Upgrade",
"Are you sure to OTA upgrade from\n{}".format(current_url),
QMessageBox.Yes | QMessageBox.No) == QMessageBox.Yes:
self.model.setData(
self.model.index(self.idx.row(), DevMdl.FIRMWARE),
"Upgrade in progress")
self.mqtt.publish("{}/upgrade".format(
self.model.commandTopic(self.idx)),
payload="1")
def show_list_ctx_menu(self, at):
self.select_device(self.device_list.indexAt(at))
self.ctx_menu.popup(self.device_list.viewport().mapToGlobal(at))
def build_header_ctx_menu(self):
self.hdr_ctx_menu = QMenu()
for c in columns.keys():
a = self.hdr_ctx_menu.addAction(columns[c][0])
a.setData(c)
a.setCheckable(True)
a.setChecked(not self.device_list.isColumnHidden(c))
a.toggled.connect(self.header_ctx_menu_toggle_col)
def show_header_ctx_menu(self, at):
self.hdr_ctx_menu.popup(
self.device_list.horizontalHeader().viewport().mapToGlobal(at))
def header_ctx_menu_toggle_col(self, state):
self.device_list.setColumnHidden(self.sender().data(), not state)
hidden_columns = [
int(c) for c in columns.keys()
if self.device_list.isColumnHidden(c)
]
self.settings.setValue("hidden_columns", hidden_columns)
self.settings.sync()
def select_device(self, idx):
self.idx = self.sorted_device_model.mapToSource(idx)
self.device = self.model.data(self.model.index(idx.row(),
DevMdl.TOPIC))
relays = self.model.data(self.model.index(self.idx.row(),
DevMdl.POWER))
if relays and len(relays.keys()) > 1:
self.ctx_menu_relays.setEnabled(True)
self.ctx_menu_relays.setEnabled(True)
self.ctx_menu_relays.clear()
for r in relays.keys():
actR = self.ctx_menu_relays.addAction("{} ON".format(r))
actR.triggered.connect(
lambda st, x=r: self.ctx_menu_power(x, "ON"))
actR = self.ctx_menu_relays.addAction("{} OFF".format(r))
actR.triggered.connect(
lambda st, x=r: self.ctx_menu_power(x, "OFF"))
self.ctx_menu_relays.addSeparator()
else:
self.ctx_menu_relays.setEnabled(False)
self.ctx_menu_relays.clear()
def device_config(self, idx=None):
if self.idx:
dev_cfg = DevicesConfigWidget(self, self.model.topic(self.idx))
self.mdi.addSubWindow(dev_cfg)
dev_cfg.setWindowState(Qt.WindowMaximized)
def device_add(self):
rc = self.model.rowCount()
self.model.insertRow(rc)
dlg = DeviceEditDialog(self.model, rc)
dlg.full_topic.setText("%prefix%/%topic%/")
if dlg.exec_() == QDialog.Accepted:
self.model.setData(
self.model.index(rc, DevMdl.FRIENDLY_NAME),
self.model.data(self.model.index(rc, DevMdl.TOPIC)))
topic = dlg.topic.text()
tele_dev = self.telemetry_model.addDevice(TasmotaDevice, topic)
self.telemetry_model.devices[topic] = tele_dev
else:
self.model.removeRow(rc)
def device_delete(self):
if self.idx:
topic = self.model.topic(self.idx)
if QMessageBox.question(
self, "Confirm",
"Do you want to remove '{}' from devices list?".format(
topic)) == QMessageBox.Yes:
self.model.removeRows(self.idx.row(), 1)
tele_idx = self.telemetry_model.devices.get(topic)
if tele_idx:
self.telemetry_model.removeRows(tele_idx.row(), 1)
def get_dump(self):
self.backup += self.dl.readAll()
def save_dump(self):
fname = self.dl.header(QNetworkRequest.ContentDispositionHeader)
if fname:
fname = fname.split('=')[1]
save_file = QFileDialog.getSaveFileName(
self, "Save config backup",
"{}/TDM/{}".format(QDir.homePath(), fname))[0]
if save_file:
with open(save_file, "wb") as f:
f.write(self.backup)
def closeEvent(self, event):
event.ignore()
class DevicesListWidget(QWidget):
def __init__(self, parent, *args, **kwargs):
super(DevicesListWidget, self).__init__(*args, **kwargs)
self.setWindowTitle("Devices list")
self.setWindowState(Qt.WindowMaximized)
self.setLayout(VLayout(margin=0, spacing=0))
self.mqtt = parent.mqtt
self.mdi = parent.mdi
self.idx = None
self.settings = QSettings()
self.hidden_columns = self.settings.value("hidden_columns", [1, 2])
self.tb = Toolbar(Qt.Horizontal, 16, Qt.ToolButtonTextBesideIcon)
self.tb.addAction(QIcon("GUI/icons/add.png"), "Add", self.device_add)
self.layout().addWidget(self.tb)
self.device_list = TableView()
self.model = parent.device_model
self.telemetry_model = parent.telemetry_model
self.sorted_device_model = QSortFilterProxyModel()
self.sorted_device_model.setSourceModel(parent.device_model)
self.device_list.setModel(self.sorted_device_model)
self.device_list.setupColumns(columns, self.hidden_columns)
self.device_list.setSortingEnabled(True)
self.device_list.setWordWrap(True)
self.device_list.setItemDelegate(DeviceDelegate())
self.device_list.sortByColumn(DevMdl.TOPIC, Qt.AscendingOrder)
self.device_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.layout().addWidget(self.device_list)
self.device_list.clicked.connect(self.select_device)
self.device_list.doubleClicked.connect(self.device_config)
self.device_list.customContextMenuRequested.connect(self.show_list_ctx_menu)
self.device_list.horizontalHeader().setContextMenuPolicy(Qt.CustomContextMenu)
self.device_list.horizontalHeader().customContextMenuRequested.connect(self.show_header_ctx_menu)
self.ctx_menu = QMenu()
self.ctx_menu_relays = None
self.create_actions()
self.build_header_ctx_menu()
def create_actions(self):
self.ctx_menu.addAction(QIcon("GUI/icons/configure.png"), "Configure", self.device_config)
self.ctx_menu.addAction(QIcon("GUI/icons/delete.png"), "Remove", self.device_delete)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/refresh.png"), "Refresh", self.ctx_menu_refresh)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/on.png"), "Power ON", lambda: self.ctx_menu_power(state="ON"))
self.ctx_menu.addAction(QIcon("GUI/icons/off.png"), "Power OFF", lambda: self.ctx_menu_power(state="OFF"))
self.ctx_menu_relays = QMenu("Relays")
self.ctx_menu_relays.setIcon(QIcon("GUI/icons/switch.png"))
relays_btn = self.ctx_menu.addMenu(self.ctx_menu_relays)
self.ctx_menu_relays.setEnabled(False)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/clear.png"), "Clear retained", self.ctx_menu_clean_retained)
self.ctx_menu.addSeparator()
self.ctx_menu_copy = QMenu("Copy")
self.ctx_menu_copy.setIcon(QIcon("GUI/icons/copy.png"))
copy_btn = self.ctx_menu.addMenu(self.ctx_menu_copy)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/restart.png"), "Restart", self.ctx_menu_restart)
self.ctx_menu.addAction(QIcon("GUI/icons/web.png"), "Open WebUI", self.ctx_menu_webui)
self.ctx_menu_copy.addAction("IP", lambda: self.ctx_menu_copy_value(DevMdl.IP))
self.ctx_menu_copy.addAction("MAC", lambda: self.ctx_menu_copy_value(DevMdl.MAC))
self.ctx_menu_copy.addAction("BSSID", lambda: self.ctx_menu_copy_value(DevMdl.BSSID))
self.ctx_menu_copy.addSeparator()
self.ctx_menu_copy.addAction("Topic", lambda: self.ctx_menu_copy_value(DevMdl.TOPIC))
self.ctx_menu_copy.addAction("FullTopic", lambda: self.ctx_menu_copy_value(DevMdl.FULL_TOPIC))
self.ctx_menu_copy.addAction("STAT topic", lambda: self.ctx_menu_copy_prefix_topic("STAT"))
self.ctx_menu_copy.addAction("CMND topic", lambda: self.ctx_menu_copy_prefix_topic("CMND"))
self.ctx_menu_copy.addAction("TELE topic", lambda: self.ctx_menu_copy_prefix_topic("TELE"))
self.tb.addActions(self.ctx_menu.actions())
self.tb.widgetForAction(relays_btn).setPopupMode(QToolButton.InstantPopup)
self.tb.widgetForAction(copy_btn).setPopupMode(QToolButton.InstantPopup)
def ctx_menu_copy_value(self, column):
if self.idx:
row = self.idx.row()
value = self.model.data(self.model.index(row, column))
QApplication.clipboard().setText(value)
def ctx_menu_copy_prefix_topic(self, prefix):
if self.idx:
if prefix == "STAT":
topic = self.model.statTopic(self.idx)
elif prefix == "CMND":
topic = self.model.commandTopic(self.idx)
elif prefix == "TELE":
topic = self.model.teleTopic(self.idx)
QApplication.clipboard().setText(topic)
def ctx_menu_clean_retained(self):
if self.idx:
relays = self.model.data(self.model.index(self.idx.row(), DevMdl.POWER))
if relays and len(relays.keys()>1):
cmnd_topic = self.model.cmndTopic(self.idx)
for r in relays.keys():
self.mqtt.publish(cmnd_topic + r, retain=True)
QMessageBox.information(self, "Clear retained", "Cleared reatined messages.")
def ctx_menu_power(self, relay=None, state=None):
if self.idx:
relays = self.model.data(self.model.index(self.idx.row(), DevMdl.POWER))
cmnd_topic = self.model.commandTopic(self.idx)
if relay:
self.mqtt.publish(cmnd_topic+relay, payload=state)
elif relays:
for r in relays.keys():
self.mqtt.publish(cmnd_topic+r, payload=state)
def ctx_menu_restart(self):
if self.idx:
self.mqtt.publish("{}/restart".format(self.model.commandTopic(self.idx)), payload="1")
def ctx_menu_refresh(self):
if self.idx:
for q in initial_queries:
self.mqtt.publish("{}/status".format(self.model.commandTopic(self.idx)), payload=q)
def ctx_menu_telemetry(self):
if self.idx:
self.mqtt.publish("{}/status".format(self.model.commandTopic(self.idx)), payload=8)
def ctx_menu_bssid(self):
if self.idx:
bssid = self.model.bssid(self.idx)
current = self.settings.value("BSSID/{}".format(bssid), "")
alias, ok = QInputDialog.getText(self, "BSSID alias", "Alias for {}. Clear to remove.".format(bssid), text=current)
if ok:
self.settings.setValue("BSSID/{}".format(bssid), alias)
self.model.refreshBSSID()
def ctx_menu_webui(self):
if self.idx:
QDesktopServices.openUrl(QUrl("http://{}".format(self.model.ip(self.idx))))
def show_list_ctx_menu(self, at):
self.select_device(self.device_list.indexAt(at))
self.ctx_menu.popup(self.device_list.viewport().mapToGlobal(at))
def build_header_ctx_menu(self):
self.hdr_ctx_menu = QMenu()
for c in columns.keys():
a = self.hdr_ctx_menu.addAction(columns[c][0])
a.setData(c)
a.setCheckable(True)
a.setChecked(not self.device_list.isColumnHidden(c))
a.toggled.connect(self.header_ctx_menu_toggle_col)
def show_header_ctx_menu(self, at):
self.hdr_ctx_menu.popup(self.device_list.horizontalHeader().viewport().mapToGlobal(at))
def header_ctx_menu_toggle_col(self, state):
self.device_list.setColumnHidden(self.sender().data(), not state)
hidden_columns = [int(c) for c in columns.keys() if self.device_list.isColumnHidden(c)]
self.settings.setValue("hidden_columns", hidden_columns)
self.settings.sync()
def select_device(self, idx):
self.idx = self.sorted_device_model.mapToSource(idx)
self.device = self.model.data(self.model.index(idx.row(), DevMdl.TOPIC))
relays = self.model.data(self.model.index(self.idx.row(), DevMdl.POWER))
if relays and len(relays.keys()) > 1:
self.ctx_menu_relays.setEnabled(True)
self.ctx_menu_relays.setEnabled(True)
self.ctx_menu_relays.clear()
for r in relays.keys():
actR = self.ctx_menu_relays.addAction("{} ON".format(r))
actR.triggered.connect(lambda st, x=r: self.ctx_menu_power(x, "ON"))
actR = self.ctx_menu_relays.addAction("{} OFF".format(r))
actR.triggered.connect(lambda st, x=r: self.ctx_menu_power(x, "OFF"))
self.ctx_menu_relays.addSeparator()
else:
self.ctx_menu_relays.setEnabled(False)
self.ctx_menu_relays.clear()
def device_config(self, idx=None):
dev_cfg = DevicesConfigWidget(self, self.model.topic(self.idx))
self.mdi.addSubWindow(dev_cfg)
dev_cfg.setWindowState(Qt.WindowMaximized)
def device_add(self):
rc = self.model.rowCount()
self.model.insertRow(rc)
dlg = DeviceEditDialog(self.model, rc)
dlg.full_topic.setText("%prefix%/%topic%/")
if dlg.exec_() == QDialog.Accepted:
self.model.setData(self.model.index(rc, DevMdl.FRIENDLY_NAME), self.model.data(self.model.index(rc, DevMdl.TOPIC)))
topic = dlg.topic.text()
tele_dev = self.telemetry_model.addDevice(TasmotaDevice, topic)
self.telemetry_model.devices[topic] = tele_dev
else:
self.model.removeRow(rc)
def device_delete(self):
if self.idx:
topic = self.model.topic(self.idx)
if QMessageBox.question(self, "Confirm", "Do you want to remove '{}' from devices list?".format(topic)) == QMessageBox.Yes:
self.model.removeRows(self.idx.row(),1)
tele_idx = self.telemetry_model.devices.get(topic)
if tele_idx:
self.telemetry_model.removeRows(tele_idx.row(),1)
def closeEvent(self, event):
event.ignore()
def __init__(self, parent, *args, **kwargs):
super(ListWidget, self).__init__(*args, **kwargs)
self.setWindowTitle("Devices list")
self.setWindowState(Qt.WindowMaximized)
self.setLayout(VLayout(margin=0, spacing=0))
self.mqtt = parent.mqtt
self.env = parent.env
self.device = None
self.idx = None
self.nam = QNetworkAccessManager()
self.backup = bytes()
self.settings = QSettings("{}/TDM/tdm.cfg".format(QDir.homePath()), QSettings.IniFormat)
views_order = self.settings.value("views_order", [])
self.views = {}
self.settings.beginGroup("Views")
views = self.settings.childKeys()
if views and views_order:
for view in views_order.split(";"):
view_list = self.settings.value(view).split(";")
self.views[view] = base_view + view_list
else:
self.views = default_views
self.settings.endGroup()
self.tb = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonTextBesideIcon)
self.tb_relays = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonIconOnly)
# self.tb_filter = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonTextBesideIcon)
self.tb_views = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonTextBesideIcon)
self.pwm_sliders = []
self.layout().addWidget(self.tb)
self.layout().addWidget(self.tb_relays)
# self.layout().addWidget(self.tb_filter)
self.device_list = TableView()
self.device_list.setIconSize(QSize(24, 24))
self.model = parent.device_model
self.model.setupColumns(self.views["Home"])
self.sorted_device_model = QSortFilterProxyModel()
self.sorted_device_model.setFilterCaseSensitivity(Qt.CaseInsensitive)
self.sorted_device_model.setSourceModel(parent.device_model)
self.sorted_device_model.setSortRole(Qt.InitialSortOrderRole)
self.sorted_device_model.setFilterKeyColumn(-1)
self.device_list.setModel(self.sorted_device_model)
self.device_list.setupView(self.views["Home"])
self.device_list.setSortingEnabled(True)
self.device_list.setWordWrap(True)
self.device_list.setItemDelegate(DeviceDelegate())
self.device_list.sortByColumn(self.model.columnIndex("FriendlyName"), Qt.AscendingOrder)
self.device_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.device_list.verticalHeader().setSectionResizeMode(QHeaderView.ResizeToContents)
self.layout().addWidget(self.device_list)
self.layout().addWidget(self.tb_views)
self.device_list.clicked.connect(self.select_device)
self.device_list.customContextMenuRequested.connect(self.show_list_ctx_menu)
self.ctx_menu = QMenu()
self.ctx_menu_relays = None
self.create_actions()
self.create_view_buttons()
# self.create_view_filter()
self.device_list.doubleClicked.connect(lambda: self.openConsole.emit())
class ListWidget(QWidget):
deviceSelected = pyqtSignal(TasmotaDevice)
openRulesEditor = pyqtSignal()
openConsole = pyqtSignal()
openTelemetry = pyqtSignal()
openWebUI = pyqtSignal()
def __init__(self, parent, *args, **kwargs):
super(ListWidget, self).__init__(*args, **kwargs)
self.setWindowTitle("Devices list")
self.setWindowState(Qt.WindowMaximized)
self.setLayout(VLayout(margin=0, spacing=0))
self.mqtt = parent.mqtt
self.env = parent.env
self.device = None
self.idx = None
self.nam = QNetworkAccessManager()
self.backup = bytes()
self.settings = QSettings("{}/TDM/tdm.cfg".format(QDir.homePath()), QSettings.IniFormat)
views_order = self.settings.value("views_order", [])
self.views = {}
self.settings.beginGroup("Views")
views = self.settings.childKeys()
if views and views_order:
for view in views_order.split(";"):
view_list = self.settings.value(view).split(";")
self.views[view] = base_view + view_list
else:
self.views = default_views
self.settings.endGroup()
self.tb = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonTextBesideIcon)
self.tb_relays = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonIconOnly)
# self.tb_filter = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonTextBesideIcon)
self.tb_views = Toolbar(Qt.Horizontal, 24, Qt.ToolButtonTextBesideIcon)
self.pwm_sliders = []
self.layout().addWidget(self.tb)
self.layout().addWidget(self.tb_relays)
# self.layout().addWidget(self.tb_filter)
self.device_list = TableView()
self.device_list.setIconSize(QSize(24, 24))
self.model = parent.device_model
self.model.setupColumns(self.views["Home"])
self.sorted_device_model = QSortFilterProxyModel()
self.sorted_device_model.setFilterCaseSensitivity(Qt.CaseInsensitive)
self.sorted_device_model.setSourceModel(parent.device_model)
self.sorted_device_model.setSortRole(Qt.InitialSortOrderRole)
self.sorted_device_model.setFilterKeyColumn(-1)
self.device_list.setModel(self.sorted_device_model)
self.device_list.setupView(self.views["Home"])
self.device_list.setSortingEnabled(True)
self.device_list.setWordWrap(True)
self.device_list.setItemDelegate(DeviceDelegate())
self.device_list.sortByColumn(self.model.columnIndex("FriendlyName"), Qt.AscendingOrder)
self.device_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.device_list.verticalHeader().setSectionResizeMode(QHeaderView.ResizeToContents)
self.layout().addWidget(self.device_list)
self.layout().addWidget(self.tb_views)
self.device_list.clicked.connect(self.select_device)
self.device_list.customContextMenuRequested.connect(self.show_list_ctx_menu)
self.ctx_menu = QMenu()
self.ctx_menu_relays = None
self.create_actions()
self.create_view_buttons()
# self.create_view_filter()
self.device_list.doubleClicked.connect(lambda: self.openConsole.emit())
def create_actions(self):
self.ctx_menu_cfg = QMenu("Configure")
self.ctx_menu_cfg.setIcon(QIcon("GUI/icons/settings.png"))
self.ctx_menu_cfg.addAction("Module", self.configureModule)
self.ctx_menu_cfg.addAction("GPIO", self.configureGPIO)
self.ctx_menu_cfg.addAction("Template", self.configureTemplate)
# self.ctx_menu_cfg.addAction("Wifi", self.ctx_menu_teleperiod)
# self.ctx_menu_cfg.addAction("Time", self.cfgTime.emit)
# self.ctx_menu_cfg.addAction("MQTT", self.ctx_menu_teleperiod)
# self.ctx_menu_cfg.addAction("Logging", self.ctx_menu_teleperiod)
self.ctx_menu.addMenu(self.ctx_menu_cfg)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/refresh.png"), "Refresh", self.ctx_menu_refresh)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/clear.png"), "Clear retained", self.ctx_menu_clear_retained)
self.ctx_menu.addAction("Clear Backlog", self.ctx_menu_clear_backlog)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/copy.png"), "Copy", self.ctx_menu_copy)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/restart.png"), "Restart", self.ctx_menu_restart)
self.ctx_menu.addAction(QIcon(), "Reset", self.ctx_menu_reset)
self.ctx_menu.addSeparator()
self.ctx_menu.addAction(QIcon("GUI/icons/delete.png"), "Delete", self.ctx_menu_delete_device)
console = self.tb.addAction(QIcon("GUI/icons/console.png"), "Console", self.openConsole.emit)
console.setShortcut("Ctrl+E")
rules = self.tb.addAction(QIcon("GUI/icons/rules.png"), "Rules", self.openRulesEditor.emit)
rules.setShortcut("Ctrl+R")
self.tb.addAction(QIcon("GUI/icons/timers.png"), "Timers", self.configureTimers)
buttons = self.tb.addAction(QIcon("GUI/icons/buttons.png"), "Buttons", self.configureButtons)
buttons.setShortcut("Ctrl+B")
switches = self.tb.addAction(QIcon("GUI/icons/switches.png"), "Switches", self.configureSwitches)
switches.setShortcut("Ctrl+S")
power = self.tb.addAction(QIcon("GUI/icons/power.png"), "Power", self.configurePower)
power.setShortcut("Ctrl+P")
# setopts = self.tb.addAction(QIcon("GUI/icons/setoptions.png"), "SetOptions", self.configureSO)
# setopts.setShortcut("Ctrl+S")
self.tb.addSpacer()
telemetry = self.tb.addAction(QIcon("GUI/icons/telemetry.png"), "Telemetry", self.openTelemetry.emit)
telemetry.setShortcut("Ctrl+T")
webui = self.tb.addAction(QIcon("GUI/icons/web.png"), "WebUI", self.openWebUI.emit)
webui.setShortcut("Ctrl+U")
# self.tb.addAction(QIcon(), "Multi Command", self.ctx_menu_webui)
self.agAllPower = QActionGroup(self)
self.agAllPower.addAction(QIcon("GUI/icons/P_ON.png"), "All ON")
self.agAllPower.addAction(QIcon("GUI/icons/P_OFF.png"), "All OFF")
self.agAllPower.setEnabled(False)
self.agAllPower.setExclusive(False)
self.agAllPower.triggered.connect(self.toggle_power_all)
self.tb_relays.addActions(self.agAllPower.actions())
self.agRelays = QActionGroup(self)
self.agRelays.setVisible(False)
self.agRelays.setExclusive(False)
for a in range(1, 9):
act = QAction(QIcon("GUI/icons/P{}_OFF.png".format(a)), "")
act.setShortcut("F{}".format(a))
self.agRelays.addAction(act)
self.agRelays.triggered.connect(self.toggle_power)
self.tb_relays.addActions(self.agRelays.actions())
self.tb_relays.addSeparator()
self.actColor = self.tb_relays.addAction(QIcon("GUI/icons/color.png"), "Color", self.set_color)
self.actColor.setEnabled(False)
self.actChannels = self.tb_relays.addAction(QIcon("GUI/icons/sliders.png"), "Channels")
self.actChannels.setEnabled(False)
self.mChannels = QMenu()
self.actChannels.setMenu(self.mChannels)
self.tb_relays.widgetForAction(self.actChannels).setPopupMode(QToolButton.InstantPopup)
def create_view_buttons(self):
self.tb_views.addWidget(QLabel("View mode: "))
ag_views = QActionGroup(self)
ag_views.setExclusive(True)
for v in self.views.keys():
a = QAction(v)
a.triggered.connect(self.change_view)
a.setCheckable(True)
ag_views.addAction(a)
self.tb_views.addActions(ag_views.actions())
ag_views.actions()[0].setChecked(True)
stretch = QWidget()
stretch.setSizePolicy(QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Minimum))
self.tb_views.addWidget(stretch)
# actEditView = self.tb_views.addAction("Edit views...")
# def create_view_filter(self):
# # self.tb_filter.addWidget(QLabel("Show devices: "))
# # self.cbxLWT = QComboBox()
# # self.cbxLWT.addItems(["All", "Online"d, "Offline"])
# # self.cbxLWT.currentTextChanged.connect(self.build_filter_regex)
# # self.tb_filter.addWidget(self.cbxLWT)
#
# self.tb_filter.addWidget(QLabel(" Search: "))
# self.leSearch = QLineEdit()
# self.leSearch.setClearButtonEnabled(True)
# self.leSearch.textChanged.connect(self.build_filter_regex)
# self.tb_filter.addWidget(self.leSearch)
#
# def build_filter_regex(self, txt):
# query = self.leSearch.text()
# # if self.cbxLWT.currentText() != "All":
# # query = "{}|{}".format(self.cbxLWT.currentText(), query)
# self.sorted_device_model.setFilterRegExp(query)
def change_view(self, a=None):
view = self.views[self.sender().text()]
self.model.setupColumns(view)
self.device_list.setupView(view)
def ctx_menu_copy(self):
if self.idx:
string = dumps(self.model.data(self.idx))
if string.startswith('"') and string.endswith('"'):
string = string[1:-1]
QApplication.clipboard().setText(string)
def ctx_menu_clear_retained(self):
if self.device:
relays = self.device.power()
if relays and len(relays.keys()) > 0:
for r in relays.keys():
self.mqtt.publish(self.device.cmnd_topic(r), retain=True)
QMessageBox.information(self, "Clear retained", "Cleared retained messages.")
def ctx_menu_clear_backlog(self):
if self.device:
self.mqtt.publish(self.device.cmnd_topic("backlog"), "")
QMessageBox.information(self, "Clear Backlog", "Backlog cleared.")
def ctx_menu_restart(self):
if self.device:
self.mqtt.publish(self.device.cmnd_topic("restart"), payload="1")
for k in list(self.device.power().keys()):
self.device.p.pop(k)
def ctx_menu_reset(self):
if self.device:
reset, ok = QInputDialog.getItem(self, "Reset device and restart", "Select reset mode", resets, editable=False)
if ok:
self.mqtt.publish(self.device.cmnd_topic("reset"), payload=reset.split(":")[0])
for k in list(self.device.power().keys()):
self.device.p.pop(k)
def ctx_menu_refresh(self):
if self.device:
for k in list(self.device.power().keys()):
self.device.p.pop(k)
for c in initial_commands():
cmd, payload = c
cmd = self.device.cmnd_topic(cmd)
self.mqtt.publish(cmd, payload, 1)
def ctx_menu_delete_device(self):
if self.device:
if QMessageBox.question(self, "Confirm", "Do you want to remove the following device?\n'{}' ({})"
.format(self.device.p['FriendlyName1'], self.device.p['Topic'])) == QMessageBox.Yes:
self.model.deleteDevice(self.idx)
def ctx_menu_teleperiod(self):
if self.device:
teleperiod, ok = QInputDialog.getInt(self, "Set telemetry period", "Input 1 to reset to default\n[Min: 10, Max: 3600]", self.device.p['TelePeriod'], 1, 3600)
if ok:
if teleperiod != 1 and teleperiod < 10:
teleperiod = 10
self.mqtt.publish(self.device.cmnd_topic("teleperiod"), teleperiod)
def ctx_menu_config_backup(self):
if self.device:
self.backup = bytes()
self.dl = self.nam.get(QNetworkRequest(QUrl("http://{}/dl".format(self.device.p['IPAddress']))))
self.dl.readyRead.connect(self.get_dump)
self.dl.finished.connect(self.save_dump)
def ctx_menu_ota_set_url(self):
if self.device:
url, ok = QInputDialog.getText(self, "Set OTA URL", '100 chars max. Set to "1" to reset to default.', text=self.device.p['OtaUrl'])
if ok:
self.mqtt.publish(self.device.cmnd_topic("otaurl"), payload=url)
def ctx_menu_ota_set_upgrade(self):
if self.device:
if QMessageBox.question(self, "OTA Upgrade", "Are you sure to OTA upgrade from\n{}".format(self.device.p['OtaUrl']), QMessageBox.Yes | QMessageBox.No) == QMessageBox.Yes:
self.mqtt.publish(self.device.cmnd_topic("upgrade"), payload="1")
def show_list_ctx_menu(self, at):
self.select_device(self.device_list.indexAt(at))
self.ctx_menu.popup(self.device_list.viewport().mapToGlobal(at))
def select_device(self, idx):
self.idx = self.sorted_device_model.mapToSource(idx)
self.device = self.model.deviceAtRow(self.idx.row())
self.deviceSelected.emit(self.device)
relays = self.device.power()
self.agAllPower.setEnabled(len(relays) >= 1)
for i, a in enumerate(self.agRelays.actions()):
a.setVisible(len(relays) > 1 and i < len(relays))
color = self.device.color().get("Color", False)
has_color = bool(color)
self.actColor.setEnabled(has_color and not self.device.setoption(68))
self.actChannels.setEnabled(has_color)
if has_color:
self.actChannels.menu().clear()
max_val = 100
if self.device.setoption(15) == 0:
max_val = 1023
for k, v in self.device.pwm().items():
channel = SliderAction(self, k)
channel.slider.setMaximum(max_val)
channel.slider.setValue(int(v))
self.mChannels.addAction(channel)
channel.slider.valueChanged.connect(self.set_channel)
dimmer = self.device.color().get("Dimmer")
if dimmer:
saDimmer = SliderAction(self, "Dimmer")
saDimmer.slider.setValue(int(dimmer))
self.mChannels.addAction(saDimmer)
saDimmer.slider.valueChanged.connect(self.set_channel)
def toggle_power(self, action):
if self.device:
idx = self.agRelays.actions().index(action)
relay = list(self.device.power().keys())[idx]
self.mqtt.publish(self.device.cmnd_topic(relay), "toggle")
def toggle_power_all(self, action):
if self.device:
idx = self.agAllPower.actions().index(action)
for r in self.device.power().keys():
self.mqtt.publish(self.device.cmnd_topic(r), str(not bool(idx)))
def set_color(self):
if self.device:
color = self.device.color().get("Color")
if color:
dlg = QColorDialog()
new_color = dlg.getColor(QColor("#{}".format(color)))
if new_color.isValid():
new_color = new_color.name()
if new_color != color:
self.mqtt.publish(self.device.cmnd_topic("color"), new_color)
def set_channel(self, value=0):
cmd = self.sender().objectName()
if self.device:
self.mqtt.publish(self.device.cmnd_topic(cmd), str(value))
def configureSO(self):
if self.device:
dlg = SetOptionsDialog(self.device)
dlg.sendCommand.connect(self.mqtt.publish)
dlg.exec_()
def configureModule(self):
if self.device:
dlg = ModuleDialog(self.device)
dlg.sendCommand.connect(self.mqtt.publish)
dlg.exec_()
def configureGPIO(self):
if self.device:
dlg = GPIODialog(self.device)
dlg.sendCommand.connect(self.mqtt.publish)
dlg.exec_()
def configureTemplate(self):
if self.device:
dlg = TemplateDialog(self.device)
dlg.sendCommand.connect(self.mqtt.publish)
dlg.exec_()
def configureTimers(self):
if self.device:
self.mqtt.publish(self.device.cmnd_topic("timers"))
timers = TimersDialog(self.device)
self.mqtt.messageSignal.connect(timers.parseMessage)
timers.sendCommand.connect(self.mqtt.publish)
timers.exec_()
def configureButtons(self):
if self.device:
backlog = []
buttons = ButtonsDialog(self.device)
if buttons.exec_() == QDialog.Accepted:
for c, cw in buttons.command_widgets.items():
current_value = self.device.p.get(c)
new_value = ""
if isinstance(cw.input, SpinBox):
new_value = cw.input.value()
if isinstance(cw.input, QComboBox):
new_value = cw.input.currentIndex()
if current_value != new_value:
backlog.append("{} {}".format(c, new_value))
for so, sow in buttons.setoption_widgets.items():
current_value = self.device.setoption(so)
new_value = -1
if isinstance(sow.input, SpinBox):
new_value = sow.input.value()
if isinstance(sow.input, QComboBox):
new_value = sow.input.currentIndex()
if current_value != new_value:
backlog.append("SetOption{} {}".format(so, new_value))
if backlog:
backlog.append("status 3")
self.mqtt.publish(self.device.cmnd_topic("backlog"), "; ".join(backlog))
def configureSwitches(self):
if self.device:
backlog = []
switches = SwitchesDialog(self.device)
if switches.exec_() == QDialog.Accepted:
for c, cw in switches.command_widgets.items():
current_value = self.device.p.get(c)
new_value = ""
if isinstance(cw.input, SpinBox):
new_value = cw.input.value()
if isinstance(cw.input, QComboBox):
new_value = cw.input.currentIndex()
if current_value != new_value:
backlog.append("{} {}".format(c, new_value))
for so, sow in switches.setoption_widgets.items():
current_value = self.device.setoption(so)
new_value = -1
if isinstance(sow.input, SpinBox):
new_value = sow.input.value()
if isinstance(sow.input, QComboBox):
new_value = sow.input.currentIndex()
if current_value != new_value:
backlog.append("SetOption{} {}".format(so, new_value))
for sw, sw_mode in enumerate(self.device.p['SwitchMode']):
new_value = switches.sm.inputs[sw].currentIndex()
if sw_mode != new_value:
backlog.append("switchmode{} {}".format(sw+1, new_value))
if backlog:
backlog.append("status")
backlog.append("status 3")
self.mqtt.publish(self.device.cmnd_topic("backlog"), "; ".join(backlog))
def configurePower(self):
if self.device:
backlog = []
power = PowerDialog(self.device)
if power.exec_() == QDialog.Accepted:
for c, cw in power.command_widgets.items():
current_value = self.device.p.get(c)
new_value = ""
if isinstance(cw.input, SpinBox):
new_value = cw.input.value()
if isinstance(cw.input, QComboBox):
new_value = cw.input.currentIndex()
if current_value != new_value:
backlog.append("{} {}".format(c, new_value))
for so, sow in power.setoption_widgets.items():
new_value = -1
if isinstance(sow.input, SpinBox):
new_value = sow.input.value()
if isinstance(sow.input, QComboBox):
new_value = sow.input.currentIndex()
if new_value != self.device.setoption(so):
backlog.append("SetOption{} {}".format(so, new_value))
new_interlock_value = power.ci.input.currentData()
new_interlock_grps = " ".join([grp.text().replace(" ", "") for grp in power.ci.groups]).rstrip()
if new_interlock_value != self.device.p.get("Interlock", "OFF"):
backlog.append("interlock {}".format(new_interlock_value))
if new_interlock_grps != self.device.p.get("Groups", ""):
backlog.append("interlock {}".format(new_interlock_grps))
for i, pt in enumerate(power.cpt.inputs):
ptime = "PulseTime{}".format(i+1)
current_ptime = self.device.p.get(ptime)
if current_ptime:
current_value = list(current_ptime.keys())[0]
new_value = str(pt.value())
if new_value != current_value:
backlog.append("{} {}".format(ptime, new_value))
if backlog:
backlog.append("status")
backlog.append("status 3")
self.mqtt.publish(self.device.cmnd_topic("backlog"), "; ".join(backlog))
def get_dump(self):
self.backup += self.dl.readAll()
def save_dump(self):
fname = self.dl.header(QNetworkRequest.ContentDispositionHeader)
if fname:
fname = fname.split('=')[1]
save_file = QFileDialog.getSaveFileName(self, "Save config backup", "{}/TDM/{}".format(QDir.homePath(), fname))[0]
if save_file:
with open(save_file, "wb") as f:
f.write(self.backup)
def check_fulltopic(self, fulltopic):
fulltopic += "/" if not fulltopic.endswith('/') else ''
return "%prefix%" in fulltopic and "%topic%" in fulltopic
def closeEvent(self, event):
event.ignore()