如何在多文件中共享Python argparse的--alation-instance参数？

方案一：逐层传递所需参数（最推荐） #

直接将解析好的alation_instance参数沿着调用链向下传递，无需全局状态或重复解析参数，逻辑清晰无副作用。

修改步骤： #

1. main.py：调用alation_cronjob.main时传入解析后的参数

import argparse
import alation_cronjob

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--alation-instance', required=True, choices=['prod', 'dev'])
    # 其他参数定义...
    args = parser.parse_args()
    
    # 仅传递需要的参数，或直接传入整个args对象（如果后续有扩展需求）
    alation_cronjob.main(alation_instance=args.alation_instance)

if __name__ == '__main__':
    main()

2. alation_cronjob.py：修改main函数接收参数，并传递给alation模块的相关调用

import alation

def main(alation_instance):
    # 调用alation模块功能时传入参数
    alation.handle_refresh_token(alation_instance)

3. alation.py：修改调用get_refresh_token的逻辑，传入参数

from utils import get_refresh_token

def handle_refresh_token(alation_instance):
    # 传递参数给utils的函数
    refresh_token = get_refresh_token(alation_instance)
    # 后续业务逻辑...

4. utils.py：修改get_refresh_token函数，直接使用传入的参数

def get_refresh_token(alation_instance):
    if alation_instance == 'prod':
        # 生产环境的token获取逻辑
        return 'prod_refresh_token'
    elif alation_instance == 'dev':
        # 开发环境的token获取逻辑
        return 'dev_refresh_token'

方案二：全局配置对象（适合参数较多场景） #

如果后续需要传递多个参数，不想修改多层函数签名，可以用一个全局配置模块存储解析后的参数。

修改步骤： #

1. 新建config.py：定义全局配置变量

# 初始化配置变量，后续由main.py赋值
alation_instance = None

2. main.py：解析参数后赋值给全局配置

import argparse
import config
import alation_cronjob

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--alation-instance', required=True, choices=['prod', 'dev'])
    args = parser.parse_args()
    
    # 将参数存入全局配置
    config.alation_instance = args.alation_instance
    
    alation_cronjob.main()

if __name__ == '__main__':
    main()

3. utils.py：导入config模块直接使用参数

import config

def get_refresh_token():
    instance = config.alation_instance
    if instance == 'prod':
        return 'prod_refresh_token'
    elif instance == 'dev':
        return 'dev_refresh_token'

方案对比 #

• 方案一：调用关系明确，无全局状态污染，测试时可直接传入参数模拟场景，适合当前短调用链的场景。
• 方案二：无需修改多层函数签名，适合参数较多或调用链复杂的情况，但全局状态可能增加测试复杂度（单进程cronjob场景无并发问题）。

内容的提问来源于stack exchange，提问作者Byron Smith