我创建了一个Click命令,该命令会将文件从源复制到目标
该命令接受3个参数:
1-文件源
2-文件目的地
3-传输模式(本地,ftp)
import click
@click.group()
def cli():
pass
@cli.command()
@click.argument('source')
@click.argument('destination')
@click.option('--mode', required = True)
def copy(source, destination, mode):
print("copying files from " + source + " to " + destination + "using " + mode + " mode")
if __name__ == '__main__':
cli()
当我使用以下命令调用脚本时:command.py copy "C:/" "D:/" --mode=network
我得到以下输出:copying files from C:/ to D:/using network mode
您可以看到我将网络指定为模式,但是我只需要两个选项:本地或ftp
那么我该如何使用“点击”设置选项的允许值?
答案 0 :(得分:1)
您要使用click.Choice
@cli.command()
@click.argument('source')
@click.argument('destination')
@click.option('--mode', type=click.Choice(['local', 'ftp']), required = True)
def copy(source, destination, mode):
print("copying files from " + source + " to " + destination + "using " + mode + " mode")