我正在为我的脚本创建一个命令行界面。我希望用户能够在命令行本身或配置文件中提供选项。
某些选项具有合理的默认值。我想要的逻辑是:
例如,这是我的文档字符串:
"""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
只给了{"--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)