python如何实现运行时才选择配置参数？

自动化测试用例往往需要在多个环境中跑测试用例，最开始我用pytest+selenium搭建了一套测试工程，使用jenkins管理定时拉起任务，每次需要在另一个环境跑自动化时就临时改下配置再拉起任务。改的次数多了就烦了，就想办法解决这个问题，实现在运行时根据用户输入的参数来决定使用哪个环境的配置。

要实现运行时根据用户输入来决定使用什么环境配置，需要借助系统环境变量，首先运行程序时提取用户输入参数并将其写入环境变量，然后程序通过读取系统环境变量，根据环境变量的值来决定要运行的环境参数，下面列出python示例代码来演示是如何实现的。

示例代码

main.py为自动化工程运行的入口，为简便起见，这里仅将选择的环境参数打印出来，代码如下：

import sys
import os
from conf import *  

class Autotest:
    def __init__(self):
        self.running_env = ""
        self.config = None  

    def get_env_config(self):
        self.running_env = os.environ["env"]
        if "dev" in self.running_env:
            self.config = DevConfig
        else:
            self.config = ProdConfig

    def show_config(self):
        print(self.config.desc)
        print(self.config.demo_url)
        print(self.config.demo_other)
  

if __name__ == "__main__":
    app = Autotest()
    # 根据运行输入参数设置环境变量
    if len(sys.argv) > 1:
        os.environ["env"] = sys.argv[1]
    else:
        # 用户没有输入默认设置环境为开发环境
        os.environ["env"] = "dev"
    # 读取环境变量并根据环境变量获取运行配置
    app.get_env_config()
    app.show_config()

conf.py为环境配置文件，和main.py在同一个目录，代码如下：

class Config:
    desc = ""
    demo_url = ""
    demo_other = "继承配置项示例"  

class DevConfig(Config):
    desc = "开发环境"
    demo_url = "https://localhost:8080"
	# 未设置的参数直接继承Config的配置

class ProdConfig(Config):
    desc = "生产环境"
    demo_url = "https://localhost:8090"
    # 未设置的参数直接继承Config的配置

验证一下

接下来实际运行测试一下，cmd下执行"python main.py dev"命令将打印如下信息：

开发环境
https://localhost:8080
继承配置项示例

执行"python main.py prod"将打印如下信息：

生产环境
https://localhost:8090
继承配置项示例

加微信即可免费领取下面的自动化测试资料和一份软件测试面试宝典