使用Python轻松写出漂亮的命令行程序
诞生之初用来作为配置管理语言的Python,现在已经成为最流行的编程语言之一,尤其是随着AI应用的兴起,变得无处不在。
文章图片
Python中的getopt
,optparse
模块都可以进行命令行程序的开发,不过它们已经被废弃,在2.7和3.2版本中引入了argparse
模块;这里介绍使用argparser
模块,只需要数十行代码,就可以做出漂亮的命令行程序。
1.基本的命令行
先看看官方文档里的一个示例:
def main():
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)
这几行代码实现的功能为:
如果命令行输入不带
--sum
选项,则对输入的多个整数执行max
操作;否则,执行sum
操作。例如:在命令行中执行:
$ python prog.py 1 2 3 4
输出为:4
在命令行中执行:
$ python prog.py 1 2 3 4 --sum
输出为:10
上面的代码中,主要的过程分为三个步骤:
- 调用
argparse.ArgumentParser
的构造器创建一个ArgumentParser
对象。 - 调用
ArgumentParser
的add_argument()
方法添加命令行的参数,包括可选参数和必选参数。其中以--
、-
开头的表示可选参数,其它的表示必须输入的参数。 - 调用
parser.parse_args()
方法对传入的sys.argv
参数进行解析。
【使用Python轻松写出漂亮的命令行程序】有时候,我们可以根据功能将命令行分组成一系列相关的子命令。例如,Git命令行可以分为以下几组:
These are common Git commands used in various situations:start a working area (see also: git help tutorial)
cloneClone a repository into a new directory
initCreate an empty Git repository or reinitialize an existing onework on the current change (see also: git help everyday)
addAdd file contents to the index
mvMove or rename a file, a directory, or a symlink
resetReset current HEAD to the specified state
rmRemove files from the working tree and from the indexexamine the history and state (see also: git help revisions)
bisectFind by binary search the change that introduced a bug
grepPrint lines matching a pattern
logShow commit logs
showShow various types of objects
statusShow the working tree statusgrow, mark and tweak your common history
branchList, create, or delete branches
checkoutSwitch branches or restore working tree files
commitRecord changes to the repository
diffShow changes between commits, commit and working tree, etc
mergeJoin two or more development histories together
rebaseForward-port local commits to the updated upstream head
tagCreate, list, delete or verify a tag object signed with GPGcollaborate (see also: git help workflows)
fetchDownload objects and refs from another repository
pullFetch from and integrate with another repository or a local branch
pushUpdate remote refs along with associated objects
在Python中,利用
argparse
模块也可以很容易做出这样的子命令行程序。例如,我们要做一个对用户信息进行管理的命令行程序,包括添加、删除、查询显示等几个功能。相应的代码可以这样写:
我们可以根据功能将其分为三个子命令:add
,delete
,show
。
def main():
parser = argparse.ArgumentParser(prog='cmd')sub_parser = parser.add_subparsers(title='subcommands',
description='valid subcommands',
help='config subscommand help')add_parser = sub_parser.add_parser('add',help='add user config')
add_parser.add_argument('--name', required=True, help='user name')
add_parser.add_argument('--addr', required=False, help='user address')
add_parser.add_argument('--phone', required=False, help='phone number')
add_parser.set_defaults(func=add_fn)delete_parser = sub_parser.add_parser('delete', help='delete user config')
delete_parser.add_argument('--name', required=True, help='user name')
delete_parser.set_defaults(func=delete_fn)show_parser = sub_parser.add_parser('show', help='show user config')
show_parser.set_defaults(func=show_fn)args = parser.parse_args()
args.func(args)
对上面的代码进行概要的解释:
ArgumentParser
对象的add_subparsers()
方法创建了一个特殊的action实例,对这个实例分别调用add_parser()
方法来创建子命令对应的ArgumentParser
,再调用这个ArgumentParser
实例的add_argument()
方法来添加子命令的参数。一个比较方便的地方在于,可以通过
ArgumentParser
的set_defaults()
方法来设置子命令对应的处理函数。上面的示例中,分别为三个子命令设置了处理函数:add_fn
,delete_fn
,show_fn
。在命令行程序中执行:
$python cmd.py --help
将显示出以下信息:
usage:
cmd [sub-command]optional arguments:
-h, --helpshow this help message and exitsubcommands:
valid subcommands{add,delete,show}config subscommand help
addadd user config
deletedelete user config
showshow user config
从帮助信息中可以看到,命令行被组织为三个子命令,分别为
add
,delete
,show
。在命令行中执行:
$ python cmd.pyadd--help
可以查看
add
子命令的帮助信息:usage: cmd [sub-command] add [-h] --name NAME [--addr ADDR]
[--phone PHONE]optional arguments:
-h, --helpshow this help message and exit
--name NAMEuser name
--addr ADDRuser address
--phone PHONEphone number
对于
add
子命令,它有3个参数,分别为用户名、联系地址、电话号码;其中用户名为必选参数,其它两个为可选参数。在命令行中执行:
$ python cmd.py add --name 'Eric' --addr 'California,USA' --phone '0076938377'`
添加一个User的处理逻辑将由
add_fn()
来完成。程序中将得到一个信息为{name='Eric',addr='California,USA', phone='0076938377'}
的User。def add_fn(args):
args_dict = vars(args)name = args_dict['name']
address = args_dict['addr']
phone_number = args_dict['phone']print 'Add User : %-10s %10s %12s' % (name, address, phone_number)
...
其它的
delete
和show
子命令与此类似,不再赘述。3.小结
使用Python的
argparse
模块,短短的几十行代码就可以搭建起来一个命令行程序的骨架。argparse
模块中最重要的是ArgumentParser
对象,它包含了以下几个方法:- add_argument()
- add_subparsers()
- parse_args()
推荐阅读
- 由浅入深理解AOP
- 【译】20个更有效地使用谷歌搜索的技巧
- mybatisplus如何在xml的连表查询中使用queryWrapper
- MybatisPlus|MybatisPlus LambdaQueryWrapper使用int默认值的坑及解决
- MybatisPlus使用queryWrapper如何实现复杂查询
- python学习之|python学习之 实现QQ自动发送消息
- 逻辑回归的理解与python示例
- python自定义封装带颜色的logging模块
- 【Leetcode/Python】001-Two|【Leetcode/Python】001-Two Sum
- iOS中的Block