我们知道在Linux下有优雅的shell终端命令行界面,shell脚本都可以优雅用命令行的方式来运行。而且shell也再带优化命令行参数解析的bash内部命令getopts和大多数发行版附带的外部命令getops。Perl语言也有Getopt::XX系列模块来实现类似功能;Golang也有flag标准库以及更加强大的第三库cobra。
说了这么多你肯定要问,那么Python呢?Python不像shell,perl一样有便捷强大的oneline脚本,虽然Python有强大的web界面下的脚本执行环境Jupyter Note,但是对终端命令行下的执行也是我们要追求的目标。本文虫虫就给大家介绍如何在Python下实现优雅的终端命令行界面CLI,涉及的模块有Click和argparse。
Click
Python开始的时候也模仿bansh和其他语言有getopt,optparse等模块来做CLI,后来推出argparser模块后getopt,optparse等都已经被废弃。在介绍argparser模块之前我们先来介绍click。
首先最简单的方式是使用click。我们先来看下python 3中的print()函数,注意该print()强制必须以带括号方式, 这是和python2比较大的区别。其定义如下:
- print(*objects, sep=' ', end='\n', file=sys.stdout, flush=False)
可以看到该函数可以支持各种规范格式的输出,而不单单一个打印输出了。
比如,通过可以自定义打印的结束字符,默认是"\n"。
- print('hello,chongchong! ', end='\r')
可以通过sep自定义分割符,而不是默认的空格。
- print('hello', ' chongchong!', sep=',')
通过file来重定向输出到文件,而不是stdout的默认标准输出。
- print(' hello,chongchong!', file=open('chongchong.out', 'w'))
我们可以利用click在命令行中实现对这些参数的输入和解析,实现一个简单的打印命令行界面。废话少说直接上代码:
- #!/usr/bin/env python3
- import click
- import sys
- @click.command()
- @click.argument('args', nargs=-1)
- @click.option('-s', '--sep', default=' ', help='分隔符,默认为空')
- @click.option('-e', '--end', default='\n', help='结尾字符,默认为 "\\n"')
- @click.option('-o', '--out', default=None, help='输出文件,默认为stdout)')
- def print_prog(args, sep, end, out):
- '''打印输出结果。
- '''
- print(*args, sep=sep, end=end, file=open(out, 'w') if out else sys.stdout)
- if __name__ == '__main__':
- print_prog()
脚本很简单,就是获取输入的参数,并将其传递给print()函数执行。执行如下:
打印文本
- python print.py hello,chongchong!
显示帮助
- python print.py –help
设置风格符号
- python print.py -s %% hello chongchong!
重定向到文件
- python print.py -o chongchong.out -s %% hello chongchong!
通过几行代码就实现了一个很棒的交互界面!
当然你可以依次为基础增加更多的内容,让它更实用这是一个演示,向您展示创建自己的CLI是多么容易。你不再需要编写普通的python脚本文件了。更多功能和高级特性可以参考click官方文档:
更多的基于click的项目和模块也可以通过其github仓clone和fork使用,你也可以给该项目添加自己的贡献。
具体地址为:github:/click-contrib/。
argparse
我们首先从官方的数字计算程序小例子来开始:
- import argparse
- parser = argparse.ArgumentParser(description='Process some integers.')
- parser.add_argument('integers', metavar='N', type=int, nargs='+',
- help='an integer for the accumulator')
- parser.add_argument('--sum', dest='accumulate', action='store_const',
- const=sum, default=max,
- help='sum the integers (default: find the max)')
- args = parser.parse_args()
- print args.accumulate(args.integers)
代码解释:
首先,通过argparse.ArgumentParser()创建一个解析对象;
接着,给对象添加两个参数,用add_argument()方法。比如本例中是一个type为一个或者一列整数;还有一个目标为accumulate属性用来选择两个函数sum()和max()求和和求最大值,默认为求最大值。
然后,通过parse_args()函数解析对象的参数,并把各个参数解析为合适的类型(int,accumulate),触发对应的行为(sum,max)。
最后通过print打印结果。
ArgumentParser.add_argument()
使用方法如下:
ArgumentParser构造对象
ArgumentParser对象的初始化原型为(构造方法):
- ArgumentParser(prog=None, usage=None, description=None, epilog=None, parents=[], formatter_class=argparse.HelpFormatter, prefix_chars='-', fromfile_prefix_chars=None, argument_default=None, conflict_handler='error', add_help=True)
其各个参数为:
prog - 程序的名称(默认为sys.argv[0])
usage - 描述程序使用帮助信息(默认值:从添加的参数来生成)
description - 在参数帮助之前显示的文本(默认值:无)
epilog - 在参数帮助后显示的文本(默认值:无)
parents - ArgumentParser也应包含其参数的对象列表
formatter_class - 用于自定义帮助输出的类
prefix_chars - 前缀可选参数的字符集(默认值:' - ')
fromfile_prefix_chars -文件前缀字符参数(默认值:None)
argument_default -为参数的全局默认值(默认值:None)
conflict_handler - 解决冲突选项的策略(通常不需要)
add_help -添加-h/--help选项解析器(默认值:True)
一般情况下,我们无需管其他情况,只需设置个程序描述即可。
add_argument()方法
add_argument()方法用来创建解析参数,定义对参数具体的解析,其原型如下:
add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
参数有:
name或者flags - 名称或选项字符串列表,必填写项。例如foo或-f, --foo。
action - 在命令行中遇到此参数时要采取的操作类型。
nargs – 要使用的命令行参数的数量,用通配符号表示,比如"*","+"和"?",分别表示不限数量,至少一个和一个参数。
const - 一些动作和 nargs 选择所需的常量。
default - 如果参数在命令行中不存在,则默认使用的值。
type - 命令行参数应该转换的类型,默认为字符串string
choices – 可选项, 一个允许参数值的容器。
required - 是否可以省略命令行选项(仅对可选项)。
help - 帮助提示信息,用来对参数进行必要的描述。
metavar – 帮助消息替代参数的显示名称。
dest - 要添加到parse_args()函数返回对象的属性的名称。
parse_args()方法
parse_args方法用来把命令行中的字符,解析到参数解析命名空间(add_argumnet()创建)定义的各个参数。其原型如下:
- parse_args(args=None, namespace=None)
args为参数列表,参数字符串通过调用sys.argv获取,Namespace为属性创建一个新的空对象。
实例学习sqlmap
最后我们节选一个著名sql注入渗透工具sqlmap的cmdline.py部分代码,来展示一个大型软件中,如何实际通过argparser来构建起命令行界面的。基本上也跟上面流程方法一样,不过额外使用一些函数和功能。
全部代码详见sqlmap github官方仓库:
(github /sqlmapproject/sqlmap/blob/master/lib/parse/cmdline.py)
总结
本文我们实例介绍了python下构建优雅终端命令行界面两种方法Click和argparser。善用他们可以为我们的代码工作减少很多繁琐的命令行参数的定义和解析过程,提高编码的效率。