如何在Python类中兼容命令行参数与代码调用传参？

我写了一个带命令行执行入口的Python类，__init__函数负责处理命令行参数：

import argparse

class Tester:

    def __init__(self):
        self.args_parser = argparse.ArgumentParser(description='Test')
        self.args = self.__parse_parameters()
...


if __name__ == "__main__":
    tester = Tester()

通过命令行执行时，比如运行：

python tester.py --test eating --lowcarb

参数能正常传递给__parse_parameters函数，运行没问题。

现在我想在其他Python代码里直接调用这个Tester类，该怎么给它传递这些参数？

方法1：修改__init__支持自定义参数列表 #

给__init__加一个可选的args参数，让它既支持从命令行读参数，也能接受外部传入的参数列表。同时调整__parse_parameters让它可以接收参数：

import argparse

class Tester:
    def __init__(self, args=None):
        self.args_parser = argparse.ArgumentParser(description='Test')
        self.args = self.__parse_parameters(args)
    
    def __parse_parameters(self, args=None):
        # 这里添加你的参数定义，示例：
        self.args_parser.add_argument('--test', type=str)
        self.args_parser.add_argument('--lowcarb', action='store_true')
        # 有传入参数就用传入的解析，否则用命令行参数
        return self.args_parser.parse_args(args)

# 命令行执行逻辑保持不变
if __name__ == "__main__":
    tester = Tester()

外部代码调用时，直接传参数列表即可：

from tester import Tester

# 传入和命令行格式一致的参数列表
tester = Tester(args=['--test', 'eating', '--lowcarb'])

方法2：拆分参数解析为公共方法 #

把参数解析逻辑抽成类的公共方法，外部可以先解析好参数再传给类实例，这样不用改动原有的__init__核心逻辑：

import argparse

class Tester:
    def __init__(self, args):
        self.args = args
    
    @classmethod
    def parse_args(cls, args=None):
        parser = argparse.ArgumentParser(description='Test')
        parser.add_argument('--test', type=str)
        parser.add_argument('--lowcarb', action='store_true')
        return parser.parse_args(args)

# 命令行执行入口
if __name__ == "__main__":
    args = Tester.parse_args()
    tester = Tester(args)

外部调用示例：

from tester import Tester

# 先调用公共方法解析参数
args = Tester.parse_args(['--test', 'eating', '--lowcarb'])
# 用解析好的参数实例化类
tester = Tester(args)

方法3：临时修改sys.argv（不推荐） #

如果不想改动类的代码，可以临时替换sys.argv，但这种方式容易影响其他依赖命令行参数的逻辑，不推荐使用：

import sys
from tester import Tester

# 保存原始命令行参数
original_argv = sys.argv.copy()
# 设置需要传入的参数
sys.argv = ['tester.py', '--test', 'eating', '--lowcarb']
# 实例化类
tester = Tester()
# 恢复原始参数，避免影响后续代码
sys.argv = original_argv

推荐优先用前两种方法，尤其是方法1，既保留了命令行执行的兼容性，又能方便地在代码内调用，逻辑清晰且无副作用。

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