from typing import Optional
from .window import FrontWindow, RunWindow, ConfigWindow, ClickWidget
from .core import WsMessageSchemes, WebSocketClient
class PdScreen(ClickWidget):
def __init__(self):
super().__init__()
# 初始化多个窗口对象
self.front_window: Optional[FrontWindow] = None
self.run_window: Optional[RunWindow] = None
self.config_window: Optional[ConfigWindow] = None
# 初始化websocket客户端
self.ws_client = WebSocketClient()
# 初始化窗口
self.initWindow()
self.bind()
self.showFullScreen()
def initWindow(self):
self.front_window = FrontWindow(self)
def bind(self):
self.quick_clicked_sgl.connect(self.toConfigWindowSolt)
self.ws_client.ws_msg_sgl.connect(self.onWebSocketMessageSolt) # type: ignore
self.ws_client.startSocket()
def toConfigWindowSolt(self):
if self.config_window is None:
self.config_window = ConfigWindow(self)
if self.config_window.isVisible():
return
self.config_window.showFullScreen()
def frontToRunWindowSolt(self):
if self.run_window is None:
self.run_window = RunWindow(self)
if self.front_window and self.front_window.isVisible():
self.front_window.hide()
self.run_window.showFullScreen()
def runToFrontWindowSolt(self):
if self.front_window is None:
self.front_window = FrontWindow(self)
if self.run_window and self.run_window.isVisible():
self.run_window.hide()
self.front_window.showFullScreen()
def onWebSocketMessageSolt(self, message: WsMessageSchemes):
if message.capture == 0:
# rank
if self.run_window and self.run_window.isVisible():
self.runToFrontWindowSolt()
self.front_window.updateRank(message)
elif message.capture == 1:
if self.front_window and self.front_window.isVisible():
self.frontToRunWindowSolt()
self.run_window.updateData(message)