1

我正在为我的脚本创建一个命令行界面。我希望用户能够在命令行本身或配置文件中提供选项。

某些选项具有合理的默认值。我想要的逻辑是这样的：

  • 如果该选项既不在命令行也不在配置文件中，请使用默认值。
  • 如果该选项仅在配置文件中或仅在命令行中，请使用该选项。
  • 如果该选项同时在配置文件和命令行中，则显式命令行优先。

例如，这是我的文档字符串：

"""Usage: my-script.py [options] <some_arg>

Options:
  -c --config=<file>       Configuration file, if it exists. 
                           [default: settings.yaml]
  -d --delay=<delay>       Delay before scraping each message, to avoid rate
                           limiting. Delays by a gaussian distribution with
                           average <delay> and standard deviation <delay>/2.
                           [default: 1]
"""

所以，没有settings.yaml我想要的：

$ python my-script.py foo        # delay is "1"
$ python my-script.py -d 5 foo   # delay is "5"

我想要一个settings.yaml：delay: 10

$ python my-script.py foo        # delay is "10"
$ python my-script.py -d 5 foo   # delay is "5"

我遇到的问题是docopt只给了我 a{"--delay": "1"}或{"--delay": "5"}. 我无法知道它是否来自默认值。此外，我想在文档字符串中指定默认值- 从用户的角度来看，这要好得多。

有没有什么好的docopt-y，Pythonic方式来完成这个？我正在考虑根据默认参数检查解析的参数，但是如果它碰巧与参数值匹配，用户将无法使用命令行参数覆盖配置文件参数。

这是我加载配置文件的方式：

args = docopt(__doc__, version='Yahoo! Groups Backup-er 0.1',
              options_first=True)
cfg_args = {}
if args['--config'] != 'settings.yaml' and not os.path.exists(args['--config']):
    sys.exit("Specified config file '%s' does not exist." % args['--config'])

if os.path.exists(args['--config']):
    settings = yaml.load(open(args['--config']))
    command_line_args = args
    args = {}
    for key, val in settings.items():
        args['--%s' % key] = val
    args.update(command_line_args)
4

1 回答 1

1

是的，要检查的语法是：

args = docopt(__doc__)
if args["--flag"]:
  print "With --flag\n"

请阅读这篇文章了解详情。

于 2019-10-08T00:36:59.393 回答