Python 如何在单击中为组操作创建异常
我有一种情况,我想为大多数单击命令运行一个公共函数(Python 如何在单击中为组操作创建异常,python,python-click,Python,Python Click,我有一种情况,我想为大多数单击命令运行一个公共函数(check\u upgrade()),但也有一些情况我不想运行它。我宁愿在默认情况下运行它,然后为不应该运行check\u upgrade()的命令添加一个修饰符(例如,@bypass\u upgrade\u check) 我希望有这样的事情: class State(object): def __init__(self): self.bypass_upgrade_check = False pass_state =
check\u upgrade()
),但也有一些情况我不想运行它。我宁愿在默认情况下运行它,然后为不应该运行check\u upgrade()
的命令添加一个修饰符(例如,@bypass\u upgrade\u check
)
我希望有这样的事情:
class State(object):
def __init__(self):
self.bypass_upgrade_check = False
pass_state = click.make_pass_decorator(State, ensure=True)
def bypass_upgrade_check(func):
@pass_state
def wrapper(state, *args, **kwargs):
state.bypass_upgrade_check = True
func(*args, **kwargs)
return wrapper
@click.group()
@pass_state
def common(state):
if not state.bypass_upgrade_check:
check_upgrade()
@common.command()
def cmd1():
# check_upgrade() runs here
pass
@bypass_upgrade_check
@common.command()
def cmd2():
# don't run check_upgrade() here
pass
但这不起作用。它实际上从未调用旁路升级检查()函数
有没有一种方法可以在组码运行之前修改命令的状态?或者另一种方法可以实现这一点?为了跟踪哪些命令绕过了升级检查,我建议您在绕过标记装饰器中将该状态存储在click.Command
对象上。然后,如果将click.Context
传递给您的组,您可以查看命令对象,查看它是否标记为允许跳过升级,如:
代码:
测试代码:
结果:
您是否意识到,就在解释器处理cmd2
的函数定义之后,decoratorpass\u upgrade\u check
将只运行一次?它不会生成您在此时可以看到的任何输出,也不会调用原始的cmd2
。不清楚你想在这里做什么。您似乎希望单击“默认”执行某些特定函数,但不希望未来事件处理程序的“开发人员”忘记调用该函数。但是你没有告诉我们你的环境是什么,也没有告诉我们开发人员是如何添加处理程序的。这非常有效!我不知道你能拉入被调用的u子命令。如此简单优雅,正是我所需要的。谢谢这个答案适用于顶级命令,但不适用于子命令。假设我有子命令“cmd1 sub1”和“cmd1 sub2”,并且我想禁用对sub1的检查,但不禁用对sub2的检查?我尝试将getattr检查添加到子命令的组定义中,但这不起作用,因为顶级组将首先捕获它,并且由于“cmd1”没有@bypass\u upgrade\u检查,因此它会运行升级检查。我认为需要有一种从顶层搜索子命令的方法。或许还有其他方法可以做到这一点?新问题发布:
def bypass_upgrade_check(func):
setattr(func, 'do_upgrade_check', False)
@click.group()
@click.pass_context
def cli(ctx):
sub_cmd = ctx.command.commands[ctx.invoked_subcommand]
if getattr(sub_cmd, 'do_upgrade_check', True):
check_upgrade()
import click
def check_upgrade():
click.echo('Checking Upgrade!')
def bypass_upgrade_check(func):
setattr(func, 'do_upgrade_check', False)
@click.group()
@click.pass_context
def cli(ctx):
sub_cmd = ctx.command.commands[ctx.invoked_subcommand]
if getattr(sub_cmd, 'do_upgrade_check', True):
check_upgrade()
@cli.command()
def cmd1():
# check_upgrade() runs here
click.echo('cmd1')
@bypass_upgrade_check
@cli.command()
def cmd2():
# don't run check_upgrade() here
click.echo('cmd2')
if __name__ == "__main__":
commands = (
'cmd1',
'cmd2',
)
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)
cli(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)]
-----------
> cmd1
Checking Upgrade!
cmd1
-----------
> cmd2
cmd2