Python 为click.options()创建包装decorator

发布于09月25日

我正在try 为点击decorator @click.options('--foo', required=True)创建一个包装decorator :

import click

def foo_option(func):
    orig_decorator = click.option('--foo', required=True)(func)
    def decorator(*args, **kwargs):
        orig_decorator(*args, **kwargs)
    return decorator

@click.command()
@foo_option
def bar1(foo: str) -> None:
    print(f"bar1: {foo}")

if __name__ == '__main__':
    bar1()

这不管用.当我以script.py --foo=1的速度运行它时，我得到错误:

Usage: script.py [OPTIONS]
Try 'script.py --help' for help.

Error: No such option: --foo

预期输出应为:

bar1: 1

我遗漏了什么？有关背景信息，请参阅Refactor @click.option() arguments.

推荐答案

问题在于你是如何构建这个装饰者的.在本例中，您使用的是click.选项，这是不必要的.当您创建一个修饰符来包裹另一个修饰符(如click.ption)时，您应该考虑修饰符是自下而上工作的.以下是更新后的代码:

import click


def foo_option(func):
    return click.option('--foo', required=True)(func)


@click.command()
@foo_option
def bar1(foo: str) -> None:
    print(f"bar1: {foo}")


if __name__ == '__main__':
    bar1()

当您运行该脚本时，您将获得预期的输出.