码迷,mamicode.com
首页 > 编程语言 > 详细

转:python命令行解析工具Argparse

时间:2015-09-15 12:35:56      阅读:278      评论:0      收藏:0      [点我收藏+]

标签:

转自:http://www.cnblogs.com/jianboqi/archive/2013/01/10/2854726.html

最近在研究pathon的命令行解析工具,argparse,它是Python标准库中推荐使用的编写命令行程序的工具。

以前老是做UI程序,今天试了下命令行程序,感觉相当好,不用再花大把时间去研究界面问题,尤其是vc++中尤其繁琐。

现在用python来实现命令行,核心计算模块可以用c自己写扩展库,效果挺好。

学习了argparse,在官方文档中找到一篇toturial,简单翻译了下。

http://docs.python.org/2/howto/argparse.html#id1

Argparse Tutorial

这篇教程简明地介绍了Python标准库推荐使用的命令行参数解析模块——Argparse的使用。

基本概念

  在这篇教程中我们通过一个常见的ls命令来展示argparse的功能。

技术分享
$ ls
cpython  devguide  prog.py  pypy  rm-unused-function.patch
$ ls pypy
ctypes_configure  demo  dotviewer  include  lib_pypy  lib-python ...
$ ls -l
total 20
drwxr-xr-x 19 wena wena 4096 Feb 18 18:51 cpython
drwxr-xr-x  4 wena wena 4096 Feb  8 12:04 devguide
-rwxr-xr-x  1 wena wena  535 Feb 19 00:05 prog.py
drwxr-xr-x 14 wena wena 4096 Feb  7 00:59 pypy
-rw-r--r--  1 wena wena  741 Feb 18 01:01 rm-unused-function.patch
$ ls --help
Usage: ls [OPTION]... [FILE]...
List information about the FILEs (the current directory by default).
Sort entries alphabetically if none of -cftuvSUX nor --sort is specified.
技术分享

从以上的四个命令中,我们能够了解以下几个基本概念:

  • ls命令在没有参数的情况下也是可以运行的,默认打印出当前目录下的所有内容。
  • 如果我们想让它展示更多内容,那么我们需要给它多一点参数。在这种情况下,我们想显示一个不同的目录,pypy。我们所做的就是明确了常见的定位 参数(positional argument),这样命名是因为需要程序根据参数在命令行中的位置而决定做什么。这个概念与命令cp更为接近,它的用法是cp src dest,src表示的是你想要拷贝的文件,dest表示你想要将文件拷贝到哪里。
  • 现在,我想要改变程序的行为。在我们的例子中,我想显示文件的向西信息而不仅仅是文件名,参数-l 则是我们所知道的可选参数(optinal argument)
  • 最后一句是显示帮助的文档的一个片段,当你遇到你从未使用过的命令时,你可以通过它学习怎么使用。

基本认识

    我们从一个基本的程序开始(它什么也不做)

import argparse
parser = argparse.ArgumentParser()
parser.parse_args()

运行结果:

技术分享
$ python prog.py
$ python prog.py --help
usage: prog.py [-h]

optional arguments:
  -h, --help  show this help message and exit
$ python prog.py --verbose
usage: prog.py [-h]
prog.py: error: unrecognized arguments: --verbose
$ python prog.py foo
usage: prog.py [-h]
prog.py: error: unrecognized arguments: foo
技术分享

结果分析:

  • 若不给参数而运行这个程序,将不会得到任何结果。
  • 第二条命名显示了使用的argparse的好处,你什么也没做,却得到了一个很好的帮助信息。
  • 我们无需人为设置--help参数,就能得到一个良好的帮助信息。但是若给其他参数(比如foo)就会产生一个错误。

位置参数

   首先,给一个例子:

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("echo")
args = parser.parse_args()
print args.echo

运行结果:

技术分享
$ python prog.py
usage: prog.py [-h] echo
prog.py: error: the following arguments are required: echo
$ python prog.py --help
usage: prog.py [-h] echo

positional arguments:
  echo

optional arguments:
  -h, --help  show this help message and exit
$ python prog.py foo
foo
技术分享

结果分析:

  • 这次,我们增加了一个add_argument()方法,用来设置程序可接受的命令行参数。
  • 现在要运行程序,就必须设置一个参数。
  • parse_args()方法实际上从我们的命令行参数中返回了一些数据,在上面的例子中是echo
  • 这个像“魔法”一样的过程,是argparse自动完成的。

尽管自动产生的帮助信息展示地很美观,但是我们仍然无法只根据echo这个参数知道它是做什么的。所以,我们增加了一些东西,使得它变得更有用。

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("echo", help="echo the string you use here")
args = parser.parse_args()
print args.echo

运行结果:

技术分享
$ python prog.py -h
usage: prog.py [-h] echo

positional arguments:
  echo        echo the string you use here

optional arguments:
  -h, --help  show this help message and exit
技术分享

在此基础上,我们再多改变一点:(计算输入参数square的平方)

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number")
args = parser.parse_args()
print args.square**2

下面是运行结果:

$ python prog.py 4
Traceback (most recent call last):
  File "prog.py", line 5, in <module>
    print args.square**2
TypeError: unsupported operand type(s) for ** or pow(): str‘ and int

这个程序并不能正确运行,因为argparse会将输入当作字符串处理,所以我们需要设置它的类型:(type=int)

技术分享
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number",
                    type=int)
args = parser.parse_args()
print args.square**2
技术分享

下面是运行结果:

$ python prog.py 4
16
$ python prog.py four
usage: prog.py [-h] square
prog.py: error: argument square: invalid int value: ‘four‘

现在,这个程序能够顺利运行,而且能够处理一些错误输入。

完整范例如下

# -*- coding: utf-8 -*-  
  
import argparse  
  
args = "-f hello.txt -n 1 2 3 -x 100 -y b -z a -q hello @args.txt i_am_bar -h".split()   
# 使用@args.txt要求fromfile_prefix_chars="@"  
# args.txt文件中应该一行一个参数,想改变行为参考convert_arg_line_to_args()  
  
  
# ArgumentParser参数的简单说明  
## description - 命令行帮助的开始文字,大部分情况下,我们只会用到这个参数  
# epilog - 命令行帮助的结尾文字  
# prog - (default: sys.argv[0])程序的名字,一般不需要修改,另外,如果你需要在help中使用到程序的名字,可以使用%(prog)s  
# prefix_chars - 命令的前缀,默认是-,例如-f/--file。有些程序可能希望支持/f这样的选项,可以使用prefix_chars="/"  
# fromfile_prefix_chars - (default: None)如果你希望命令行参数可以从文件中读取,就可能用到。例如,如果fromfile_prefix_chars=‘@‘,命令行参数中有一个为"@args.txt",args.txt的内容会作为命令行参数  
# add_help - 是否增加-h/-help选项 (default: True),一般help信息都是必须的,所以不用设置啦。  
##  parents - 类型是list,如果这个parser的一些选项跟其他某些parser的选项一样,可以用parents来实现继承,例如parents=[parent_parser]  
## formatter_class - 自定义帮助信息的格式(description和epilog)。默认情况下会将长的帮助信息进行<自动换行和消除多个连续空白>。  
#三个允许的值:  
# class argparse.RawDescriptionHelpFormatter 直接输出description和epilog的原始形式(不进行自动换行和消除空白的操作)  
# class argparse.RawTextHelpFormatter 直接输出description和epilog以及add_argument中的help字符串的原始形式(不进行自动换行和消除空白的操作)  
## class argparse.ArgumentDefaultsHelpFormatter 在每个选项的帮助信息后面输出他们对应的缺省值,如果有设置的话。这个最常用吧!  
# argument_default - (default: None)设置一个全局的选项的缺省值,一般每个选项单独设置,所以这个参数用得少,不细说  
# usage - (default: generated)如果你需要修改usage的信息(usage: PROG [-h] [--foo [FOO]] bar [bar ...]),那么可以修改这个,一般不要修改。  
# conflict_handler - 不建议使用。这个在极端情况下才会用到,主要是定义两个add_argument中添加的选项的名字发生冲突时怎么处理,默认处理是抛出异常。  
#注释一行有##表示这几个参数比较常用  
parser = argparse.ArgumentParser(description="This is a description of %(prog)s", epilog="This is a epilog of %(prog)s", prefix_chars="-+", fromfile_prefix_chars="@", formatter_class=argparse.ArgumentDefaultsHelpFormatter)  
  
# ArgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])  
# add_argument的参数是比较复杂的。。。  
  
# name or flags - 指定参数的形式,想写几个写几个,不过我们一般就写两个,一个短参数,一个长参数,看下面的例子"-f", "--file"  
# 可选的选项,位置不固定,想怎么写就怎么写,默认是可选的  
parser.add_argument("-f", "--file", help="test test test")  
# 位置固定的选项,例如"prog i_am_bar",这样子的话,i_am_bar就是bar选项的值啦,默认是必须有的  
parser.add_argument("bar", help="test test test")  
  
# nargs - 指定这个参数后面的value有多少个,例如,我们希望使用-n 1 2 3 4,来设置n的值为[1, 2, 3, 4]  
parser.add_argument("-n", "--num", nargs="+", type=int)  
# 这里nargs="+"表示,如果你指定了-n选项,那么-n后面至少要跟一个参数,+表示至少一个,?表示一个或0个,*0个或多个,  
  
# default - 如果命令行没有出现这个选项,那么使用default指定的默认值  
parser.add_argument("+g", "++gold", help="test test test", default="test_gold")#需要prefix_chars包含"+"  
  
# type - 如果希望传进来的参数是指定的类型(例如 float, int or file等可以从字符串转化过来的类型),可以使用  
parser.add_argument("-x", type=int)  
  
# choices - 设置参数值的范围,如果choices中的类型不是字符串,记得指定type哦  
parser.add_argument("-y", choices=[a, b, d])  
  
# required - 通常-f这样的选项是可选的,但是如果required=True那么就是必须的了  
parser.add_argument("-z", choices=[a, b, d], required=True)  
  
# metavar - 参数的名字,在显示 帮助信息时才用到.  
parser.add_argument("-o", metavar="OOOOOO")  
  
# help - 设置这个选项的帮助信息  
# dest - 设置这个选项的值就是解析出来后放到哪个属性中  
parser.add_argument("-q", dest="world")   
  
args = parser.parse_args(args) # 如果你没有args参数,那么就使用sys.argv,也就是命令行参数啦。有这个参数,就方便我们调试啊  
# args.world就是-q的值啦  
  
# action - The basic type of action to be taken when this argument is encountered at the command line.  
# const - A constant value required by some action and nargs selections.  
# 这两个自己看帮助文档啦,比较复杂  
# http://docs.python.org/library/argparse.html  
  
print args

Shell:

usage: argparse_sample.py [-h] [-f FILE] [-n NUM [NUM ...]] [+g GOLD] [-x X]  
                          [-y {a,b,d}] -z {a,b,d} [-o OOOOOO] [-q WORLD]  
                          bar  
   
This is a description of argparse_sample.py  
   
positional arguments:  
  bar                   test test test  
   
optional arguments:  
  -h, --help            show this help message and exit  
  -f FILE, --file FILE  test test test (default: None)  
  -n NUM [NUM ...], --num NUM [NUM ...]  
  +g GOLD, ++gold GOLD  test test test (default: test_gold)  
  -x X  
  -y {a,b,d}  
  -z {a,b,d}  
  -o OOOOOO  
  -q WORLD  
   
This is a epilog of argparse_sample.py  
   

 

转:python命令行解析工具Argparse

标签:

原文地址:http://www.cnblogs.com/alecyrus/p/4809558.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!