python实现自动化群控的步骤

admin

一. 前言
' T0 q- e* m  [0 R4 a$ w  U       群控，相信大部分人都不会陌生！印象里是一台电脑控制多台设备完成一系列的操作，更多的人喜欢把它和 Hui 产绑定在一起！事实上，群控在自动化测试中也被广泛使用！本篇文章从基础开始，聊聊使用「 Python + adb 」命令如何编写一套群控脚本。
二. 准备
       在本机安装 Android 开发环境，保证 adb 被添加到环境变量，将准备好的多台设备，使用数据线（ 或者通过 Hub ）连接到电脑上，通过 adb devices 命令查看已经连接的所有设备。

复制代码
# 下面显示连接了3台设备
xag:Test xingag[        DISCUZ_CODE_2        ]nbsp;adb devices
List of devices attached
822QEDTL225T7    device
ca2b3455        device
DE45d9323SE96   device

三. 实战
( j/ o" p% L3 B8 D; R% s) ]       自动化群控以闲鱼 App 的一次关键字搜索为例，步骤包含：打开应用、点击到搜索界面、输入内容、点击搜索按钮，下面通过7步来完成这一操作。
. x- z# d. f9 D" |1、获取目标应用的包名及初始化 Activity
7 a' i) D/ N8 i5 X; H; l. k1 L       获取方式有很多种，主流方式包含：adb 命令、解析 APK、第三方 APK、无障碍服务，这里推荐使用 adb 命令这种方式。

复制代码
# 获取当前运行应用的包名及初始Activity
adb shell dumpsys activity | grep -i run

      打开闲鱼 App，在命令终端输入上面的命令，终端会将包名及 Activity 名称显示出来。
2、获取所有在线的设备
       通过 adb devices 命令，通过输出内容，进行一次过滤，得到所有连接到 PC 端的设备。
9 ?6 m: K5 ]' ?$ R  G

复制代码
# 所有设备ID
devices = []

def get_online_devices(self):
    """
    获取所有在线的设备
    :return:
    """
    global devices
    try:
        for device_serias_name in exec_cmd("adb devices"):
           # 过滤掉第一条数据及不在线的设备
           if "device" in device_serias_name:
              devices.append(device_serias_name.split("\t")[0])
           devices = devices[1:]
    except Exception as e:
            print(e)

    # 连上的所有设备及数量
    return devices

3、群控打开目标应用
       遍历设备列表，使用 adb -s 设备ID shell am start -W 命令分别打开目标应用。

复制代码
def start_app(self):
    """
    打开App
    :return: 
    """
    for device in devices:
        os.popen("adb -s " + device + " shell am start -W {}/{}".format(self.packageName, self.home_activity))
    print('等待加载完成...')
    sleep(10)

4、封装执行步骤
       为了方便管理设备，将每一步的操作写入到YAML文件中，可以通过 ID 查找元素并执行点击操作、在输入框中输入内容、调用本地方法及输入参数。
$ i6 O  P. T. u       这里分别对应：保存 UI 树控件、查找输入框元素并执行点击操作、保存 UI 树控件（界面变化了）、输入文本内容、查看搜索按钮元素并执行点击操作。
( v: Y3 T3 r6 I& |' Y9 v3 m4 U' N
# q3 O+ s0 J, F- P0 J5 ~: s0 h7 U

复制代码
# steps_adb.yaml

# 包名和Activity
package_name:  com.taobao.idlefish
home_activity:  com.taobao.fleamarket.home.activity.InitActivity

# 执行步骤
steps:
  - save_ui_tree_to_local:
      method:  save_ui_tree_to_local
      args:
  - find_element_and_click:
      id:  com.taobao.idlefish:id/tx_id
  - save_ui_tree_to_local:
      method:  save_ui_tree_to_local
  - input_content:
      content:  Python
  - find_element_and_click:
      id:  com.taobao.idlefish:id/search_button

      需要指出的是，为了提高群控的适配性，控件的实际坐标需要通过下面的步骤去获取：
% g* a$ X; p6 h) Q: h, v

• 导出界面的控件树
• 解析控件树 XML 文件，利用正则表达式得到目标控件的坐标值
• 计算出控件的中心点坐标
  8 A' J6 U, I% K  f

/ w2 l3 Z0 D/ s( z/ N
, g9 I3 B' q! ^% z+ ^
0 v9 J6 ^4 @* g5 ?  b+ O       利用控件 ID 获取元素中心点坐标的实现代码如下：
7 d; J9 {+ h3 u0 W. Y6 a3 t' y% z

复制代码
def get_element_position(element_id, uidump_name):
    """
    通过元素的id，使用ElementTree，解析元素控件树，查找元素的坐标中心点
    :param element_id: 元素id，比如：
    :return: 元素坐标
    """

    # 解析XML
    tree = ET.parse('./../%s.xml' % uidump_name)
    root = tree.getroot()

    # 待查找的元素
    result_element = None

    # print('查找数目', len(root.findall('.//node')))

    # 遍历查找node元素
    # 通过元素id
    for node_element in root.findall('.//node'):
        if node_element.attrib['resource-id'] == element_id:
            result_element = node_element
            break

    # 如果找不到元素，直接返回空
    if result_element is None:
        print('抱歉！找不到元素！')
        return None

    # 解析数据
    coord = re.compile(r"\d+").findall(result_element.attrib['bounds'])

    # 中心点坐标
    position_center = int((int(coord[0]) + int(coord[2])) / 2), int((int(coord[1]) + int(coord[3])) / 2)

    return position_center

5、区分设备
8 u8 e; o$ q( q3 h% ?       为了保证群控脚本执行不会产生干扰，在每个步骤执行之前，都应该将设备 ID 作为参数进行区分。比如：将控件的界面控件树按照设备保存为不同的名称、点击界面和输入的命令传相应设备 ID 作为入参。

& V( t8 k# _0 s/ d  Q5 x1 E

复制代码
def save_ui_tree_to_local(dName):
    """
    获取当前Activity控件树，保存到本地
    文件名固定为：uidump.xml
    :param dName: 设备id
    :return:
    """

    exec_cmd("adb  -s %s shell uiautomator dump /data/local/tmp/%s.xml" % (dName, dName))

    sleep(2)

    exec_cmd("adb -s %s pull /data/local/tmp/%s.xml ./../" % (dName, dName))

6、执行步骤
       从 YAML 文件中读取执行步骤，遍历步骤集合，内部遍历设备列表，以保证每一个步骤，分别执行到每台设备上。

. |7 b/ a7 @5 {- L1 G0 B

复制代码
# 执行步骤
for step in self.steps:
    # 设备
    for device in devices: 
        pass

      接着，通过步骤名称匹配不同的操作，即可操作设备了。
) z' H8 K8 x  K5 {% h+ ^( d# R

复制代码
# 操作名称
step_name = list(step)[0]

if step_name == 'save_ui_tree_to_local':
    # 保存UI数到本地
    method = step.get(step_name).get('method')
    save_ui_tree_to_local(device)
elif step_name == 'find_element_and_click':
    element_id = step.get(step_name).get('id')
    # 获取元素的坐标
    bound_search_input = get_element_position(element_id, device)
    # 点击元素
    exec_cmd('adb -s %s shell input tap %s %s' % (device, bound_search_input[0], bound_search_input[1]))
elif step_name == 'input_content':
    input_content = step.get(step_name).get('content')
    # 模拟输入
    exec_cmd('adb -s %s shell input text %s' % (device, input_content))
else:
    print('其他操作步骤')

7、关闭应用
  N4 d1 u0 H0 u" P/ Q  K       当所有的操作完成之后，同样是遍历设备，利用 adb 命令去关闭 App 即可。
8 u* c% j: E/ Q" U1 K
8 @$ E: C% @4 c* x, i* s

复制代码
def stop_all(self):
   """
   关闭应用
   :return:
   """
   for device in devices:
       os.popen("adb -s " + device + " shell am force-stop  %s" % self.packageName)

四. 最后
) ]1 v$ ]+ F) A# e6 q. X6 l4 N8 ~       本篇仅仅是 Python 自动化群控最简单的实现方式，后面将和大家讨论更加复杂的实现方式。
项目地址：https://github.com/xingag/test_auto/tree/master/group_control

7 [. g6 C, S. Y. s, Q# }