appium常用封装（一）

一、Dos下执行adb devices获取udid封装

#coding=utf-8
import os
class DosCmd:
    def excute_cmd_result(self,command):
        result_list = []
        result = os.popen(command).readlines()
        for i in result:
            if i =='\n':
                continue
            result_list.append(i.strip('\n'))
        return result_list

    def excute_cmd(self,command):
        os.system(command)

if __name__ == '__main__':
    dos = DosCmd()
    print dos.excute_cmd_result('adb devices')

二、判断端口是否被占用简单封装

#coding=utf-8
from dos_cmd import DosCmd
class Port:
    def port_is_used(self,port_num):
        '''
        判断端口是否被占用
        '''
        flag = None
        self.dos = DosCmd()
        command = 'netstat -ano | findstr '+str(port_num)
        result = self.dos.excute_cmd_result(command)
        if len(result)>0:
            flag = True
        else:
            flag = False
        return flag

    def create_port_list(self,start_port,device_list):
        '''start_port 4701
        生成可用端口
        @parameter start_port
        @parameter device_list
        '''
        port_list = []
        if device_list != None:
            while len(port_list) != len(device_list):
                if self.port_is_used(start_port) != True:
                    port_list.append(start_port)
                start_port = start_port +1
            return port_list
        else:
            print "生成可用端口失败"
            return None


if __name__ == '__main__':
    port = Port()
    li = [1,2,3,4,5]
    print port.create_port_list(4722,li)

三、对yaml文件读写操作，获取命令行返回

#coding=utf-8
import yaml
class WriteUserCommand:
    def read_data(self):
        '''
        加载yaml数据
        '''
        with open("../config/userconfig.yaml") as fr:
            data = yaml.load(fr)
        return data

    def get_value(self,key,port):
        '''
        获取value
        '''
        data = self.read_data()
        value = data[key][port]
        return value

    def write_data(self,i,device,bp,port):
        '''
        写入数据
        '''
        data = self.join_data(i,device,bp,port)
        with open("../config/userconfig.yaml","a") as fr:
            yaml.dump(data,fr)

    def join_data(self,i,device,bp,port):
        data = {
        "user_info_"+str(i):{
        "deviceName":device,
        "bp":bp,
        "port":port
        }
        }
        return data

    def clear_data(self):
        with open("../config/userconfig.yaml","w") as fr:
            fr.truncate()
        fr.close()

    def get_file_lines(self):
        data = self.read_data()
        return len(data)


if __name__ == '__main__':
    write_file = WriteUserCommand()
    print write_file.get_value('user_info_2','bp')

四、简单的获取设备信息、创建可用端口、生成command命令、多线程启动服务封装

#coding=utf-8
from dos_cmd import DosCmd
from port import Port
import threading
import time
from write_user_command import WriteUserCommand
class Server:
    def __init__(self):
        self.dos = DosCmd()
        self.device_list = self.get_devices()
        self.write_file = WriteUserCommand()
    def get_devices(self):
        '''
        获取设备信息
        '''
        
        devices_list = []
        result_list = self.dos.excute_cmd_result('adb devices')
        if len(result_list)>=2:
            for i in result_list:
                if 'List' in i:
                    continue
                devices_info = i.split('\t')
                if devices_info[1] == 'device':
                    devices_list.append(devices_info[0])
            return devices_list
        else:
            return None
    def create_port_list(self,start_port):
        '''
        创建可用端口
        '''
        port = Port()
        port_list = []
        port_list = port.create_port_list(start_port,self.device_list)
        return port_list

    def create_command_list(self,i):
        '''
        生成命令
        '''
        command_list = []
        appium_port_list = self.create_port_list(4700)
        bootstrap_port_list = self.create_port_list(4900)
        device_list = self.device_list
        command = "appium -p "+str(appium_port_list[i])+" -bp "+str(bootstrap_port_list[i])+" -U "+device_list[i]+" --no-reset --session-override --log D:/log/test02.log"
        command_list.append(command)
        self.write_file.write_data(i,device_list[i],str(bootstrap_port_list[i]),str(appium_port_list[i]))

        return command_list

    def start_server(self,i):
        '''
        启动服务
        '''
        self.start_list = self.create_command_list(i)
        print self.start_list

        self.dos.excute_cmd(self.start_list[0])

    def kill_server(self):
        server_list = self.dos.excute_cmd_result('tasklist | find "node.exe"')
        if len(server_list)>0:
            self.dos.excute_cmd('taskkill -F -PID node.exe')

    def main(self):
        thread_list = []
        self.kill_server()
        self.write_file.clear_data()
        for i in range(len(self.device_list)):
            appium_start = threading.Thread(target=self.start_server,args=(i,))
            thread_list.append(appium_start)
        for j in thread_list:
            j.start()
        time.sleep(25)


if __name__ == '__main__':
    server = Server()
    print server.main()