def __init__(self, *args, **kwargs): super(QSH, self).__init__(*args, **kwargs) self.setQuitOnLastWindowClosed(False) # signals self.aboutToQuit.connect(self.beforeQuit) # dialogs self.screenViewDialog = ScreenViewDialog(self) # networking self.connector = Connector() # tray self.trayIcon = MainTrayIcon(self, callbacks= { "quit": self.quit, "configuration": self.showConfigurationDialog, "incoming": self.showScreenViewDialog }) self.trayIcon.middle_click_callback = self.trayIconMiddleClick # networking callbacks self.connector.known_hosts_updated_callback = self.trayIcon.updateMenu self.connector.got_image_callback = self.processReceivedImage self.connector.receiving_start_callback = self.trayIcon.setIconLoading self.connector.sending_end_callback = self.trayIcon.setIconDefault # hi there! self.connector.updateKnownHosts() # periodically check whether hosts alive self.helloAllTimer = QtCore.QTimer(self) self.connect(self.helloAllTimer, QtCore.SIGNAL("timeout()"), self.connector.updateKnownHosts) self.helloAllTimer.start(AppConfig.get_heartbeat_interval())
def __init__(self, *args, **kwargs): super(QSH, self).__init__(*args, **kwargs) self.setQuitOnLastWindowClosed(False) # signals self.aboutToQuit.connect(self.beforeQuit) # dialogs self.screenViewDialog = ScreenViewDialog(self) # networking self.connector = Connector() # tray self.trayIcon = MainTrayIcon(self, callbacks={ "quit": self.quit, "configuration": self.showConfigurationDialog, "incoming": self.showScreenViewDialog }) self.trayIcon.middle_click_callback = self.trayIconMiddleClick # networking callbacks self.connector.known_hosts_updated_callback = self.trayIcon.updateMenu self.connector.got_image_callback = self.processReceivedImage self.connector.receiving_start_callback = self.trayIcon.setIconLoading self.connector.sending_end_callback = self.trayIcon.setIconDefault # hi there! self.connector.updateKnownHosts() # periodically check whether hosts alive self.helloAllTimer = QtCore.QTimer(self) self.connect(self.helloAllTimer, QtCore.SIGNAL("timeout()"), self.connector.updateKnownHosts) self.helloAllTimer.start(AppConfig.get_heartbeat_interval())
class QSH(QApplication): def __init__(self, *args, **kwargs): super(QSH, self).__init__(*args, **kwargs) self.setQuitOnLastWindowClosed(False) # signals self.aboutToQuit.connect(self.beforeQuit) # dialogs self.screenViewDialog = ScreenViewDialog(self) # networking self.connector = Connector() # tray self.trayIcon = MainTrayIcon(self, callbacks= { "quit": self.quit, "configuration": self.showConfigurationDialog, "incoming": self.showScreenViewDialog }) self.trayIcon.middle_click_callback = self.trayIconMiddleClick # networking callbacks self.connector.known_hosts_updated_callback = self.trayIcon.updateMenu self.connector.got_image_callback = self.processReceivedImage self.connector.receiving_start_callback = self.trayIcon.setIconLoading self.connector.sending_end_callback = self.trayIcon.setIconDefault # hi there! self.connector.updateKnownHosts() # periodically check whether hosts alive self.helloAllTimer = QtCore.QTimer(self) self.connect(self.helloAllTimer, QtCore.SIGNAL("timeout()"), self.connector.updateKnownHosts) self.helloAllTimer.start(AppConfig.get_heartbeat_interval()) def trayIconMiddleClick(self): self.trayIcon.icon.setContextMenu(None) if not self.screenViewDialog.isVisible(): if self.trayIcon.incomingTotal: self.trayIcon.actionShowScreenViewDialog.trigger() else: self.screenViewDialog.close() def processReceivedImage(self, data_uuid=None, data=None): """ Show received screenshot """ if data_uuid and data.size(): receivedImagesCount = self.screenViewDialog.processReceivedImage(data_uuid=data_uuid, data=data, known_hosts=self.connector.known_hosts) self.trayIcon.incomingTotal += receivedImagesCount if self.screenViewDialog.isVisible(): self.screenViewDialog.showWindow() else: self.trayIcon.incomingUnread += receivedImagesCount self.trayIcon.updateMenu() self.trayIcon.setIconDefault() def shareScreen(self, host_data): """ Send screenshot """ self.trayIcon.setIconLoading() # capture screenshot screenBA = QtCore.QByteArray() screenBuf = QtCore.QBuffer(screenBA) screenBuf.open(QtCore.QBuffer.WriteOnly) QPixmap.grabWindow(self.desktop().winId()).save(screenBuf, SCREEN_IMAGE_TYPE, SCREEN_IMAGE_QUALITY) self.connector.submitScreen(host_data["host"], host_data["port"], screenBA) def showConfigurationDialog(self): self.config_dialog = ConfigurationDialog(self) self.config_dialog.showNormal() self.config_dialog.activateWindow() self.config_dialog.raise_() def showScreenViewDialog(self): self.trayIcon.incomingUnread = 0 self.trayIcon.updateMenu() self.trayIcon.setIconDefault() self.screenViewDialog.showWindow() def beforeQuit(self): self.connector.byeAll()
class QSH(QApplication): def __init__(self, *args, **kwargs): super(QSH, self).__init__(*args, **kwargs) self.setQuitOnLastWindowClosed(False) # signals self.aboutToQuit.connect(self.beforeQuit) # dialogs self.screenViewDialog = ScreenViewDialog(self) # networking self.connector = Connector() # tray self.trayIcon = MainTrayIcon(self, callbacks={ "quit": self.quit, "configuration": self.showConfigurationDialog, "incoming": self.showScreenViewDialog }) self.trayIcon.middle_click_callback = self.trayIconMiddleClick # networking callbacks self.connector.known_hosts_updated_callback = self.trayIcon.updateMenu self.connector.got_image_callback = self.processReceivedImage self.connector.receiving_start_callback = self.trayIcon.setIconLoading self.connector.sending_end_callback = self.trayIcon.setIconDefault # hi there! self.connector.updateKnownHosts() # periodically check whether hosts alive self.helloAllTimer = QtCore.QTimer(self) self.connect(self.helloAllTimer, QtCore.SIGNAL("timeout()"), self.connector.updateKnownHosts) self.helloAllTimer.start(AppConfig.get_heartbeat_interval()) def trayIconMiddleClick(self): self.trayIcon.icon.setContextMenu(None) if not self.screenViewDialog.isVisible(): if self.trayIcon.incomingTotal: self.trayIcon.actionShowScreenViewDialog.trigger() else: self.screenViewDialog.close() def processReceivedImage(self, data_uuid=None, data=None): """ Show received screenshot """ if data_uuid and data.size(): receivedImagesCount = self.screenViewDialog.processReceivedImage( data_uuid=data_uuid, data=data, known_hosts=self.connector.known_hosts) self.trayIcon.incomingTotal += receivedImagesCount if self.screenViewDialog.isVisible(): self.screenViewDialog.showWindow() else: self.trayIcon.incomingUnread += receivedImagesCount self.trayIcon.updateMenu() self.trayIcon.setIconDefault() def shareScreen(self, host_data): """ Send screenshot """ self.trayIcon.setIconLoading() # capture screenshot screenBA = QtCore.QByteArray() screenBuf = QtCore.QBuffer(screenBA) screenBuf.open(QtCore.QBuffer.WriteOnly) QPixmap.grabWindow(self.desktop().winId()).save( screenBuf, SCREEN_IMAGE_TYPE, SCREEN_IMAGE_QUALITY) self.connector.submitScreen(host_data["host"], host_data["port"], screenBA) def showConfigurationDialog(self): self.config_dialog = ConfigurationDialog(self) self.config_dialog.showNormal() self.config_dialog.activateWindow() self.config_dialog.raise_() def showScreenViewDialog(self): self.trayIcon.incomingUnread = 0 self.trayIcon.updateMenu() self.trayIcon.setIconDefault() self.screenViewDialog.showWindow() def beforeQuit(self): self.connector.byeAll()