class DesktopApp(App, QRCodeButtonObserver, CommandHandlerObserver,
ConnectionObserver):
def __init__(self, **kwargs):
super().__init__(**kwargs)
current_dir = os.scandir('./')
for entry in current_dir:
if entry.name == 'qrcode' and entry.is_dir():
break
else:
os.mkdir('./qrcode', 0o777)
try:
self.server = Server(self)
except OSError as err:
print("Failed to create Server instance,", err.strerror)
exit(err.errno)
# self.qrcode_generated = False
self.command_handler = CommandHandler()
self.command_handler.add_observer(self)
self.server.add_observer(self)
self.server_info = self.server.info
self.server.run()
def build(self):
self.screen_manager = ScreenManager()
self.main_screen = MainScreen('main', self.server_info)
self.main_screen.qrcode_button.add_observer(self)
self.qrcode_screen = QRCodeScreen('qrcode')
self.qrcode_screen.add_back_main_button_observer(self)
self.screen_manager.add_widget(self.main_screen)
self.screen_manager.add_widget(self.qrcode_screen)
return self.screen_manager
def handle_message(self, data: bytes):
try:
command, output = self.command_handler(data.decode('utf-8'))
self.main_screen.on_command_output(command, output)
except ValueError as err:
print(f'ValueError for input {data.decode("utf-8")}')
except TypeError as err:
print(f'TypeError for input {data.decode("utf-8")}')
def show_qrcode(self, *args):
server_info = self.server.info
qrcode_img = qrcode.make(f"{server_info.ip},{server_info.port}")
qrcode_img.save('qrcode/qrcode.png', format='png')
self.qrcode_screen.qrcode = Image(source='qrcode/qrcode.png')
self.screen_manager.current = 'qrcode'
def on_qr_button_pressed(self):
self.show_qrcode()
def on_back_main_pressed(self):
self.screen_manager.current = 'main'
def on_disconnect(self):
self.server.closeConnection()
def on_connection_lost(self, server_info: ServerInfo) -> None:
self.main_screen.on_connection_lost(server_info)
def on_connection_made(self, server_info: ServerInfo) -> None:
self.screen_manager.current = 'main'
self.main_screen.on_connection_made(server_info)
