自动化测试往往需要在多个环境中跑测试用例，最开始我用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
继承配置项示例