实现各个环境切换，一套脚本只要调通了，基本接口没有变动，可以大大的降低测试成本，主要用的是argparse
    �库。
    关键代码如下：

    1. #!/usr/bin/python
    2. # -*- coding: UTF-8 -*-
    3. # @Author   :zaygee
    4. # @time     :2020/12/3 17:17
    5. # @file     :runtest.py
    6. import argparse
    7. import shutil
    8. import sys
    9. import time
    10. from shutil import copyfile
    11. from allure_env import write_allure_env
    12. from common import logger
    13. from config import *
    15. project_path = os.path.dirname(os.path.abspath(__file__))
    16. sys.path.append(project_path)
    19. def mk_report_dir():
    20.     # 创建测试报告目录
    21.     report_path = os.path.join(project_path, 'report')
    22.     if not os.path.exists(report_path):
    23.         os.mkdir(report_path)
    24.     # 每次的测试报告
    25.     report_dir_path = os.path.abspath('{0}/{1}'.format(report_path, time.strftime("%Y-%m-%d", time.localtime())))
    26.     if not os.path.exists(report_dir_path):
    27.         os.mkdir(report_dir_path)
    28.     # 临时生成的文件
    29.     xml_tmp_path = os.path.join(report_dir_path, "xml_tmp_dir")
    30.     if not os.path.exists(xml_tmp_path):
    31.         os.mkdir(xml_tmp_path)
    32.     # html报告
    33.     now_time = time.strftime("%H%M%S", time.localtime())
    34.     allure_report_path = os.path.join(report_dir_path, "AllureReport{}".format(now_time))
    35.     if not os.path.exists(allure_report_path):
    36.         os.mkdir(allure_report_path)
    38.     return xml_tmp_path, allure_report_path
    41. # 生成html报告
    42. def mk_allure_report(xml_tmp_path, allure_report_path):
    43.     # 写入allure报告的执行环境信息
    44.     write_allure_env()
    45.     # 复制environment.properties文件到allure根目录
    46.     copyfile('environment.properties', '{}'.format(os.path.join(xml_tmp_path, 'environment.properties')))
    47.     os.system("allure generate {0} -o {1} -c".format(xml_tmp_path, allure_report_path))
    48.     shutil.rmtree(xml_tmp_path)
    51. # 生成json
    52. def create_report_xml(case_path, xml_tmp_path):
    53.     os.system("python -m  pytest {} -v -s --alluredir={}".format(case_path, xml_tmp_path))
    56. def pytest_main():
    57.     """用例执行主函数"""
    58.     parser = argparse.ArgumentParser(description='命令行执行用例参数')
    59.     parser.add_argument('--env', '-e', help="运行环境 test、staging、prod", type=str, default='dev')
    60.     parser.add_argument('--case_path', '-c', help="执行用例,默认执行全部", type=str, default='kitten_cases')
    61.     args = parser.parse_args()
    62.     run_env = args.env
    63.     case_path = args.case_path
    65.     if run_env is not None:
    66.         if run_env not in ('dev', 'test', 'staging', 'prod'):
    67.             raise ValueError('环境请输入:dev、test、staging、prod')
    68.         os.environ['env'] = run_env
    69.         logger.info("当前运行环境为： " + str(run_env))
    70.     xml_tmp_path, allure_report_path = mk_report_dir()
    71.     create_report_xml(case_path, xml_tmp_path)
    72.     if os.listdir(xml_tmp_path):
    73.         mk_allure_report(xml_tmp_path, allure_report_path)
    76. if __name__ == '__main__':
    77.     pytest_main()
    78.     """
    79.     执行命令：  python main.py -e {运行环境} -c {执行用例}
    80.     eg：  python main.py -e prod -c cases    """