pytest多執行緒與多裝置並行appium

"""
@Title   : app多裝置並行-appium+pytest
@Author  : 檸檬班-小簡
@Email   : lemonban_simple@qq.com
"""
 
class ManageDevices:
    """
       1、重啟adb服務。
       2、通過adb devices命令獲取當前平臺中,已連線的裝置個數，和裝置uuid.
       3、通過adb -P 5037 -s 裝置uuid shell getprop ro.build.version.release獲取每一個裝置的版本號。
       4、將所有已連線裝置的裝置名稱、裝置版本號儲存在一個列表當中。
       5、通過呼叫get_devices_info函數，即可獲得4中的列表。
    """
 
    def __init__(self):
        self.__devices_info = []
        # 重啟adb服務
        self.__run_command_and_get_stout("adb kill-server")
        self.__run_command_and_get_stout("adb start-server")
 
    def get_devices_info(self):
        """
        獲取已連線裝置的uuid,和版本號。
        :return: 所有已連線裝置的uuid,和版本號。
        """
        self.__get_devices_uuid()
        print(self.__devices_info)
        self.__get_device_platform_vesion()
        return self.__devices_info

def devices_pool(port=4723,system_port=8200):
    """
    裝置啟動引數管理池。含啟動引數和對應的埠號
    :param port: appium服務的埠號。每一個裝置對應一個。
    :param system_port: appium服務指定的本地埠，用來轉發資料給安卓裝置。每一個裝置對應一個。
    :return: 所有已連線裝置的啟動引數和appium埠號。
    """
    desired_template = __get_yaml_data()
    devs_pool = []
    # 獲取當前連線的所有裝置資訊
    m = ManageDevices()
    all_devices_info = m.get_devices_info()
    # 補充每一個裝置的啟動資訊，以及設定對應的appium server埠號
    if all_devices_info:
        for dev_info in all_devices_info:
            dev_info.update(desired_template)
            dev_info["systemPort"] = system_port
            new_dict = {
                "caps": dev_info,
                "port": port
            }
            devs_pool.append(new_dict)
            port += 4
            system_port += 4
    return devs_pool

import subprocess
import os
 
from Common.handle_path import appium_logs_dir
 
class ManageAppiumServer:
    """
    appium desktop通過命令列啟動appium服務。
    不同平臺上安裝的appium，預設的appium服務路徑不一樣。
    初始化時，設定appium服務啟動路徑
    再根據給定的埠號啟動appium
    """
 
    def __init__(self,appium_server_apth):
        self.server_apth = appium_server_apth
 
    # 啟動appium server服務
    def start_appium_server(self,port=4723):
        appium_log_path = os.path.join(appium_logs_dir,"appium_server_{0}.log".format(port))
        command = "node {0} -p {1} -g {2} " 
                  "--session-override " 
                  "--local-timezone " 
                  "--log-timestamp & ".format(self.server_apth, port, appium_log_path)
        subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE,shell=True).communicate()
 
    # 關閉appium服務
    @classmethod
    def stop_appium(cls,pc,post_num=4723):
        '''關閉appium服務'''
        if pc.upper() == 'WIN':
            p = os.popen(f'netstat  -aon|findstr {post_num}')
            p0 = p.read().strip()
            if p0 != '' and 'LISTENING' in p0:
                p1 = int(p0.split('LISTENING')[1].strip()[0:4])  # 獲取程序號
                os.popen(f'taskkill /F /PID {p1}')  # 結束程序
                print('appium server已結束')
        elif pc.upper() == 'MAC':
            p = os.popen(f'lsof -i tcp:{post_num}')
            p0 = p.read()
            if p0.strip() != '':
                p1 = int(p0.split('n')[1].split()[1])  # 獲取程序號
                os.popen(f'kill {p1}')  # 結束程序
                print('appium server已結束')

def pytest_addoption(parser):
    parser.addoption(
        "--cmdopt", action="store", default="{platformName:'Android',platformVersion:'5.1.1'}",
        help="my devices info"
    )
 
 
@pytest.fixture(scope="session")
def cmdopt(request):
    return request.config.getoption("--cmdopt")
 
 
@pytest.fixture
def start_app(cmdopt):
    device = eval(cmdopt)
    print("開始與裝置 {} 進行對談，並執行測試用例 ！！".format(device["caps"]["deviceName"]))
    driver = start_appium_session(device)
    yield driver
    driver.close_app()
    driver.quit()

# 根據裝置啟動資訊，通過pytest.main來收集並執行用例。
def run_cases(device):
  """
  引數：device為裝置啟動引數。在pytest.main當中，傳遞給--cmdopt選項。
  """
    print(["-s", "-v", "--cmdopt={}".format(device)])
    reports_path = os.path.join(reports_dir,"test_result_{}_{}.html".format(device["caps"]["deviceName"], device["port"]))
    pytest.main(["-s", "-v",
                 "--cmdopt={}".format(device),
                 "--html={}".format(reports_path)]
                )

# 第一步：從裝置池當中，獲取當前連線的裝置。若裝置池為空，則無裝置連線。
devices = devices_pool()
 
# 第二步：若裝置池不為空，啟動appium server.與裝置個數對應。起始server埠為4723，每多一個裝置，埠號預設+4
if devices and platform_name and appium_server_path:
    # 建立執行緒池
    T = ThreadPoolExecutor()
    # 範例化appium服務管理類。
    mas = ManageAppiumServer(appium_server_path)
    for device in devices:
        # kill 埠，以免佔用
        mas.stop_appium(platform_name,device["port"])
        # 啟動appium server
        task = T.submit(mas.start_appium_server,device["port"])
        time.sleep(1)
 
    # 第三步：若裝置池不為空，在appium server啟動的情況下，執行app自動化測試。
    time.sleep(15)
    obj_list = []
    for device in devices:
        index = devices.index(device)
        task = T.submit(run_cases,device)
        obj_list.append(task)
        time.sleep(1)
 
    # 等待自動化任務執行完成
    for future in as_completed(obj_list):
        data = future.result()
        print(f"sub_thread: {data}")
 
    # kill 掉appium server服務,釋放埠。
    for device in devices:
        ManageAppiumServer.stop_appium(platform_name, device["port"])