python - 单击带有可选参数的工具

Question

我想编写一个以 FILENAME 作为参数的 CLI hello，除非给出了选项 -s STRING，在这种情况下直接处理 STRING。程序应该打印 hello {NAME}其中 name 要么通过 STRING 给出，要么被视为文件 FILENAME 的内容。
例子:

$ cat myname.txt
john

$ hello myname.txt
hello john

$ hello -s paul
hello paul

一种可能的解决方法是使用两个选项:

@click.command()
@click.option("-f", "--filename", type=str)
@click.option("-s", "--string", type=str)
def main(filename: str, string: str):
    if filename:
        ...
    if string:
        ....

但这意味着我必须调用 hello有一面旗帜。

Answer 1

您可以使用点击 argument并使其不需要，例如:

import click
@click.command()
@click.argument('filename', required=False)
@click.option("-s", "--string")
def main(filename, string):
    if None not in (filename, string):
        raise click.ClickException('filename argument and option string are mutually exclusive!')
    click.echo(f'Filename: {filename}')
    click.echo(f'String: {string}')