python - 如何不退出单击 CLI？

Question

我正在编写一个 python 脚本，它的行为应该像一个典型的 shell 并提供一些自写函数。
它已经工作得很好，但它总是在成功执行命令后退出，因此必须再次启动它才能执行第二个任务。
我怎样才能做到，所以它不会以退出代码 0 结束，而是返回到 shell 等待新输入？那么我将如何实现退出方法？
以下示例总是在输入 print-a 或 print-b 后退出:

import click
import click_repl
from prompt_toolkit.history import FileHistory
import os

@click.group(invoke_without_command=True)
@click.pass_context
def cli(ctx):
     if ctx.invoked_subcommand is None:
         ctx.invoke(repl)

@cli.command()
def print_a():
    print("a")


@cli.command()
def print_b():
    print("b")


@cli.command()
def repl():
    prompt_kwargs = {
        'history': FileHistory(os.path.expanduser('~/.repl_history'))
    }
    click_repl.repl(click.get_current_context(), prompt_kwargs)

def main():
    while True:
        cli(obj={})


if __name__ == "__main__":
    main()

(还有一个额外的问题:在 cmd 包中可以自定义 > 提示标签，这是否可以通过单击来实现？所以它类似于 App>？)

Answer 1

使用 standalone_mode论证，试试这个:

rv = cli(obj={}, standalone_mode=False)

当解析失败时，上面的代码会抛出 UsageError .当 --help已通过， rv将是整数 0 .在大多数其他情况下，会返回处理命令的函数的返回值，尽管有一堆异常并且行为通常非常复杂，更多解释在这里:
https://click.palletsprojects.com/en/master/commands/#command-return-values

这种方法的优点是您可以使用命令处理程序的返回值。缺点是当解析失败时你会丢失漂亮的帮助信息(也许有办法恢复它？)。

另一种选择是不使用 standalone_mode而是将您的电话转至 cli在 try/except阻止您捕获 SystemExit :

try:
    cli(obj={})
except SystemExit as e:
    if e.code != 0:
        raise

通过捕捉 SystemExit您可以停止单击启动的程序退出过程。如果命令解析成功，则 SystemExit(0)被捕获了。再次注意解析 --help也算作“成功”解析，因此也返回 SystemExit(0) .

这种方法的缺点是不能使用命令处理程序的返回值，这使得更难知道何时 --help通过了。好处是所有到控制台的帮助消息都被恢复了。

我还应该注意到 SystemExit继承自 BaseException但不是来自 Exception .所以要真正捕获 SystemExit你可以直接捕获它，也可以捕获它 BaseException .