Airtest 如何添加自定义参数给脚本

最近碰到一个有意思的小问题，命令行执行air脚本时，需要动态的更改脚本里某一参数。为此各种百度  谷歌，最后还是在官方文档找到对应的思路。

通常情况下，我们用的启动方式为命令行如：

airtest run /Users/a1/soft/berserker/airtest.shell/phone.air --device android://127.0.0.1:5037/device

这种方式无法动态更改脚本参数，在官方文档中，有自定义启动器的方法可以实现。

在这个里面有自定义参数的添加方法，是不是很简单。刚开始漏过这一方法，导致走了很多弯路。分析下对应的air脚本运行过程源码runner.py

下面分析airtest的运行过程,通过分析我们将知道airtest是怎么运行.air文件脚本的

关键代码有两个地方，一个是main文件，作为接收命令行参数，另一个是/core/cli/runner.py文件里面

先来看看main.py文件

def main(argv=None):
    ap = get_parser()
    args = ap.parse_args(argv)
    if args.action == "info":
        from airtest.cli.info import get_script_info
        print(get_script_info(args.script))
    elif args.action == "report":
        from airtest.report.report import main as report_main
        report_main(args)
    elif args.action == "run":
        from airtest.cli.runner import run_script
        run_script(args)
    elif args.action == "version":
        from airtest.utils.version import show_version
        show_version()
    else:
        ap.print_help()

所以执行脚本是通过run参数，run参数再调用runner.py文件的run_script方法，因此呢我们可以自定义执行脚本，只需要接收参数并调用run_script方法即可，此处官方文档已经给出了自定义启动器的教程，我们已经知道原理了，可以参考官方文档。

详细分析过程可以参考下这位大佬的，传送门

下面对应出我的方法

首先 设置好自定义的启动器，添加参数

from airtest.cli.parser import runner_parser
from airtest.cli.runner import AirtestCase, run_script


class CustomAirtestCase(AirtestCase):
    def setUp(self):
        if self.args.phone:
            print(self.args.phone)


if __name__ == '__main__':
    ap = runner_parser()
    ap.add_argument("--phone", help="phone number arg")
    args = ap.parse_args()
    run_script(args, CustomAirtestCase)

如此一来，启动脚本方式就更改为（命令行，python3启动）

python3 start.py airtest.shell/phone.air --phone *****

下面就简单了，python中有获取命令行参数的方法：sys.argv 参考菜鸟教程

因此，脚本中可直接获取上面启动命令中的 phone值。

另外，如需指定手机。可如下操作：

python3 start.py  airtest.shell/your.air --device android://127.0.0.1:5037/设备号 --phone ***