Python:“导入”输入参数

2024-04-30 17:45:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个python脚本A和{},它们都有一个main和不同的输入参数(我使用argparse)。我通常用以下方式启动它们:

$ python A.py --paramA1 valA1 --paramA2 valA2
$ python B.py --paramB1 valB1 --paramB2 valB2

我希望能够从A内调用B,并在启动A时指定B的参数。简而言之,我想用A的参数和^{的参数来调用A。从外表看应该是这样的:

^{pr2}$

有没有比复制并粘贴Bargparse代码到A然后在它们上调用B更优雅的方法吗?在

编辑:为了简化操作,下面是一些示例代码:

A.py:

import argparse
def main():
    parser = argparse.ArgumentParser(description="Args for A.py")
    parser.add_argument("--param1A", type=int)
    parser.add_argument("--param2A", type=int)
    args = parser.parse_args()
    valA1 = args.param1A
    valA2 = args.param2A
    ...
    return 0

备份:

import argparse
def main():
    parser = argparse.ArgumentParser(description="Args for B.py")
    parser.add_argument("--param1B", type=int)
    parser.add_argument("--param2B", type=int)
    args = parser.parse_args()
    valA1 = args.param1B
    valA2 = args.param2B
    ...
    return 0

建议的C.py,它包括A.py和{}的参数,它到底是什么样子的?那么A.py和{}会是什么样子呢?在

编辑2:我忘了要在A.py中创建B.py的一个参数,因此执行顺序是A,然后B,并且{}必须能够向B传递一个参数。在


Tags: 代码pyimportaddparser编辑参数main
2条回答

您可以创建一个模块C.py,它包含通用的argparse逻辑,并由a和B导入

从A调用B时,也可以将已解析的参数传递给B。我的意思是,A和B都应该有类似于main函数的东西,它接受已经解析的参数,因此不必进行两次解析。在

C.py公司

import argparse

def create_parser():
    parser = argparse.ArgumentParser(description="Args for B.py")
    parser.add_argument(" param1B", type=int)
    parser.add_argument(" param2B", type=int)
    return parser

B.py公司

^{pr2}$

A.py

import argparse
import sys
import C
import B

def main(args):
    valA1 = args.param1A
    valA2 = args.param2A
    # do stuff
    return 0


if __name__ == '__main__':
    parser = C.create_parser()
    parser.add_argument(" param1A", type=int)
    parser.add_argument(" param2A", type=int)
    args = parser.parse_args()
    B.main(args)
    sys.exit(main(args))

我想@Ioan Alexandru Cucu所描述的内容如下:

import argparse

class A():
    __init__(self, param1A, param2A, param1B=None, param2B=None):
        if(param1B and param2B):
            myB = B(param1B, param2B)
        #Do processing here ...
        return 0


class B():
    __init__(self, param1B, param2B):
        #Do processing here ...
        return 0

if(__name__ == '__main__'):
    parser = argparse.ArgumentParser(description="Args for A and/or B")

    parser.add_argument(" param1A", type=int)
    parser.add_argument(" param2A", type=int)

    parser.add_argument(" param1B", type=int)
    parser.add_argument(" param2B", type=int)

    args = parser.parse_args()

    if(args.param1A and args.param2A and args.param1B and args.param2B):
        A(args.param1A, args.param2A, args.param1B, args.param2B)
    if(args.param1A and args.param2A):
        A(args.param1A, args.param2A)
    elif(args.param1B and args.param2B):
        B(args.param1B, args.param2B)

而不是在一个模块中有多个类,而是可以有多个具有不同类的模块。在

相关问题 更多 >