使用标志变量从另一个python脚本执行python求解器

Question

我目前正在使用Python Pipline工作，在该管道中，我需要从Python代码调用不同的求解器/脚本并执行它们。由于将涉及不同的代码，因此我想根据标志参数称这些求解器为其称为这些求解器，并相应地运行它们。以下是我打算做的示例：

因此，有两个或多个Python代码，通常具有100行代码，但是在这里我以通用的示例

#script1.py
import numpy
        
def test1():
    print('This is script One')


if __name__ == '__main__':
    test1()

#script2.py
import numpy
        
def test2():
    print('This is script two')


if __name__ == '__main__':
    test2()

从另一个脚本main执行了一个通用示例。 py使用

#main.py
import numpy
from script1 import *
from script2 import *
    
if __name__=='__main__':
    test1()
    test2()

此main.py执行我不想要的两个。我想使用标志变量并执行我需要的任何一种。可以使用argparse命令行解析完成此操作吗？因为每个script.py在它运行中可能会有更多标志。那也是我更喜欢的东西。

谢谢

Answer 1

您可以做类似的事情：

#main.py
import numpy
import sys
from script1 import test1
from script2 import test2
    
if __name__=='__main__':
    # sys.argv[0] is the executable so we skip it.
    for arg in sys.argv[1:]   
        if arg == "test1": 
            test1()
        elif arg == "test2": 
            test2()

现在您可以执行测试

python main.py test1

python main.py test2

# or if you want to repeat a test several times or run multiple tests

# will execute test1, then test2 then test1 again
python main.py test1 test2 test1