def __init__(self, device, picPath): super(UpdateScreen, self).__init__() self.isStop = False self._device = device self.delay = 0 self.picPath = picPath self.lastFrameTime = 0 self.isRefresh = False self.nconcurrent = threading.BoundedSemaphore(5) self.miniServer = MiniServer(self._device) self.miniReader = MiniReader(self._device)
class RunTest(QtCore.QThread): """ Handle the reproduced actions and send back to device """ actionResult = State.FAIL isStopRun = False limit = 0 picData = None picImage = QtGui.QImage() finish = QtCore.Signal() currentAction = QtCore.Signal(AutoSenseItem) actionDone = QtCore.Signal(AutoSenseItem, int, int) def __init__(self, times=1, device=None): super(RunTest, self).__init__() self.setTimes(times) self.setDevice(device) self.setMinicap() self.actionSwitcher = { 'Power': self.actionPower, 'Unlock': self.actionUnlock, 'Back': self.actionBack, 'Click': self.actionClick, 'Home': self.actionHome, 'Swipe': self.actionSwipe, 'Drag': self.actionDrag, 'LongClick': self.actionLongClick, 'Delay': self.actionDelay, 'Menu': self.actionMenu, 'Exist': self.actionCheckExist, 'NoExist': self.actionCheckNoExist, 'IsBlank': self.actionCheckBlank, 'RelativeCheck': self.actionRelativeCheck, 'Type': self.actionType, 'MediaCheck': self.actionMediaCheck, 'HideKeyboard': self.actionHideKeyboard, 'CheckPoint': self.actionCheckPoint } def setMinicap(self): self.miniReader = MiniReader(self._device) self.miniServer = MiniServer(self._device) def setTimes(self, times): """ Set repeat times of running a script. It should set before running. :param times: Set by user. """ self.times = times def setActions(self, actions): """ Assign actions that want to play. :param actions: List of AutoSense item. """ self.actions = actions def setDevice(self, device): """ Assign a instance of device. :param device: AdbDevice """ self._device = device def setPlayName(self, name): self.playName = name def checkMedia(self, passTime, timeout, isHold): decryptRE = re.compile( '[lL]ast\s+write\s+occurred\s+\(msecs\):\s+(?P<last_write>\d+)') current_sec_time = lambda: time.time() isAlreadyPass = False idle = current_sec_time() work = idle hold = True if isHold == 'True' else False while not self.isStopRun: output = self._device.cmd.dumpsys(['media.audio_flinger']) decryptMatch = decryptRE.search(output) if decryptMatch is not None: lastWriteTime = int(decryptMatch.group('last_write')) if lastWriteTime < 1000: idle = current_sec_time() playTime = current_sec_time() - work print 'play time = ' + str( playTime) + ' passTime: ' + str(passTime) if playTime > passTime: if hold: isAlreadyPass = True timeout = 2 else: return {'result': True, 'reason': ''} else: work = current_sec_time() idleTime = current_sec_time() - idle print 'idleTime = ' + str( idleTime) + ', idleTime > timeout: ' + str( idleTime > timeout) if idleTime > timeout: if isAlreadyPass: return {'result': True, 'reason': ''} else: return {'result': False, 'reason': 'timeout'} else: print 'It can\'t figure out last write time.' return { 'result': False, 'reason': 'Program can\'t define whether media is playing.' } time.sleep(0.1) def actionMediaCheck(self, param, refer=None): t, timeout, isHold = param result = self.checkMedia(int(t), int(timeout), isHold) if result is not None: if result['result']: self.actionResult = State.PASS def actionRelativeCheck(self, param, refer=None): pass def actionCheckPoint(self, param, refer=None): self.actionResult = State.SEMI def actionType(self, param, refer=None): """ Action of typing text. :param param: a string :param refer: None """ self._device.type(param[0]) self.actionResult = State.PASS def actionDelay(self, param, refer=None): """ :param param: (int). in second. :param refer: None """ t = int(param[0]) D_value = 0 container = [] startTime = float( self._device.cmd.shell(['echo', '$EPOCHREALTIME'], output=True)) while not self.isStopRun and D_value < t: currentTime = float( self._device.cmd.shell(['echo', '$EPOCHREALTIME'], output=True)) D_value = currentTime - startTime self.removeLastLog.emit() container.append(D_value) if len(container) > 1: if not self._device.isScreenOn(): self._device.powerBtn() if self._device.isLocked(): self._device.unlock() del container[:] time.sleep(1) self.actionResult = State.PASS def actionUnlock(self, param, refer=None): """ Unlock screen. If screen is not locked, it will do nothing. :param param: None :param refer: None """ self._device.unlock() self.actionResult = State.PASS def actionMenu(self, param, refer=None): """ Simulate menu button is pressed. :param param: None :param refer: None """ self._device.cmd.inputKeyevnt('MENU') self.actionResult = State.PASS def actionPower(self, param, refer=None): """ Simulate power button is pressed. :param param: None :param refer: None """ self._device.cmd.inputKeyevnt('POWER') self.actionResult = State.PASS def actionHome(self, param, refer=None): """ Simulate home button is pressed. :param param: None :param refer: None """ self._device.cmd.inputKeyevnt('HOME') self.actionResult = State.PASS def actionLongClick(self, param, refer=None): """ Simulate long click. :param param: [x, y, duration] :param refer: A json object include view information. """ info = json.loads(refer) x, y, duration = param point = (int(x), int(y)) self._device.longClick(point[0], point[1], int(duration)) self.actionResult = State.PASS def actionDrag(self, param, refer=None): """ Simulate drag action. It can drag a view object if it can be dragged. :param param: [x1, y1, x2, y2, duration] :param refer: None """ start = (param[0], param[1]) end = (param[2], param[3]) self._device.drag(start, end, int(param[4])) self.actionResult = State.PASS def actionSwipe(self, param, refer=None): """ Simulate swipe action. View won't be dragged :param param: [x1, y1, x2, y2, duration] :param refer: None """ start = (param[0], param[1]) end = (param[2], param[3]) # ensure page flow self._device.swipe(start, end, int(param[4])) self.actionResult = State.PASS def actionBack(self, param, refer=None): """ Simulate back button. :param param: None :param refer: None """ self._device.backBtn() self.actionResult = State.PASS def actionClick(self, param, refer=None): """ Simulate click action. :param param: [x, y] :param refer: A json object include view information. """ info = json.loads(refer) if len(param) == 2: x, y = param else: x, y, _ = param point = (int(x), int(y)) noShow = False wait = 5 while self._device.isConnected() and not self.isStopRun and wait > 0: try: if not self._device.isScreenOn(): self._device.powerBtn() if self._device.isLocked(): self._device.unlock() result = self._device.checkSamePoint(point, info) print 'reason = ' + result['reason'] if result['answer']: self.actionResult = State.PASS break elif not noShow: print 'wait view' time.sleep(1) wait -= 1 except uiautomator.JsonRPCError as m: print 'JsonRPCError: ' + m.message break def actionCheckBlank(self, param, refer=None): """ Check the target view should not have any child view. :param param: [Identification, x, y]. There are 4 kinds of identification(resourceId, text, description, className). :param refer: None """ benchmark, x, y = param attribute, value = benchmark.split('=') point = (int(x), int(y)) if attribute == 'text': selector = self._device.retrieveSelector( point, self._device.d(text=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS elif attribute == 'description': selector = self._device.retrieveSelector( point, self._device.d(description=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS elif attribute == 'resourceId': selector = self._device.retrieveSelector( point, self._device.d(resourceId=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS elif attribute == 'className': selector = self._device.retrieveSelector( point, self._device.d(className=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS def actionCheckExist(self, param, refer=None): """ Check the target view whether is existed on the screen. :param param: [Identification, x, y]. There are 4 kinds of identification(resourceId, text, description, className). :param refer: None """ benchmark, x, y = param attribute, value = benchmark.split('=') point = (int(x), int(y)) if attribute == 'text': if self._device.retrieveSelector(point, self._device.d(text=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True elif attribute == 'description': if self._device.retrieveSelector( point, self._device.d(description=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True elif attribute == 'resourceId': if self._device.retrieveSelector(point, self._device.d(resourceId=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True elif attribute == 'className': if self._device.retrieveSelector(point, self._device.d(className=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True def actionCheckNoExist(self, param, refer=None): """ Check the target view whether isn't existed on the screen. :param param: [Identification, x, y]. There are 4 kinds of identification(resourceId, text, description, className). :param refer: None """ benchmark, x, y = param attribute, value = benchmark.split('=') point = (int(x), int(y)) if attribute == 'text': if not self._device.retrieveSelector(point, self._device.d(text=value)): self.actionResult = State.PASS elif attribute == 'description': if not self._device.retrieveSelector( point, self._device.d(description=value)): self.actionResult = State.PASS elif attribute == 'resourceId': if not self._device.retrieveSelector( point, self._device.d(resourceId=value)): self.actionResult = State.PASS elif attribute == 'className': if not self._device.retrieveSelector( point, self._device.d(className=value)): self.actionResult = State.PASS def actionHideKeyboard(self, param, refer=None): """ Hide soft keyboard. If keyboard isn't showed, it will do nothing. :param param: None :param refer: None """ self._device.hideKeyboard() self.actionResult = State.PASS def stop(self): """ Exit this process. """ self.isStopRun = True def setStartIndex(self, index): """ chose a start index of actions. :param index: integer """ self.startIndex = index def takeScreenShot(self, path): self.picData = self.miniReader.getDisplay() self.picImage.loadFromData(self.picData) self.picImage.save(path, 'JPEG') def run(self): self.isStopRun = False self.limit = 0 try: self.miniServer.start() ''' wait for active ''' while not self.miniServer.isActive(): time.sleep(0.5) self.takeScreenShot(Global.IMAGE_FOLDER + '/%s_%d.jpg' % (self.playName, 0)) while self.limit < self.times and not self.isStopRun: self.limit += 1 for index in range(len(self.actions)): if not self.isStopRun and self.startIndex <= index: self.index = index + 1 item = self.actions[index] if self._device.isConnected(): self.actionResult = State.FAIL self.currentAction.emit(item) self.actionSwitcher.get(item.action())( item.parameter(), item.information()) self.actionDone.emit(item, self.actionResult, index) self.takeScreenShot(Global.IMAGE_FOLDER + '/%s_%d.jpg' % (self.playName, self.index)) time.sleep(0.5) else: self.isStopRun = True break except socket.error: print 'socket.error: device offline' self.isStopRun = True except ValueError as e: print 'ValueError: ' + e.message self.isStopRun = True except CalledProcessError, exc: print 'CalledProcessError: ' + str(exc) self.isStopRun = True finally:
class UpdateScreen(QtCore.QThread): """ Keep capturing screen in the background. """ loadDone = QtCore.Signal() startLoad = QtCore.Signal() deviceOffline = QtCore.Signal() isStop = False isPause = False def __init__(self, device, picPath): super(UpdateScreen, self).__init__() self.isStop = False self._device = device self.delay = 0 self.picPath = picPath self.lastFrameTime = 0 self.isRefresh = False self.nconcurrent = threading.BoundedSemaphore(5) self.miniServer = MiniServer(self._device) self.miniReader = MiniReader(self._device) def stop(self): """ Stop capturing """ self.isStop = True self.miniServer.stop() # def pause(self): # """ Pause capturing """ # self.isPause = True # self.miniServer.stop() # # def resume(self): # """ resume capturing """ # self.isPause = False # # self.start() def setDelay(self, delay=0): """ Set a delay time to start capturing """ self.delay = delay def run(self): time.sleep(self.delay) try: self.miniServer.start() self.isStop = False while not self.isStop: if self.miniServer.isActive(): if not self.miniServer.needRestart(): self.startLoad.emit() self.screenshot(path=self.picPath + '/' + self._device.serialno + '_screen.jpg') else: self.miniServer.reStart() else: time.sleep(0.5) # self.isPause = True except: print 'device not found' self.lastFrameTime = 0 self.deviceOffline.emit() finally: self.miniServer.stop() def monitorFrame(self): """ monitor the screen has changed, it helps to reduce capturing frequency. """ decryptRE = re.compile('\s+events-delivered:\s+(?P<number>\d+)') output = self._device.cmd.dumpsys(['SurfaceFlinger']) if output: m = decryptRE.search(output) if m is not None: tmpTime = m.group('number') if tmpTime != self.lastFrameTime: self.lastFrameTime = tmpTime return True def screenshot(self, path, delay=0): """ Do screen capture :param path: save pic path :param delay: delay to capture screen. In second. """ try: # self.nconcurrent.acquire() time.sleep(delay) pp = time.time() if self._device.isConnected(): if self.miniServer.isActive(): byteData = self.miniReader.getDisplay() im = QtGui.QImage.fromData( QtCore.QByteArray.fromRawData(byteData)) im.loadFromData(byteData) im.save(path, 'JPEG') else: if not self.isStop: self.deviceOffline.emit() print time.time() - pp if not self.isStop: self.loadDone.emit() except AttributeError as e: print e.message except IOError as e: print e.message # finally: # self.nconcurrent.release() def __byteToHex(self, bs): return ''.join(['%02X ' % ord(byte) for byte in bs]) def __parsePicSize(self, data): return int(binascii.hexlify(data[::-1]), 16) def __parseBanner(self, data): if len(data) == 24: banner = dict() banner['version'] = int(binascii.hexlify(data[0]), 16) banner['length'] = int(binascii.hexlify(data[1]), 16) banner['pid'] = int(binascii.hexlify(data[2:5][::-1]), 16) banner['real.width'] = int(binascii.hexlify(data[6:9][::-1]), 16) banner['real.height'] = int(binascii.hexlify(data[10:13][::-1]), 16) banner['virtual.width'] = int(binascii.hexlify(data[14:17][::-1]), 16) banner['virtual.height'] = int(binascii.hexlify(data[18:21][::-1]), 16) banner['orient'] = int(binascii.hexlify(data[22]), 16) banner['policy'] = int(binascii.hexlify(data[23]), 16) return banner
def setMinicap(self): self.miniReader = MiniReader(self._device) self.miniServer = MiniServer(self._device)
class RunTest(QtCore.QThread): """ Handle the reproduced actions and send back to device """ actionResult = State.FAIL isStopRun = False limit = 0 picData = None picImage = QtGui.QImage() finish = QtCore.Signal() currentAction = QtCore.Signal(AutoSenseItem) actionDone = QtCore.Signal(AutoSenseItem, int, int) def __init__(self, times=1, device=None): super(RunTest, self).__init__() self.setTimes(times) self.setDevice(device) self.setMinicap() self.actionSwitcher = { 'Power': self.actionPower, 'Unlock': self.actionUnlock, 'Back': self.actionBack, 'Click': self.actionClick, 'Home': self.actionHome, 'Swipe': self.actionSwipe, 'Drag': self.actionDrag, 'LongClick': self.actionLongClick, 'Delay': self.actionDelay, 'Menu': self.actionMenu, 'Exist': self.actionCheckExist, 'NoExist': self.actionCheckNoExist, 'IsBlank': self.actionCheckBlank, 'RelativeCheck': self.actionRelativeCheck, 'Type': self.actionType, 'MediaCheck': self.actionMediaCheck, 'HideKeyboard': self.actionHideKeyboard, 'CheckPoint': self.actionCheckPoint} def setMinicap(self): self.miniReader = MiniReader(self._device) self.miniServer = MiniServer(self._device) def setTimes(self, times): """ Set repeat times of running a script. It should set before running. :param times: Set by user. """ self.times = times def setActions(self, actions): """ Assign actions that want to play. :param actions: List of AutoSense item. """ self.actions = actions def setDevice(self, device): """ Assign a instance of device. :param device: AdbDevice """ self._device = device def setPlayName(self, name): self.playName = name def checkMedia(self, passTime, timeout, isHold): decryptRE = re.compile('[lL]ast\s+write\s+occurred\s+\(msecs\):\s+(?P<last_write>\d+)') current_sec_time = lambda: time.time() isAlreadyPass = False idle = current_sec_time() work = idle hold = True if isHold == 'True' else False while not self.isStopRun: output = self._device.cmd.dumpsys(['media.audio_flinger']) decryptMatch = decryptRE.search(output) if decryptMatch is not None: lastWriteTime = int(decryptMatch.group('last_write')) if lastWriteTime < 1000: idle = current_sec_time() playTime = current_sec_time() - work print 'play time = ' + str(playTime) + ' passTime: ' + str(passTime) if playTime > passTime: if hold: isAlreadyPass = True timeout = 2 else: return {'result': True, 'reason': ''} else: work = current_sec_time() idleTime = current_sec_time() - idle print 'idleTime = ' + str(idleTime) + ', idleTime > timeout: ' + str(idleTime > timeout) if idleTime > timeout: if isAlreadyPass: return {'result': True, 'reason': ''} else: return {'result': False, 'reason': 'timeout'} else: print 'It can\'t figure out last write time.' return {'result': False, 'reason': 'Program can\'t define whether media is playing.'} time.sleep(0.1) def actionMediaCheck(self, param, refer=None): t, timeout, isHold = param result = self.checkMedia(int(t), int(timeout), isHold) if result is not None: if result['result']: self.actionResult = State.PASS def actionRelativeCheck(self, param, refer=None): pass def actionCheckPoint(self, param, refer=None): self.actionResult = State.SEMI def actionType(self, param, refer=None): """ Action of typing text. :param param: a string :param refer: None """ self._device.type(param[0]) self.actionResult = State.PASS def actionDelay(self, param, refer=None): """ :param param: (int). in second. :param refer: None """ t = int(param[0]) D_value = 0 container = [] startTime = float(self._device.cmd.shell(['echo', '$EPOCHREALTIME'], output=True)) while not self.isStopRun and D_value < t: currentTime = float(self._device.cmd.shell(['echo', '$EPOCHREALTIME'], output=True)) D_value = currentTime - startTime self.removeLastLog.emit() container.append(D_value) if len(container) > 1: if not self._device.isScreenOn(): self._device.powerBtn() if self._device.isLocked(): self._device.unlock() del container[:] time.sleep(1) self.actionResult = State.PASS def actionUnlock(self, param, refer=None): """ Unlock screen. If screen is not locked, it will do nothing. :param param: None :param refer: None """ self._device.unlock() self.actionResult = State.PASS def actionMenu(self, param, refer=None): """ Simulate menu button is pressed. :param param: None :param refer: None """ self._device.cmd.inputKeyevnt('MENU') self.actionResult = State.PASS def actionPower(self, param, refer=None): """ Simulate power button is pressed. :param param: None :param refer: None """ self._device.cmd.inputKeyevnt('POWER') self.actionResult = State.PASS def actionHome(self, param, refer=None): """ Simulate home button is pressed. :param param: None :param refer: None """ self._device.cmd.inputKeyevnt('HOME') self.actionResult = State.PASS def actionLongClick(self, param, refer=None): """ Simulate long click. :param param: [x, y, duration] :param refer: A json object include view information. """ info = json.loads(refer) x, y, duration = param point = (int(x), int(y)) self._device.longClick(point[0], point[1], int(duration)) self.actionResult = State.PASS def actionDrag(self, param, refer=None): """ Simulate drag action. It can drag a view object if it can be dragged. :param param: [x1, y1, x2, y2, duration] :param refer: None """ start = (param[0], param[1]) end = (param[2], param[3]) self._device.drag(start, end, int(param[4])) self.actionResult = State.PASS def actionSwipe(self, param, refer=None): """ Simulate swipe action. View won't be dragged :param param: [x1, y1, x2, y2, duration] :param refer: None """ start = (param[0], param[1]) end = (param[2], param[3]) # ensure page flow self._device.swipe(start, end, int(param[4])) self.actionResult = State.PASS def actionBack(self, param, refer=None): """ Simulate back button. :param param: None :param refer: None """ self._device.backBtn() self.actionResult = State.PASS def actionClick(self, param, refer=None): """ Simulate click action. :param param: [x, y] :param refer: A json object include view information. """ info = json.loads(refer) if len(param) == 2: x, y = param else: x, y, _ = param point = (int(x), int(y)) noShow = False wait = 5 while self._device.isConnected() and not self.isStopRun and wait > 0: try: if not self._device.isScreenOn(): self._device.powerBtn() if self._device.isLocked(): self._device.unlock() result = self._device.checkSamePoint(point, info) print 'reason = '+result['reason'] if result['answer']: self.actionResult = State.PASS break elif not noShow: print 'wait view' time.sleep(1) wait -= 1 except uiautomator.JsonRPCError as m: print 'JsonRPCError: ' + m.message break def actionCheckBlank(self, param, refer=None): """ Check the target view should not have any child view. :param param: [Identification, x, y]. There are 4 kinds of identification(resourceId, text, description, className). :param refer: None """ benchmark, x, y = param attribute, value = benchmark.split('=') point = (int(x), int(y)) if attribute == 'text': selector = self._device.retrieveSelector(point, self._device.d(text=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS elif attribute == 'description': selector = self._device.retrieveSelector(point, self._device.d(description=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS elif attribute == 'resourceId': selector = self._device.retrieveSelector(point, self._device.d(resourceId=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS elif attribute == 'className': selector = self._device.retrieveSelector(point, self._device.d(className=value)) if selector: if selector.childCount == 0: self.actionResult = State.PASS else: self.actionResult = State.PASS def actionCheckExist(self, param, refer=None): """ Check the target view whether is existed on the screen. :param param: [Identification, x, y]. There are 4 kinds of identification(resourceId, text, description, className). :param refer: None """ benchmark, x, y = param attribute, value = benchmark.split('=') point = (int(x), int(y)) if attribute == 'text': if self._device.retrieveSelector(point, self._device.d(text=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True elif attribute == 'description': if self._device.retrieveSelector(point, self._device.d(description=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True elif attribute == 'resourceId': if self._device.retrieveSelector(point, self._device.d(resourceId=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True elif attribute == 'className': if self._device.retrieveSelector(point, self._device.d(className=value)): self.actionResult = State.PASS # self.report.emit(INFO, 'Pass: View exist', True) else: pass # self.report.emit(ERROR, 'Fail: View not exist', True) # self.isStopRun = True def actionCheckNoExist(self, param, refer=None): """ Check the target view whether isn't existed on the screen. :param param: [Identification, x, y]. There are 4 kinds of identification(resourceId, text, description, className). :param refer: None """ benchmark, x, y = param attribute, value = benchmark.split('=') point = (int(x), int(y)) if attribute == 'text': if not self._device.retrieveSelector(point, self._device.d(text=value)): self.actionResult = State.PASS elif attribute == 'description': if not self._device.retrieveSelector(point, self._device.d(description=value)): self.actionResult = State.PASS elif attribute == 'resourceId': if not self._device.retrieveSelector(point, self._device.d(resourceId=value)): self.actionResult = State.PASS elif attribute == 'className': if not self._device.retrieveSelector(point, self._device.d(className=value)): self.actionResult = State.PASS def actionHideKeyboard(self, param, refer=None): """ Hide soft keyboard. If keyboard isn't showed, it will do nothing. :param param: None :param refer: None """ self._device.hideKeyboard() self.actionResult = State.PASS def stop(self): """ Exit this process. """ self.isStopRun = True def setStartIndex(self, index): """ chose a start index of actions. :param index: integer """ self.startIndex = index def takeScreenShot(self, path): self.picData = self.miniReader.getDisplay() self.picImage.loadFromData(self.picData) self.picImage.save(path, 'JPEG') def run(self): self.isStopRun = False self.limit = 0 try: self.miniServer.start() ''' wait for active ''' while not self.miniServer.isActive(): time.sleep(0.5) self.takeScreenShot(Global.IMAGE_FOLDER + '/%s_%d.jpg' % (self.playName, 0)) while self.limit < self.times and not self.isStopRun: self.limit += 1 for index in range(len(self.actions)): if not self.isStopRun and self.startIndex <= index: self.index = index + 1 item = self.actions[index] if self._device.isConnected(): self.actionResult = State.FAIL self.currentAction.emit(item) self.actionSwitcher.get(item.action())(item.parameter(), item.information()) self.actionDone.emit(item, self.actionResult, index) self.takeScreenShot(Global.IMAGE_FOLDER + '/%s_%d.jpg' % (self.playName, self.index)) time.sleep(0.5) else: self.isStopRun = True break except socket.error: print 'socket.error: device offline' self.isStopRun = True except ValueError as e: print 'ValueError: ' + e.message self.isStopRun = True except CalledProcessError, exc: print 'CalledProcessError: ' + str(exc) self.isStopRun = True finally:
class UpdateScreen(QtCore.QThread): """ Keep capturing screen in the background. """ loadDone = QtCore.Signal() startLoad = QtCore.Signal() deviceOffline = QtCore.Signal() isStop = False isPause = False def __init__(self, device, picPath): super(UpdateScreen, self).__init__() self.isStop = False self._device = device self.delay = 0 self.picPath = picPath self.lastFrameTime = 0 self.isRefresh = False self.nconcurrent = threading.BoundedSemaphore(5) self.miniServer = MiniServer(self._device) self.miniReader = MiniReader(self._device) def stop(self): """ Stop capturing """ self.isStop = True self.miniServer.stop() # def pause(self): # """ Pause capturing """ # self.isPause = True # self.miniServer.stop() # # def resume(self): # """ resume capturing """ # self.isPause = False # # self.start() def setDelay(self, delay=0): """ Set a delay time to start capturing """ self.delay = delay def run(self): time.sleep(self.delay) try: self.miniServer.start() self.isStop = False while not self.isStop: if self.miniServer.isActive(): if not self.miniServer.needRestart(): self.startLoad.emit() self.screenshot(path=self.picPath + '/' + self._device.serialno + '_screen.jpg') else: self.miniServer.reStart() else: time.sleep(0.5) # self.isPause = True except: print 'device not found' self.lastFrameTime = 0 self.deviceOffline.emit() finally: self.miniServer.stop() def monitorFrame(self): """ monitor the screen has changed, it helps to reduce capturing frequency. """ decryptRE = re.compile('\s+events-delivered:\s+(?P<number>\d+)') output = self._device.cmd.dumpsys(['SurfaceFlinger']) if output: m = decryptRE.search(output) if m is not None: tmpTime = m.group('number') if tmpTime != self.lastFrameTime: self.lastFrameTime = tmpTime return True def screenshot(self, path, delay=0): """ Do screen capture :param path: save pic path :param delay: delay to capture screen. In second. """ try: # self.nconcurrent.acquire() time.sleep(delay) pp = time.time() if self._device.isConnected(): if self.miniServer.isActive(): byteData = self.miniReader.getDisplay() im = QtGui.QImage.fromData(QtCore.QByteArray.fromRawData(byteData)) im.loadFromData(byteData) im.save(path, 'JPEG') else: if not self.isStop: self.deviceOffline.emit() print time.time() - pp if not self.isStop: self.loadDone.emit() except AttributeError as e: print e.message except IOError as e: print e.message # finally: # self.nconcurrent.release() def __byteToHex(self, bs): return ''.join(['%02X ' % ord(byte) for byte in bs]) def __parsePicSize(self, data): return int(binascii.hexlify(data[::-1]), 16) def __parseBanner(self, data): if len(data) == 24: banner = dict() banner['version'] = int(binascii.hexlify(data[0]), 16) banner['length'] = int(binascii.hexlify(data[1]), 16) banner['pid'] = int(binascii.hexlify(data[2:5][::-1]), 16) banner['real.width'] = int(binascii.hexlify(data[6:9][::-1]), 16) banner['real.height'] = int(binascii.hexlify(data[10:13][::-1]), 16) banner['virtual.width'] = int(binascii.hexlify(data[14:17][::-1]), 16) banner['virtual.height'] = int(binascii.hexlify(data[18:21][::-1]), 16) banner['orient'] = int(binascii.hexlify(data[22]), 16) banner['policy'] = int(binascii.hexlify(data[23]), 16) return banner