时间戳

首页 > 解决方案 > 使用预先生成的输入自动化 python argparse 脚本

python-3.x - 使用预先生成的输入自动化 python argparse 脚本

问题描述

我有一个程序，它通过带有 argparse 的命令行获取文件夹路径和其他输入。我希望这个脚本在服务器上自动运行，但我也想保留它的 argparse 功能，以防我想手动运行脚本。有没有办法让脚本使用文件中预先生成的输入，但也可以使用 argparse 保留其基于标志的输入系统？这是我当前的实现：

parser = argparse.ArgumentParser(description='runs batch workflow on root directory')
parser.add_argument("--root", type=str, default='./', help="the path to the root directory 
                    to process")
parser.add_argument("--data", type=str, default='MS', help="The type of data to calculate ")
args = parser.parse_args()

root_dir = args.root
option = args.data

我对这些东西很陌生，阅读 argparse 文档和这个堆栈溢出问题并不是我真正想要的，如果可能的话，我想保留根和数据标志，而不仅仅是用输入文件或标准输入替换它们.

标签： python-3.xstdinargparse

解决方案

如果使用argparse，default关键字参数是解决问题的一种很好的标准方法；将程序的默认行为嵌入脚本源中，而不是外部配置文件。但是，如果您有多个想要以不同方式部署的配置文件，那么您提到的方法（从输入预先生成）是可取的。

argparse到字典

argparse命名空间可以转换为字典。这很方便，因为我们可以创建一个接受字典或关键字参数的函数，并让它使用方便的函数签名来处理程序。此外，文件解析器可以很容易地加载字典并与相同的函数进行交互。以pythonjson模块为例。当然，也可以使用其他的。

示例 Python

def main(arg1=None, arg2=None, arg3=None):
    print(f"{arg1}, {arg2}, {arg3}")

if __name__ == "__main__":
    import sys
    import json
    import argparse

    # script called with nothing -- load default
    if len(sys.argv) == 1:
        with open("default.json", "r") as dfp:
            conf = json.load(dfp)
        main(**conf)

    else: # parse arguments
        parser = argparse.ArgumentParser()
        parser.add_argument('-a1', dest='arg1', metavar='arg1', type=str)
        parser.add_argument('-a2', dest='arg2', metavar='arg2', type=str)
        parser.add_argument('-a3', dest='arg3', metavar='arg3', type=str)
        args = parser.parse_args()
        conf = vars(args)
        main(**conf)

默认.json

{
    "arg1" : "str1",
    "arg2" : "str2",
    "arg3" : "str3"
}

使用Fire

pythonFire模块也可以更方便地使用。它具有多种模式，可以轻松地与文件进行交互。github 存储库可在此处获得。

推荐阅读