保留Python字符串格式的帮助

2024-04-27 03:38:32 发布

您现在位置:Python中文网/ 问答频道 /正文

在寻找更快的方法来解析脚本中的命令行参数时,我遇到了argh library。在

我真的很喜欢argh的特性,但是我遇到了一个阻碍我使用它的缺点,这与我调用-help选项时显示的默认帮助消息有关: 默认情况下,函数的docstring显示在参数列表的顶部。这很好,但是初始格式丢失了。例如,请参见下面的示例脚本

import argh

def func(foo=1, bar=True):
    """Sample function.

        Parameters:
            foo: float
                An example argument.
            bar: bool
                Another argument.
    """
    print foo, bar

argh.dispatch_command(func, argv=['-h'])

这将导致以下输出

^{pr2}$

有没有一种(简单)方法可以得到如下输出?在

usage: script.py [-h] [-f FOO] [-b]

Sample function.

    Parameters:
        foo: float
            An example argument.
        bar: bool
            Another argument.

optional arguments:
  -h, --help         show this help message and exit
  -f FOO, --foo FOO
  -b, --bar

我不希望使用注释来定义参数帮助消息,因为这将要求我在每次需要更改某些内容时同时更改函数的docstring和帮助文本。在


Tags: sample方法函数脚本消息参数foobar
3条回答

感谢您对Argh图书馆的兴趣。这里讨论的解决方案将被纳入下一个版本(argh≥0.25)。另请参见issue #64(已修复)。在

我不熟悉argh,但显然它是{}的包装器。我的猜测是,它将您的函数__doc__,并使其成为解析器的description,例如

parser = argparse.ArgumentParser(description=func.__doc__)

{a1}

argparse有一个RawDescriptionHelpFormatter,按原样显示描述。在

^{pr2}$

所以问题是,有没有办法让argh使用这个格式化程序?在

argparse脚本生成所需的帮助:

import argparse

def func(foo=1, bar=True):
    """Sample function.

        Parameters:
            foo: float
                An example argument.
            bar: bool
                Another argument.
    """
    print foo, bar

parser = argparse.ArgumentParser(prog='script.py',
    description=func.__doc__,
    formatter_class=argparse.RawDescriptionHelpFormatter)
parser.add_argument('-f', '--foo', type=float)
parser.add_argument('-b', '--bar', action='store_false')
parser.print_help()

argh/dispatching.py

def dispatch_command(function, *args, **kwargs):
    ...
    parser = argparse.ArgumentParser(formatter_class=PARSER_FORMATTER)
    set_default_command(parser, function)
    dispatch(parser, *args, **kwargs)

所以你可以设置:

PARSER_FORMATTER = argparse.RawDescriptionHelpFormatter

或者编写自己的函数:

def raw_dispatch_command(function, *args, **kwargs):
    ...
    parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter)
    set_default_command(parser, function)
    dispatch(parser, *args, **kwargs)

在@hpaulj的帮助下,我终于获得了想要的行为。为了便于实现这一点,我定义了一个类似于啊。啊,目标是不必为每个参数分别编写@argh.arg(‘—param’, help=“%(default)s”),而是在我的函数上只使用一个@arg_custom()修饰符:

def arg_custom():
    from argh.constants import ATTR_ARGS
    from argh.assembling import  _get_args_from_signature, _fix_compat_issue29

    def wrapper(func):
        declared_args = getattr(func, ATTR_ARGS, [])
        for a in list(_get_args_from_signature(func)):
             declared_args.insert(0, dict(option_strings=a['option_strings'], help="(default: %(default)s)"))
        setattr(func, ATTR_ARGS, declared_args)
        _fix_compat_issue29(func)
        return func
    return wrapper

这里的关键点是for循环负责确保所有参数都得到相应的help=“%(default)s”选项。 同时更改argh/constants.py中的相应行

^{pr2}$

我们现在可以方便地使用

@arg_custom()
def func(foo=1, bar=True):
    """Sample function.

        Parameters:
            foo: float
                An example argument.
            bar: bool
                Another argument.
    """
    print foo, bar

argh.dispatch_command(func)

终于屈服了

usage: script.py [-h] [-f FOO] [-b]

Sample function.

        Parameters:
            foo: float
                An example argument.
            bar: bool
                Another argument.


optional arguments:
  -h, --help         show this help message and exit
  -f FOO, --foo FOO  (default: 1)
  -b, --bar          (default: True) 

使用-h选项执行脚本时。在

相关问题 更多 >