autoMonkey框架源代码

# -*- coding: utf-8 -*- import os import time import datetime import threading import re from wsgiref.validate import validator from conf.config import config class AppShell(threading.Thread): def __init__(self): threading.Thread.__init__(self) self.thread_stop = False def __raise_error(self, msg=None): # Raise Exception with self-defined message. raise Exception(msg) def kill_adb_process(self): """ Kill ADB Process. Examples: | Kill Adb Process | """ try: status = os.system('taskkill /f /im adb.exe') if status == 0: print 'Pass! Killed ADB Process.' else: self.__raise_error('Fail! Could NOT kill ADB Process.') except Exception as e: print str(e) def start_adb_service(self): """ Start ADB Service. Examples: | Start Adb Service | """ try: text = os.popen('adb start-server') content = text.read() print content if '5037' not in content: self.__raise_error('Error: start ADB server failed. %s' % content) else: print 'Pass! ADB server started.' except Exception, e: print str(e) def get_device_name(self): """ Get Devices Name through ADB command. Examples: | ${device_name} | Get Adb Devices | """ device_name = '' adb_devices = "adb devices" try: text = os.popen(adb_devices) content = text.read().strip() print content res = content.splitlines() if 'device' not in content: self.__raise_error('Error: Could Not get device -> {}'.format(res[-1].split()[1])) device_name = res[-1].split()[0] except Exception, e: if str(e) == 'list index out of range': self.__raise_error('Error: Could NOT find device! Please check the phone has been attached to TestBed.') else: print str(e) return device_name def set_device_name(self, name=None): """ Set Devices Name after get its name through ADB command. Examples: | Set Device Name | """ if name == None: device_name = self.get_device_name() else: device_name = name if config['G_APP_DEVICE_NAME'] != '' and config['G_APP_DEVICE_NAME'] == device_name: print '-'*30 print 'Pass! Set device name!' print 'Device Name:' + config['G_APP_DEVICE_NAME'] print '-'*30 elif device_name == '': self.__raise_error('FAIL: Could NOT get device name!') else: config['G_APP_DEVICE_NAME'] = str(device_name) print '-'*30 print 'Pass! Set device name!' print 'Device Name: ' + config['G_APP_DEVICE_NAME'] print '-'*30 def get_phone_info(self, dev=None): if dev is None: dev = config['G_APP_DEVICE_NAME'] cmd = "adb -s {} shell cat /system/build.prop".format(dev) print cmd lines = os.popen(cmd).readlines() for line in lines: if 'version.release' in line: config['G_APP_DEVICE_VERSION'] = line.strip().split('=')[1] # Android 系统 elif 'ro.product.model' in line: config['G_APP_DEVICE_MODEL'] = line.strip().split('=')[1] # 手机名 elif 'ro.product.brand' in line: config['G_APP_DEVICE_BRAND'] = line.strip().split('=')[1] # 手机品牌 elif 'vm.heapsize' in line: config['G_APP_DEVICE_HEAPSIZE'] = line.strip().split('=')[1] # 单个虚拟机最大内存 elif 'vm.heapgrowthlimit' in line: config['G_APP_DEVICE_HEAPGROWTHLIMIT'] = line.strip().split('=')[1] # 每个APP最大内存 def get_total_mem(self, dev=None): # 获取手机内存 if dev is None: dev = config['G_APP_DEVICE_NAME'] try: cmd = 'adb -s {} shell cat /proc/meminfo'.format(dev) print cmd res = os.popen(cmd) for line in res.readlines(): if 'MemTotal:' in line: config['G_APP_DEVICE_MEMORY'] = line.split('MemTotal:')[1].strip() break except Exception, e: print str(e) def get_cpu_kernels(self, dev=None): # 获取手机CPU内核数 if dev is None: dev = config['G_APP_DEVICE_NAME'] try: cmd = 'adb -s {} shell cat /proc/cpuinfo |findstr processor'.format(dev) print cmd content = os.popen(cmd) res = content.readlines() if 'processor' in res[0]: config['G_APP_DEVICE_CPU_KERNELS'] = len(res) else: print 'Could NOT get CPU kenels!' except Exception, e: print str(e) def adb_install_package(self, apk=None, package_name=None): """ Install packages through ADB command. Examples: | Adb Install Package | ${apk_name} | ${package_name} | """ if apk is None: apk = config['G_APP_PACKAGE_NAME'] # else: # apk = os.path.join(config['G_APP_PACKAGE_NAME'], apk) if package_name is None: package_name = config['G_APP_PACKAGE_NAME'] if self.is_package_installed(package_name): self.adb_uninstall_package(package_name) try: cmd = 'adb -s {} install -r {}'.format(config['G_APP_DEVICE_NAME'], apk) print cmd text = os.popen(cmd) content = text.read() print content if 'Success' in content: print 'Pass: Install {} succeeded. Version: {}'.format(package_name, apk) else: self.__raise_error('Fail: Could NOT install {}'.format(package_name)) except Exception, e: print str(e) def adb_uninstall_package(self, package_name=None): """ Uninstall packages through ADB command. Examples: | Adb Uninstall Package | ${package_name} | """ if package_name is None: package_name = config['G_APP_PACKAGE_NAME'] if not self.is_package_installed(package_name): self.__raise_error('Fail: APP {} is not installed.'.format(package_name)) try: cmd = 'adb -s {} uninstall {}'.format(config['G_APP_DEVICE_NAME'], package_name) print cmd text = os.popen(cmd) content = text.read() print content if 'Success' in content: print 'Pass: Uninstall {} succeeded.'.format(package_name) else: self.__raise_error('Fail: Could NOT uninstall {}'.format(package_name)) except Exception as e: print str(e) def is_package_installed(self, package_name): """ Check target package is installed through ADB command. Examples: | ${status} | Is Package Installed | """ packages = self.get_third_party_packages() if package_name in packages: return True else: return False def get_third_party_packages(self): """ Get Third-party packages through ADB command. Examples: | ${apk_name} | Get Third Party Packages | """ apks = [] try: f = os.popen('adb shell pm list package -3') for x in f.readlines():