forked from M-Labs/thermostat
Make Ui_MainWindow a superclass of our main window
Gets rid of the global ui.
This commit is contained in:
parent
299ef7dcc3
commit
c6ca2b3490
208
pytec/tec_qt.py
208
pytec/tec_qt.py
|
|
@ -1,4 +1,4 @@
|
|||
from PyQt6 import QtWidgets, uic
|
||||
from PyQt6 import QtWidgets
|
||||
from PyQt6.QtCore import pyqtSignal, QObject, QSignalBlocker, pyqtSlot
|
||||
from pyqtgraph import PlotWidget
|
||||
from pyqtgraph.parametertree import Parameter, ParameterTree, ParameterItem, registerParameterType
|
||||
|
|
@ -16,11 +16,7 @@ from ui_tec_qt import Ui_MainWindow
|
|||
|
||||
tec_client: Client = None
|
||||
|
||||
# ui = None
|
||||
ui: Ui_MainWindow = None
|
||||
|
||||
client_watcher = None
|
||||
client_watcher_task = None
|
||||
|
||||
|
||||
def get_argparser():
|
||||
|
|
@ -62,105 +58,118 @@ class ClientWatcher(QObject):
|
|||
self.running = False
|
||||
|
||||
@pyqtSlot()
|
||||
def set_update_s(self):
|
||||
self.update_s = ui.report_refresh_spin.value()
|
||||
def set_update_s(self, update_s):
|
||||
self.update_s = update_s
|
||||
|
||||
|
||||
def on_connection_changed(result):
|
||||
global client_watcher, client_watcher_task
|
||||
ui.graph_group.setEnabled(result)
|
||||
ui.hw_rev_lbl.setEnabled(result)
|
||||
ui.fan_group.setEnabled(result)
|
||||
ui.report_group.setEnabled(result)
|
||||
class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
|
||||
def __init__(self, args):
|
||||
super().__init__()
|
||||
|
||||
ui.ip_set_line.setEnabled(not result)
|
||||
ui.port_set_spin.setEnabled(not result)
|
||||
ui.status_lbl.setText("Connected" if result else "Disconnected")
|
||||
ui.connect_btn.setText("Disconnect" if result else "Connect")
|
||||
if not result:
|
||||
ui.hw_rev_lbl.setText("Thermostat vX.Y")
|
||||
ui.fan_group.setStyleSheet("")
|
||||
if client_watcher:
|
||||
client_watcher.stop_watching()
|
||||
client_watcher = None
|
||||
client_watcher_task = None
|
||||
self.setupUi(self)
|
||||
|
||||
self.connect_btn.clicked.connect(self.connect)
|
||||
self.fan_power_slider.valueChanged.connect(self.fan_set)
|
||||
self.fan_auto_box.stateChanged.connect(self.fan_auto_set)
|
||||
|
||||
def hw_rev(hw_rev_d: dict):
|
||||
logging.debug(hw_rev_d)
|
||||
ui.hw_rev_lbl.setText(f"Thermostat v{hw_rev_d['rev']['major']}.{hw_rev_d['rev']['major']}")
|
||||
ui.fan_group.setEnabled(hw_rev_d["settings"]["fan_available"])
|
||||
if hw_rev_d["settings"]["fan_pwm_recommended"]:
|
||||
ui.fan_group.setStyleSheet("")
|
||||
ui.fan_group.setToolTip("")
|
||||
else:
|
||||
ui.fan_group.setStyleSheet("background-color: yellow")
|
||||
ui.fan_group.setToolTip("Changing the fan settings of not recommended")
|
||||
self.client_watcher_task = None
|
||||
|
||||
if args.connect:
|
||||
if args.IP:
|
||||
self.ip_set_line.setText(args.IP)
|
||||
if args.PORT:
|
||||
self.port_set_spin.setValue(int(args.PORT))
|
||||
self.connect_btn.click()
|
||||
|
||||
def fan_update(fan_settings):
|
||||
logging.debug(fan_settings)
|
||||
if fan_settings is None:
|
||||
return
|
||||
with QSignalBlocker(ui.fan_power_slider) as _:
|
||||
ui.fan_power_slider.setValue(fan_settings["fan_pwm"])
|
||||
ui.fan_power_slider.setEnabled(not fan_settings["auto_mode"])
|
||||
with QSignalBlocker(ui.fan_auto_box) as _:
|
||||
ui.fan_auto_box.setChecked(fan_settings["auto_mode"])
|
||||
def _on_connection_changed(self, result):
|
||||
global client_watcher
|
||||
self.graph_group.setEnabled(result)
|
||||
self.hw_rev_lbl.setEnabled(result)
|
||||
self.fan_group.setEnabled(result)
|
||||
self.report_group.setEnabled(result)
|
||||
|
||||
self.ip_set_line.setEnabled(not result)
|
||||
self.port_set_spin.setEnabled(not result)
|
||||
self.status_lbl.setText("Connected" if result else "Disconnected")
|
||||
self.connect_btn.setText("Disconnect" if result else "Connect")
|
||||
if not result:
|
||||
self.hw_rev_lbl.setText("Thermostat vX.Y")
|
||||
self.fan_group.setStyleSheet("")
|
||||
if client_watcher:
|
||||
client_watcher.stop_watching()
|
||||
client_watcher = None
|
||||
self.client_watcher_task = None
|
||||
|
||||
@asyncSlot()
|
||||
async def fan_set(_):
|
||||
global tec_client
|
||||
if tec_client is None or ui.fan_auto_box.isChecked():
|
||||
return
|
||||
await tec_client.set_param("fan", ui.fan_power_slider.value())
|
||||
|
||||
|
||||
@asyncSlot()
|
||||
async def fan_auto_set(enabled):
|
||||
global tec_client
|
||||
if tec_client is None:
|
||||
return
|
||||
ui.fan_power_slider.setEnabled(not enabled)
|
||||
if enabled:
|
||||
await tec_client.set_param("fan", "auto")
|
||||
else:
|
||||
await tec_client.set_param("fan", ui.fan_power_slider.value())
|
||||
|
||||
|
||||
@asyncSlot()
|
||||
async def connect(_):
|
||||
global tec_client, client_watcher, client_watcher_task
|
||||
ip, port = ui.ip_set_line.text(), ui.port_set_spin.value()
|
||||
try:
|
||||
if tec_client:
|
||||
await tec_client.disconnect()
|
||||
tec_client = None
|
||||
on_connection_changed(False)
|
||||
def _hw_rev(self, hw_rev_d: dict):
|
||||
logging.debug(hw_rev_d)
|
||||
self.hw_rev_lbl.setText(f"Thermostat v{hw_rev_d['rev']['major']}.{hw_rev_d['rev']['major']}")
|
||||
self.fan_group.setEnabled(hw_rev_d["settings"]["fan_available"])
|
||||
if hw_rev_d["settings"]["fan_pwm_recommended"]:
|
||||
self.fan_group.setStyleSheet("")
|
||||
self.fan_group.setToolTip("")
|
||||
else:
|
||||
ui.status_lbl.setText("Connecting...")
|
||||
tec_client = Client()
|
||||
await tec_client.connect(host=ip, port=port, timeout=30)
|
||||
on_connection_changed(True)
|
||||
hw_rev(await tec_client.hw_rev())
|
||||
# fan_update(await tec_client.fan())
|
||||
if client_watcher is None:
|
||||
client_watcher = ClientWatcher(ui.main_widget, ui.report_refresh_spin.value())
|
||||
client_watcher.fan_update.connect(fan_update)
|
||||
ui.report_apply_btn.clicked.connect(
|
||||
lambda: client_watcher.set_update_s(ui.report_refresh_spin.value())
|
||||
)
|
||||
QtWidgets.QApplication.instance().aboutToQuit.connect(client_watcher.stop_watching)
|
||||
client_watcher_task = asyncio.create_task(client_watcher.run())
|
||||
except Exception as e:
|
||||
logging.error(f"Failed communicating to the {ip}:{port}: {e}")
|
||||
on_connection_changed(False)
|
||||
self.fan_group.setStyleSheet("background-color: yellow")
|
||||
self.fan_group.setToolTip("Changing the fan settings of not recommended")
|
||||
|
||||
def fan_update(self, fan_settings):
|
||||
logging.debug(fan_settings)
|
||||
if fan_settings is None:
|
||||
return
|
||||
with QSignalBlocker(self.fan_power_slider) as _:
|
||||
self.fan_power_slider.setValue(fan_settings["fan_pwm"])
|
||||
self.fan_power_slider.setEnabled(not fan_settings["auto_mode"])
|
||||
with QSignalBlocker(self.fan_auto_box) as _:
|
||||
self.fan_auto_box.setChecked(fan_settings["auto_mode"])
|
||||
|
||||
@asyncSlot()
|
||||
async def fan_set(self):
|
||||
global tec_client
|
||||
if tec_client is None or self.fan_auto_box.isChecked():
|
||||
return
|
||||
await tec_client.set_param("fan", self.fan_power_slider.value())
|
||||
|
||||
@asyncSlot(int)
|
||||
async def fan_auto_set(self, enabled):
|
||||
global tec_client
|
||||
if tec_client is None:
|
||||
return
|
||||
self.fan_power_slider.setEnabled(not enabled)
|
||||
if enabled:
|
||||
await tec_client.set_param("fan", "auto")
|
||||
else:
|
||||
await tec_client.set_param("fan", self.fan_power_slider.value())
|
||||
|
||||
@asyncSlot()
|
||||
async def connect(self):
|
||||
global tec_client, client_watcher
|
||||
|
||||
ip, port = self.ip_set_line.text(), self.port_set_spin.value()
|
||||
try:
|
||||
if tec_client:
|
||||
await tec_client.disconnect()
|
||||
tec_client = None
|
||||
self._on_connection_changed(False)
|
||||
else:
|
||||
self.status_lbl.setText("Connecting...")
|
||||
tec_client = Client()
|
||||
await tec_client.connect(host=ip, port=port, timeout=30)
|
||||
self._on_connection_changed(True)
|
||||
self._hw_rev(await tec_client.hw_rev())
|
||||
# self.fan_update(await tec_client.fan())
|
||||
if client_watcher is None:
|
||||
client_watcher = ClientWatcher(self.main_widget, self.report_refresh_spin.value())
|
||||
client_watcher.fan_update.connect(self.fan_update)
|
||||
self.report_apply_btn.clicked.connect(
|
||||
lambda: client_watcher.set_update_s(self.report_refresh_spin.value())
|
||||
)
|
||||
QtWidgets.QApplication.instance().aboutToQuit.connect(client_watcher.stop_watching)
|
||||
self.client_watcher_task = asyncio.create_task(client_watcher.run())
|
||||
except Exception as e:
|
||||
logging.error(f"Failed communicating to the {ip}:{port}: {e}")
|
||||
self._on_connection_changed(False)
|
||||
|
||||
|
||||
async def coro_main():
|
||||
global ui
|
||||
|
||||
args = get_argparser().parse_args()
|
||||
if args.logLevel:
|
||||
logging.basicConfig(level=getattr(logging, args.logLevel))
|
||||
|
|
@ -170,22 +179,7 @@ async def coro_main():
|
|||
app = QtWidgets.QApplication.instance()
|
||||
app.aboutToQuit.connect(app_quit_event.set)
|
||||
|
||||
main_window = QtWidgets.QMainWindow()
|
||||
ui = Ui_MainWindow()
|
||||
ui.setupUi(main_window)
|
||||
# ui = uic.loadUi('tec_qt.ui', main_window)
|
||||
|
||||
ui.connect_btn.clicked.connect(connect)
|
||||
ui.fan_power_slider.valueChanged.connect(fan_set)
|
||||
ui.fan_auto_box.stateChanged.connect(fan_auto_set)
|
||||
|
||||
if args.connect:
|
||||
if args.IP:
|
||||
ui.ip_set_line.setText(args.IP)
|
||||
if args.PORT:
|
||||
ui.port_set_spin.setValue(int(args.PORT))
|
||||
ui.connect_btn.click()
|
||||
|
||||
main_window = MainWindow(args)
|
||||
main_window.show()
|
||||
|
||||
await app_quit_event.wait()
|
||||
|
|
|
|||
Loading…
Reference in New Issue