python - 使用 Python click 命令调用具有可变参数的类方法
问题描述
我有一个用以前未知数量的参数初始化的类,我希望它使用 Python 的click
包在 CLI 上完成。我的问题是我无法初始化它并运行click
命令:
$ python mycode.py arg1 arg2 ... argN click_command
设置定义数量的参数,如nargs=5
,解决了缺少命令的问题,但要求我在命令之前输入 5 个参数。使用可变参数,如nargs=-1
,click
不能识别click_command
为命令。
如何输入 n 个参数,然后使用 运行命令click
?
import click
class Foo(object):
def __init__(self, *args):
self.args = args
def log(self):
print('self.args:', self.args)
pass_foo = click.make_pass_decorator(Foo)
@click.group()
@click.argument('myargs', nargs=-1)
@click.pass_context
def main(ctx, myargs):
ctx.obj = Foo(myargs)
print("arguments: ", myargs)
@main.command()
@pass_foo
def log(foo):
foo.log()
main()
click
我希望在将 n-many args 传递给我的类后能够运行命令Foo()
,因此我可以对其进行初始化并将其log()
方法作为 CLI 命令运行,但输出为:
错误:缺少命令
解决方案
我不完全确定您要尝试做的是解决此问题的最佳方法。我认为在命令之后放置可变参数会更合乎逻辑,并且肯定会更符合 click 的工作方式。但是,你可以用这个来做你想做的事:
自定义类:
class CommandAfterArgs(click.Group):
def parse_args(self, ctx, args):
parsed_args = super(CommandAfterArgs, self).parse_args(ctx, args)
possible_command = ctx.params['myargs'][-1]
if possible_command in self.commands:
ctx.protected_args = [possible_command]
ctx.params['myargs'] = ctx.params['myargs'][:-1]
elif possible_command in ('-h', '--help'):
if len(ctx.params['myargs']) > 1 and \
ctx.params['myargs'][-2] in self.commands:
ctx.protected_args = [ctx.params['myargs'][-2]]
parsed_args = ['--help']
ctx.params['myargs'] = ctx.params['myargs'][:-2]
ctx.args = [possible_command]
return parsed_args
使用自定义类:
然后使用自定义类,将其作为cls
参数传递给组装饰器,如:
@click.group(cls=CommandAfterArgs)
@click.argument('myargs', nargs=-1)
def main(myargs):
...
测试代码:
import click
class Foo(object):
def __init__(self, *args):
self.args = args
def log(self):
print('self.args:', self.args)
pass_foo = click.make_pass_decorator(Foo)
@click.group(cls=CommandAfterArgs)
@click.argument('myargs', nargs=-1)
@click.pass_context
def main(ctx, myargs):
ctx.obj = Foo(*myargs)
print("arguments: ", myargs)
@main.command()
@pass_foo
def log(foo):
foo.log()
if __name__ == "__main__":
commands = (
'arg1 arg2 log',
'log --help',
'--help',
)
import sys, time
time.sleep(1)
print('Click Version: {}'.format(click.__version__))
print('Python Version: {}'.format(sys.version))
for cmd in commands:
try:
time.sleep(0.1)
print('-----------')
print('> ' + cmd)
time.sleep(0.1)
main(cmd.split())
except BaseException as exc:
if str(exc) != '0' and \
not isinstance(exc, (click.ClickException, SystemExit)):
raise
结果:
Click Version: 6.7
Python Version: 3.6.3 (v3.6.3:2c5fed8, Oct 3 2017, 18:11:49) [MSC v.1900 64 bit (AMD64)]
-----------
> arg1 arg2 log
arguments: ('arg1', 'arg2')
self.args: ('arg1', 'arg2')
-----------
> log --help
arguments: ()
Usage: test.py log [OPTIONS]
Options:
--help Show this message and exit.
-----------
> --help
Usage: test.py [OPTIONS] [MYARGS]... COMMAND [ARGS]...
Options:
--help Show this message and exit.
Commands:
log
推荐阅读
- javascript - 试图切换 dd 元素的可见性
- javascript - 类中的 PHP 函数返回空值
- html - alt="" 在技术上不可行的空 alt 属性的最佳实践
- excel - 如果单元格字符串包含列中列表中的单词,则将值返回到新单元格
- json - 从 http.get 解组嵌套的 json
- c# - 在包含所述接口类型成员的类上定义接口类型的正确方法
- graphql-java - 读取通过 GraphQL Java 中的变量传递的字段的参数
- javascript - 在 JavaScript 中,将类名分配给常规元素和命名空间元素是否不同?
- javascript - eval 可以用来声明多个全局类吗?
- python - 沿每个轴应用 DCT 矩阵未给出所需结果