49

我想使用一些有用的功能作为命令。为此，我正在测试click图书馆。我定义了我的三个原始函数，然后装饰为click.command：

import click
import os, sys

@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=True)
def add_name(content, to_stdout=False):
    if not content:
        content = ''.join(sys.stdin.readlines())
    result = content + "\n\tadded name"
    if to_stdout is True:
        sys.stdout.writelines(result)
    return result


@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=True)
def add_surname(content, to_stdout=False):
    if not content:
        content = ''.join(sys.stdin.readlines())
    result = content + "\n\tadded surname"
    if to_stdout is True:
        sys.stdout.writelines(result)
    return result

@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=False)
def add_name_and_surname(content, to_stdout=False):
    result = add_surname(add_name(content))
    if to_stdout is True:
        sys.stdout.writelines(result)
    return result

这样我就可以生成三个命令add_name，add_surname并add_name_and_surname使用一个setup.py文件，pip install --editable .然后我可以管道：

$ echo "original content" | add_name | add_surname 
original content

    added name
    added surname

但是，当使用不同的单击命令作为函数编写时，我需要解决一个小问题：

$echo "original content" | add_name_and_surname 
Usage: add_name_and_surname [OPTIONS] [CONTENT]

Error: Got unexpected extra arguments (r i g i n a l   c o n t e n t 
)

我不知道为什么它不起作用，我需要这个add_name_and_surname命令来调用add_name而add_surname不是作为命令而是作为函数调用，否则它违背了我将函数用作库函数和命令的最初目的。

4

4 回答 4

53

当您直接从另一个函数调用add_name()时add_surname()，您实际上调用了它们的修饰版本，因此预期的参数可能与您定义的不同（有关原因的一些详细信息，请参见How to strip decorators from a function in python的答案）。

我建议修改您的实现，以便您保持原始功能未修饰并为它们创建瘦点击特定包装器，例如：

def add_name(content, to_stdout=False):
    if not content:
        content = ''.join(sys.stdin.readlines())
    result = content + "\n\tadded name"
    if to_stdout is True:
        sys.stdout.writelines(result)
    return result

@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=True)
def add_name_command(content, to_stdout=False):
    return add_name(content, to_stdout)

然后，您可以直接调用这些函数，也可以通过 setup.py 创建的 CLI 包装脚本调用它们。

这似乎是多余的，但实际上可能是正确的方法：一个函数代表您的业务逻辑，另一个（单击命令）是通过命令行公开此逻辑的“控制器”（为了例如，还有一个通过 Web 服务公开相同逻辑的函数）。

事实上，我什至建议将它们放在单独的 Python 模块中 - 您的“核心”逻辑和特定于单击的实现，如果需要，可以将其替换为任何其他接口。

于 2016-10-17T19:43:05.170 回答
50

由于点击装饰器，函数不能再通过指定参数来调用。Context类是你的朋友，特别是：

  1. Context.invoke() - 使用您提供的参数调用另一个命令
  2. Context.forward() - 填充当前命令的参数

因此，您的 add_name_and_surname 代码应如下所示：

@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=False)
@click.pass_context
def add_name_and_surname(ctx, content, to_stdout=False):
    result = ctx.invoke(add_surname, content=ctx.forward(add_name))
    if to_stdout is True:
        sys.stdout.writelines(result)
    return result

参考： http ://click.pocoo.org/6/advanced/#invoking-other-commands

于 2016-10-17T23:00:32.780 回答
2

我发现这些解决方案更复杂。我希望从另一个包的另一个地方调用下面的这个函数：

@click.command(help='Clean up')
@click.argument('path', nargs=1, default='def')
@click.option('--info', '-i', is_flag=True,
              help='some info1')
@click.option('--total', '-t', is_flag=True,
              help='some info2')
def clean(path, info, total):
#some definition, some actions

#this function will help us
def get_function(function_name):
if function_name == 'clean':
    return clean

我有另一个包，所以，我想在这个包中点击上面的命令

import somepackage1 #here is clean up click command
from click.testing import CliRunner


@check.command(context_settings=dict(
           ignore_unknown_options=True,
          ))
@click.argument('args', nargs=-1)
@click.pass_context
def check(ctx, args):
    runner = CliRunner()

    if len(args[0]) == 0:
        logger.error('Put name of a command')

    if len(args) > 0:
        result = runner.invoke(somepackage1.get_function(args[0]), args[1:])
        logger.print(result.output)
    else:
        result = runner.invoke(somepackage1.get_function(args[0]))
    logger.print(result.output)

所以，它有效。

python somepackage2 check clean params1 --info
于 2020-01-13T10:44:54.313 回答
2

callback在某些情况下，您还可以使用成员函数调用 Click 函数。根据这个 GitHub 问题：

假设您知道给定命令是您编写的函数的直接包装器（而不是组或其他类型的命令），您可以使用 command.callback 获取该函数。但是，调用它只会调用函数，它不会调用任何 Click 管道进行验证、回调等。

您的示例将变为：

(...)
def add_name_and_surname(content, to_stdout=False):
    result = add_surname.callback(add_name.callback(content))
    (...)
于 2021-11-05T14:47:37.763 回答