如何解决寻找一种在主脚本也使用 argparse 时显示导入脚本的 args 的方法
我正在尝试创建一个包装器/框架,它将导入一组 API 脚本/工具,然后将必要的参数传递给这些工具以执行不同的功能。
主脚本/包装器将使用 argparse
列出它需要的参数,但我也希望能够让用户选择他们需要的工具,然后打印出该工具需要运行的参数适当地。这甚至可能吗?
main.py:
import argparse,sys
import tools
parser = argparse.ArgumentParser(prog='API Tools',description='tools.py - wrapper API tools')
parser.add_argument('-l','--list_tools',help="list all loaded tools",action='store_true'
)
parser.add_argument('-t','--tool',help='tool to use'
)
args = parser.parse_args()
def list_tools_module():
tools = []
for module in sys.modules.keys():
if module.startswith('tools.'):
tools.append(module.split('tools.')[1])
return tools
def main():
if args.list_tools:
module_list = list_tools_module()
print("|-- Tools loaded ({}):\n| |-- "
.format(len(module_list))+'\n| |-- '.join(module_list)
)
print("------------")
sys.exit()
if args.tool:
#Print out that tool's necessary args
if __name__ == "__main__":
main()
工具位于一个目录中,并通过 __init__.py
文件从那里导入。
所以我们的想法是,用户将运行 main.py -l
,获取可用工具列表,选择带有 main.py -t tool1
的工具,然后输出将显示该特定工具所需的参数。
我更愿意让主脚本保持整洁,并添加其他工具需要的额外参数。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。