python的argpare和click模块详解

一、argparse模块

1、模块说明

# argparse是python的标准库中用来解析命令行参数的模块，用来替代已经过时的optparse模块，argparse能够根据程序中的定义的sys.argv中解析出这些参数，
# 并自动生成帮助和使用信息

　　

2、模块常用的参数

    # 参数说明：
    # name/flag:参数的名字
    # action:遇到参数的动作，默认值是store
    # nargs:参数的个数，可以是具体的数字，或者是+或者是*,*表示0个或者多个参数，+号表示1个或者多个参数
    # default:不指定参数时的默认值
    # type:# 参数的类型
    # choice:参数允许的值
    # required:可选参数是否可以省略
    # help:参数的帮助信息
    # dest:解析后参数的名称

　　

3、使用方法

import argparse

def _argparse():
    parseobj = argparse.ArgumentParser(description="This is script help")
    # 参数说明：
    # name/flag:参数的名字
    # action:遇到参数的动作，默认值是store
    # nargs:参数的个数，可以是具体的数字，或者是+或者是*,*表示0个或者多个参数，+号表示1个或者多个参数
    # default:不指定参数时的默认值
    # type:# 参数的类型
    # choice:参数允许的值
    # required:可选参数是否可以省略
    # help:参数的帮助信息
    # dest:解析后参数的名称

    parseobj.add_argument("--host",action='store',dest='host',required=True,default="127",help="This is a host ip address",type=int)
    parseobj.add_argument("--P",'--passwd',action='store', dest='pwd', required=True, default="admin123.",help="This is a host password", type=str)
    parseobj.add_argument("--V", '--version', action='version', version="%(prog)s 0.1")

    return parseobj.parse_args()


if __name__ == '__main__':
    res = _argparse()
    print(res.pwd)
    print(res.host)

　　

这里还可以加一个choices的选项，限制参数为指定的参数

parseobj.add_argument("--type", action='store', dest='t', required=True,
                      help="type", type=str,choices=["update","create"])

 

这里我们在试一下nargs参数，切记这里的type也是str，不能是list，这里的type的意思是每个参数的类型

    parseobj.add_argument("--index", action='store', dest='index', required=True,
                          help="index names", type=str,nargs="+")

 

 

 

 

 

 4、最后我们测试一下这个模块

a、测试 -h选项，这里-h和--help的效果是一样的

 

 b、测试--V选项和--version选项

 

c、测试一下输入的正确的参数

 

二、click模块

1、模块介绍

click模块的作者就是Flask的作者,(Armin Ronacher)开发的一个第三方的模块，用于快速创建命令行。他的作用用python标准库中的argparse相同，但是
使用更加简单，click相对于标准库的argparse，就好比requests库相当于标准库的urllib库，click是一个第三的库，因此在使用之前需要安装

　　

2、模块安装

E:\python3\Scripts>pip3.6.exe install click

　　

3、使用步骤

a、使用@click.command()装饰一个函数，使之成为命令行的接口

b、使用@click.option()等装饰函数，为其添加命令行选项等

c、先看一个官方的例子

import click

# click模块的作者就是Flask的作者,(Armin Ronacher)开发的一个第三方的模块，用于快速创建命令行。他的作用用python标准库中的argparse相同，但是
# 使用更加简单，click相对于标准库的argparse，就好比requests库相当于标准库的urllib库，click是一个第三的库，因此在使用之前需要安装

@click.command()
@click.option('--count',default=1,help='Number of greetings')
@click.option('--name',prompt='your name',help='The person to greet')
def hello(count,name):
    for x in range(count):
        click.echo("hello {name}".format(name = name))

if __name__ == '__main__':
    hello()

　　

 

 

其他的应该大家都可以看懂，这个prompt的作用是什么呢，实际就是如果我们没有为name传参数，他就会给出一个提示

下面这个例子是完整的传参

 

 4、常用参数

常用参数
default：设置命令行参数的默认值
help：参数说明
type：参数的类型，可以是string，int，float
prompt：当在命令行中没有输入相应的参数，会根据prompt提示用户输入
nargs：指定命令行参数接受的值的个数

　　

a、测试一下nargs参数

@click.command()
@click.option('--post',nargs=2,help='Number of post')
def hello(post):
    print(post)


if __name__ == '__main__':
    hello()

　　

测试结果

 

 b、测试click.choice选项

@click.command()
@click.option('--hash',type=click.Choice(["md5","sha1"]),help='type of hash')
def hello(hash):
    print(hash)


if __name__ == '__main__':
    hello()

　　

测试结果

 

 c、如果使用命令行输入密码，则默认的情况是有很大的安全隐患的，因为输入密码的命令在history中，其他用户就可以通过命令的历史列表，拿到我们的密码，click可以为我们解决这个问题

@click.command()
@click.option('--pwd',prompt=True,hide_input=True,help='passwd of user',confirmation_prompt=True)
def hello(pwd):
    print(pwd)


if __name__ == '__main__':
    hello()

　　

prompt：要设置为True
hide_input：要设置为True
confirmation_prompt：会自动为我们进行密码的二次验证

　　

测试结果如下

错误的输入

正确的输入