def open_setup(self, p, l): self.setup_form = QtWidgets.QWidget() self.ui_setup = Ui_setup_form() self.ui_setup.setupUi(self.setup_form) self.setup_form.show() self.ui_setup.pushButton_save.clicked.connect(lambda: self.save_setup()) vm_list = ['Direct mouse control'] try: vm = VirtualBoxController() vm_list += vm.get_vbox_list() except: pass # no virtual machine self.ui_setup.comboBox_vm.addItems(vm_list) timeouts = ['8', '9', '10', '11', '12'] self.ui_setup.comboBox_2.addItems(timeouts) config = ConfigObj("config.ini") try: mouse_control = config['control'] except: mouse_control = 'Direct mouse control' for i in [i for i, x in enumerate(vm_list) if x == mouse_control]: idx = i self.ui_setup.comboBox_vm.setCurrentIndex(idx) try: timeout = config['montecarlo_timeout'] except: timeout = 10 for i in [i for i, x in enumerate(timeouts) if x == timeout]: idx = i self.ui_setup.comboBox_2.setCurrentIndex(idx)
def take_screenshot(self, initial, p): if initial: self.gui_signals.signal_status.emit("") self.gui_signals.signal_progressbar_reset.emit() if self.gui_signals.exit_thread == True: sys.exit() if self.gui_signals.pause_thread == True: while self.gui_signals.pause_thread == True: time.sleep(.2) if self.gui_signals.exit_thread == True: sys.exit() time.sleep(0.1) config = ConfigObj("config.ini") control = config['control'] if control == 'Direct mouse control': self.take_screenshot2() self.entireScreenPIL = self.screenshot else: try: vb = VirtualBoxController() self.entireScreenPIL = vb.get_screenshot_vbox() self.logger.debug("Screenshot taken from virtual machine") except: self.logger.warning( "No virtual machine found. Press SETUP to re initialize the VM controller" ) # gui_signals.signal_open_setup.emit(p,L) self.take_screenshot2() self.entireScreenPIL = self.screenshot self.gui_signals.signal_status.emit(str(p.current_strategy)) self.gui_signals.signal_progressbar_increase.emit(5) return True
def take_screenshot(virtual_box=False): """ Take screenshot directly from screen or via virtualbox Args: virtual_box: bool Returns: PIL screenshot """ if not virtual_box: log.debug("Calling screen grabber") screenshot = ImageGrab.grab() log.debug("Direct screenshot successful") else: # virtual_box try: vb = VirtualBoxController() screenshot = vb.get_screenshot_vbox() log.debug("Screenshot taken from virtual machine") except: log.warning("No virtual machine found. Press SETUP to re initialize the VM controller") # gui_signals.signal_open_setup.emit(p,L) screenshot = ImageGrab.grab() return screenshot
def take_screenshot(self): """Take a screenshot""" log.info("Clearing window") self.signal_update_screenshot_pic.emit(Image.new('RGB', (3, 3))) log.info("Taking screenshot") config = ConfigObj(CONFIG_FILENAME) control = config['control'] if control == 'Direct mouse control': self.original_screenshot = take_screenshot() else: try: vb = VirtualBoxController() self.original_screenshot = vb.get_screenshot_vbox() log.debug("Screenshot taken from virtual machine") except: log.warning( "No virtual machine found. Press SETUP to re initialize the VM controller" ) self.original_screenshot = take_screenshot() log.info("Emitting update signal") self.signal_update_screenshot_pic.emit(self.original_screenshot) log.info("signal emission complete")
def take_screenshot(self, initial, p): if initial: self.gui_signals.signal_status.emit("") self.gui_signals.signal_progressbar_reset.emit() if self.gui_signals.exit_thread == True: sys.exit() if self.gui_signals.pause_thread == True: while self.gui_signals.pause_thread == True: time.sleep(1) if self.gui_signals.exit_thread == True: sys.exit() time.sleep(0.1) config = ConfigObj("config.ini") control = config['control'] if control == 'Direct mouse control': self.entireScreenPIL = ImageGrab.grab() else: try: vb = VirtualBoxController() self.entireScreenPIL = vb.get_screenshot_vbox() self.logger.debug("Screenshot taken from virtual machine") except: self.logger.warning("No virtual machine found. Press SETUP to re initialize the VM controller") # gui_signals.signal_open_setup.emit(p,L) self.entireScreenPIL = ImageGrab.grab() self.gui_signals.signal_status.emit(str(p.current_strategy)) self.gui_signals.signal_progressbar_increase.emit(5) return True
def open_setup(self, p, l): self.setup_form = QtWidgets.QWidget() self.ui_setup = Ui_setup_form() self.ui_setup.setupUi(self.setup_form) self.setup_form.show() self.ui_setup.pushButton_save.clicked.connect(lambda: self.save_setup()) vm_list = ['Direct mouse control'] try: vm = VirtualBoxController() vm_list += vm.get_vbox_list() except: pass # no virtual machine self.ui_setup.comboBox_vm.addItems(vm_list) timeouts = ['8','9','10', '11','12'] self.ui_setup.comboBox_2.addItems(timeouts) config = ConfigObj("config.ini") try: mouse_control = config['control'] except: mouse_control = 'Direct mouse control' for i in [i for i, x in enumerate(vm_list) if x == mouse_control]: idx = i self.ui_setup.comboBox_vm.setCurrentIndex(idx) try: timeout = config['montecarlo_timeout'] except: timeout = 10 for i in [i for i, x in enumerate(timeouts) if x == timeout]: idx = i self.ui_setup.comboBox_2.setCurrentIndex(idx)
def open_setup(self): self.ui_setup = SetupForm() self.ui_setup.pushButton_save.clicked.connect( lambda: self.save_setup()) vm_list = ['Direct mouse control'] try: vm = VirtualBoxController() vm_list += vm.get_vbox_list() except: pass # no virtual machine self.ui_setup.comboBox_vm.addItems(vm_list) timeouts = ['8', '9', '10', '11', '12'] self.ui_setup.comboBox_2.addItems(timeouts) config = get_config() try: mouse_control = config.config.get('main', 'control') except: mouse_control = 'Direct mouse control' for i in [i for i, x in enumerate(vm_list) if x == mouse_control]: idx = i self.ui_setup.comboBox_vm.setCurrentIndex(idx) try: timeout = config.config.get('main', 'montecarlo_timeout') except: timeout = 10 for i in [i for i, x in enumerate(timeouts) if x == timeout]: idx = i self.ui_setup.comboBox_2.setCurrentIndex(idx) login = config.config.get('main', 'login') password = config.config.get('main', 'password') db = config.config.get('main', 'db') self.ui_setup.login.setText(login) self.ui_setup.password.setText(password)