python – argparse augment子命令默认通过全局选项

前端之家收集整理的这篇文章主要介绍了python – argparse augment子命令默认通过全局选项前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我希望能够使用argparse.ArgumentParser对象中的全局选项来覆盖/扩充子命令的默认值.

一个例子是显示的帮助反映了全局更新,即以下玩具示例:

import argparse
import os
import sys

# Global parser and options.
parser = argparse.ArgumentParser(
    formatter_class=argparse.ArgumentDefaultsHelpFormatter)

parser.add_argument("--user",dest="user",default=os.environ.get("USER"),help="Override the setting of the $USER variable.")

# Sub-command parser and options.
subparsers = parser.add_subparsers()

command = subparsers.add_parser(
    "command",formatter_class=argparse.ArgumentDefaultsHelpFormatter)

command.add_argument("--config",dest="config",default="~%s/config" % os.environ.get("USER"),help="The config file.")

options = parser.parse_args()

理想情况下,当我在帮助模式下运行时,我会得到,

> python example.py --user thing command --help
usage: example.py command [-h] [--config CONFIG]

optional arguments:
  -h,--help       show this help message and exit
  --config CONFIG  The config file. (default: ~thing/config)

即,配置文件路径是用户特定的(事物).我意识到我可以将默认配置更改为“〜%(user)s / config”,然后在运行时使用options命名空间解决此问题,但我希望帮助更明确.

我收集一个替代解决方案是尝试解析一次参数以获得全局选项,即

if "--help" in sys.argv:

    # Parse the command minus the help to obtain the global options. 
    args = sys.argv[1:]
    args.remove("--help")

    # Update the defaults with the global options.
    options = parser.parse_args(args) 
    command.set_defaults(config="~%s/config" % options.user)

    # Re-parse the options.
    parser.parse_args()

虽然这看起来有点hacky.有更好的方法/解决方案吗?

最佳答案
定义全局选项后,但在定义子命令之前,请调用parse_known_args以找出–user的值.然后,完成定义subparser命令,使用–user值定义之前的–config的默认值
调用parse_args来解析命令行上的所有选项.

这与您的替代方案略有不同,但它将所有命令行处理保留在argparse对象中.

(我把你的代码缩减了一点,只是为了缩短它的答案.)

import os
import argparse

# Global preparser and options.
preparser = argparse.ArgumentParser(add_help=False)
preparser.add_argument("--user",default=os.environ.get("USER"))

# ****** NEW *******
options,_ = preparser.parse_known_args() # Ignore what we haven't defined yet
user = options.user                       # Use this to define the default of --config

parser = argparse.ArgumentParser(parents=[ preparser ],add_help=True,formatter_class=argparse.ArgumentDefaultsHelpFormatter)

# Sub-command parser and options.
subparsers = parser.add_subparsers()

command = subparsers.add_parser("command")

# ****** MODIFIED *******
command.add_argument("--config",default="~%s/config" % (user,))

options = parser.parse_args()
原文链接:https://www.f2er.com/python/439357.html

猜你在找的Python相关文章