python - 如何在 Python 3 Click 应用程序中遍历 shell 提供的文件列表？

Question

我正在使用 Click 库(几乎没有使用经验)和 Python 3.6 编写控制台应用程序。

我需要一个可以从 OS shell 调用的程序:

myapp --myoptiona=123 --myoptionb=456 *

遍历当前目录(或掩码指定的任何内容)中所有文件的列表，并对它们执行某些操作。

如何通过 Click 实现这一目标？

Answer 1

如果您需要接受可变数量的文件参数，您可以使用 nargs=-1 和 click.argument 来生成如下命令:

代码:

@click.command()
@click.option('--myoptiona')
@click.option('--myoptionb')
@click.argument('files', nargs=-1)
def cli(myoptiona, myoptionb, files):
    """The Great CLI APP"""
    filenames = []
    for filename in files:
        # if our shell does not do filename globbing
        expanded = list(glob(filename))
        if len(expanded) == 0 and '*' not in filename:
            raise(click.BadParameter('{}: file not found'.format(filename)))
        filenames.extend(expanded)

    click.echo('myoptiona: %s' % myoptiona)
    click.echo('myoptionb: %s' % myoptionb)
    for name in filenames:
        click.echo('a file: %s' % name)

测试代码:

import click
from glob import glob

@click.command()
@click.option('--myoptiona')
@click.option('--myoptionb')
@click.argument('files', nargs=-1)
def cli(myoptiona, myoptionb, files):
    """The Great CLI APP"""
    filenames = []
    for filename in files:
        expanded = list(glob(filename))
        if len(expanded) == 0 and '*' not in filename:
            raise(click.BadParameter(
                "file '{}' not found".format(filename)))
        filenames.extend(expanded)

    click.echo('myoptiona: %s' % myoptiona)
    click.echo('myoptionb: %s' % myoptionb)
    for name in filenames:
        click.echo('a file: %s' % name)


if __name__ == "__main__":
    commands = (
        '--myoptiona 3 --myoptionb 4',
        '--myoptiona 3 file1 file2',
        '--myoptiona 3 --myoptionb 4 file1 file4',
        'file1 file2 file3',
        'file*',
        '--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)
            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)]
-----------
> --myoptiona 3 --myoptionb 4
myoptiona: 3
myoptionb: 4
-----------
> --myoptiona 3 file1 file2
myoptiona: 3
myoptionb: None
a file: file1
a file: file2
-----------
> --myoptiona 3 --myoptionb 4 file1 file4
Usage: test.py [OPTIONS] [FILES]...

Error: Invalid value: 'file4': file not found
-----------
> file1 file2 file3
myoptiona: None
myoptionb: None
a file: file1
a file: file2
a file: file3
-----------
> file*
myoptiona: None
myoptionb: None
a file: file.conf
a file: file1
a file: file2
a file: file3
-----------
> --help
Usage: test.py [OPTIONS] [FILES]...

  The Great CLI APP

Options:
  --myoptiona TEXT
  --myoptionb TEXT
  --help            Show this message and exit.