最近在 TesterHome 游戏测试群里，有时候会看到有童鞋问，游戏测试人员学了 Python，可以干点什么。

很多童鞋初学 Python，学习了语法和基础类库后，开始迷茫如何实际使用到工作中去，其实 Python 可以做的事情是很多的，将日常工作的一些事情自动化，对我们的工作效率有很大的提升。

本文面向 Py 新手，分享一些辅助工作的小工具思路。以下例子都是在 Win10 + Py3.5 下完成。

调用 CMD

subprocess 是 Python 自带的子进程管理模块，定义有数个创建子进程的函数，也提供了一些管理标准流 (standard stream) 和管道 (pipe) 的工具，从而在进程间使用文本通信。

简单理解就是，你通过 CMD 敲的命令，都基本可以用 subprocess 来实现批量处理。

例子 1：批量 SVN 操作

以更新 SVN 为例，这是一个频繁的操作，尤其是多个 SVN 目录需要一一更新的时候，手动起来是挺麻烦的。

import subprocess
subprocess.Popen(r'TortoiseProc.exe /command:update /path:"C:\project\策划文档" /closeonend:0')
subprocess.Popen(r'TortoiseProc.exe /command:update /path:"C:\project\配置文档" /closeonend:0')

例子 2：adb 命令的封装

做安卓手游测试的时候，adb 是常用工具，我们可以通过它，进行 apk 的安装，卸载，截图，获取 APK 信息，性能数据，获取手机信息等等操作。

比如获取当前运行在前台的 apk 的 package 和 activity 名称

def run_cmd(cmd):
    """执行CMD命令"""
    p = subprocess.Popen(cmd, stdout=subprocess.PIPE)
    return [i.decode() for i in p.communicate()[0].splitlines()]


def get_apk_info():
    """获取apk的package，activity名称

    :return: list  eg ['com.android.calendar', 'com.meizu.flyme.calendar.AllInOneActivity']
    """
    result = run_cmd("adb shell dumpsys activity top")
    for line in result:
        if line.strip().startswith('ACTIVITY'):
            return line.split()[1].split('/')

print(get_apk_info())

output: ['com.android.calendar', 'com.meizu.flyme.calendar.AllInOneActivity']

比如查看当前 apk 的内存占用

def get_mem_using(package_name=None):
    """查看apk的内存占用

    :param package_name:
    :return: 单位KB
    """
    if not package_name:
        package_name = get_apk_info()[0]
    result = run_cmd("adb shell dumpsys meminfo {}".format(package_name))
    info = re.search('TOTAL\W+\d+', str(result)).group()
    mem = ''
    try:
        mem = info.split()
    except Exception as e:
        print(info)
        print(e)
    return mem[-1]

output: 37769

比如备份当前 apk 到桌面

def backup_current_apk(path=r"C:\Users\jianbing\Desktop\apks"):
    package = get_apk_info()[0]
    result = run_cmd("adb shell pm path {}".format(package))
    cmd = "adb pull {} {}".format(result[0].split(":")[-1], os.path.join(path, "{}.apk".format(package)))
    print(cmd)
    run_cmd(cmd)

再进一步，将常用的 adb 操作封装为一个 ADB 工具类。社区里也有童鞋之前分享过，传送门。

处理文本

例子：在整个文件夹中搜索关键字

某天策划说，这个版本他删掉了某个道具，让我检查下有没有删漏的地方，这个道具产出的地方不少，最佳的检查方式是各个相关配置表看下还有没有配置这个道具。

那就写个脚本遍历整个文件夹来搜索指定关键字吧。

import os


def get_files_by_suffix(path, suffixes=("txt", "xml"), traverse=True):
    """从path路径下，找出全部指定后缀名的文件

    :param path: 根目录
    :param suffixes: 指定查找的文件后缀名
    :param traverse: 如果为False，只遍历一层目录
    :return:
    """
    file_list = []
    for root, dirs, files in os.walk(path):
        for file in files:
            file_suffix = os.path.splitext(file)[1][1:].lower()   # 后缀名
            if file_suffix in suffixes:
                file_list.append(os.path.join(root, file))
        if not traverse:
            return file_list

    return file_list


if __name__ == '__main__':

    keyword = "XXX宝箱"

    files = get_files_by_suffix(r"C:\project\config")

    for file in files:

        with open(file, 'r', encoding='utf-8', errors='ignore') as f:
            content = f.read().lower()
            position = content.find(keyword.lower())

            if position != -1:
                print("Find in {0}".format(file))
                start = position - 100 if position - 100 > 0 else 0
                end = position + 100 if position + 100 < len(content) else len(content)
                print(content[start:end])
                print("_" * 100)

操作远程服务器

例子 1：查看内网发版时间

有时候问开发，最近一次内网服务端发版是什么时候？开发回答：有点忘记了。。那就得自力更生了~

手动方式：使用 FTP 软件连入内网服务器，查看文件的更新日期，从而知道发版时间。

懒人方式：Py 大法好~

paramiko 是 Python 很有名的第三方库，遵循 SSH2 协议，支持以加密和认证的方式，进行远程服务器的连接。

import paramiko
import time

_transport = paramiko.Transport("192.168.1.10:22")
_transport.connect(username="root", password="XXXXXX")
sftp = paramiko.SFTPClient.from_transport(_transport)

result = sftp.listdir_attr("/data/www/sg/sg_dev/socket/conf/config/treasure")
print("发版时间是：{}".format(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(result[0].st_mtime))))

sftp.close()

例子 2：查看内网报错信息

在进行测试的时候，需要多留意服务端是否有新的报错信息，有些报错在客户端并没有什么表现，比如数据进库失败，手动方式：通过 SecureCRT 连入内网服务器，CD 到 Log 目录下，然后 tail -n 200 sg_error.log 查看最新的报错信息。

于是萌生了写一个小工具来定时检测，发现报错信息就保存起来的想法。

import datetime
import paramiko
import time
import os


class ScanError(object):

    def __init__(self):

        self._ssh = paramiko.SSHClient()
        self.last_error_log = None
        self._init()

    def _init(self):
        os.chdir("data")   # 打算将报错信息保存到data目录下
        self._ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self._ssh.connect("192.168.1.10", username="root", password="XXXXXX")

        error_log = self.get_error_log(500)
        self.last_error_log = error_log

        # 检测最近三天有没有报错信息
        today = datetime.date.today()
        yesterday = today - datetime.timedelta(days=1)
        the_day_before_yesterday = today - datetime.timedelta(days=2)

        error_log_str = "\n".join(error_log)
        if error_log_str.find(str(today)) > -1 or error_log_str.find(str(yesterday)) > -1 or error_log_str.find(str(the_day_before_yesterday)) > -1:
            self.save_error_log("error.txt", error_log)
            print('内网最近三天有错误信息，请查看')
            os.popen('error.txt')

    def get_error_log(self, num=200):
        cmd = 'cd /data/www/sg/sg_dev/socket/log&&tail -n {} sg_error.log'.format(num)
        stdin, stdout, stderr = self._ssh.exec_command(cmd)
        error_log = [i.decode("utf-8") for i in stdout.read().splitlines() if i]
        return error_log

    @staticmethod
    def save_error_log(file_name, log: list):
        with open(file_name, 'w', encoding='utf-8') as f:
            f.write("\n".join(log))

    def run_forever(self, interval=30, show_error=True):
        """运行检测工具

        :param interval: 检测间隔
        :param show_error: 是否检测到报错就自动弹出显示
        :return:
        """
        while 1:
            time.sleep(interval)
            error_log = self.get_error_log()

            if error_log != self.last_error_log and "\n".join(set(error_log) - set(self.last_error_log)).find("ERROR") > -1:

                self.last_error_log = error_log
                file_name = time.strftime("%Y-%m-%d-%H-%M-%S.txt", time.localtime(time.time()))
                print('{} 检测到内网有新的错误信息'.format(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time()))))
                self.save_error_log(file_name, error_log)

                if show_error:
                    os.popen(file_name)


if __name__ == '__main__':
    ScanError().run_forever()

数据库操作

例子：找号功能

内网的服务器建了 N 多的号，有时候看着排行榜某个帐号，想登录看下数据，可以使用 Python 写一个连接数据库的找号脚本。

import cymysql

player_name = "XXXXXX"
conn = cymysql.connect(host='XXXXXX', user='sg', passwd='XXXXXX', db="dev", charset='utf8')
cur = conn.cursor()

sql = "select * from Player where name like '%{0}%'".format(player_name)   # 模糊搜索，从玩家名称搜索玩家ID
cur.execute(sql)
for r in cur.fetchall():
    sql = "select * from Account where uid = '{0}'".format(r[0])   # 从玩家ID搜索玩家帐号
    cur.execute(sql)
    for row in cur.fetchall():
        print('{0}, {1}, {2}'.format(r[0], r[1], row[2]))  # 打印相关信息

conn.close()

扩展开发提供的工具

在之前某个项目，开发做了一个给游戏帐号发道具的网页，提供测试使用，操作流程是这样的，在网页上的表单里边，填写玩家的 ID，在下拉列表选中要发送的道具（支持模糊搜索），填写数量。

这个网页使用起来，工作效率不高的地方就是，每次添加道具，都需要重新选择道具和填写数量，且添加过程没有记录下来，无法复用。

优化方案，在网页上点击添加道具，其实就是网页给游戏服务器发送了一个 HTTP 请求，那就直接让 Python 来代劳吧~

import requests

player_id = 1100000103
server_ip = "192.168.1.21:5000"

data = {"stuffList": []}
url = "http://{}/api/{}/stuff".format(server_ip, player_id)

data["stuffList"].append({"itemID": 1104000007, "number": 1000})  # itemID为1104000007的物品，数量1000
data["stuffList"].append({"itemID": 1104000008, "number": 1000})
data["stuffList"].append({"itemID": 1104000009, "number": 1000})
data["stuffList"].append({"itemID": 1104000010, "number": 1000})
data["stuffList"].append({"itemID": 1104000011, "number": 1000})
data["stuffList"].append({"itemID": 1104000012, "number": 1000})

requests.post(url, json=data, timeout=5)    # 添加道具
# requests.delete(url, json=data, timeout=5)   # 删除道具 

有没有发现，每个脚本都很简短~