python 命令行参数-Python 命令行参数解析

Python用于编写脚本的场景非常多，如何处理脚本（命令行）的参数当然非常关键

首先导入sys，参数列表在sys.arg中

import sys

if __name__ == '__main__':

for arg in sys.argv:

print arg

可以发现sys.arg[0]，也就是第一个参数是执行脚本的名字，所以一般获取参数是sys.arg[1:]

如果参数比较简单，或者做给自己用用，那就自己判断下吧

但如果想让脚本看起来专业点：比如支持长短格式，长格式添加=，帮助信息等完善的参数操作，通过相应的类库可以较快实现

1.getopt

import sys

import getopt

def main():

try:

opts, args = getopt.getopt(sys.argv[1:],'d:f:h',['days=','files=','help']) # 当一个选项只是表示开关状态时，即后面不带附加参数时，在分析串中写入选项字符。当选项后面是带一个附加参数时，在分析串中写入选项字符同时后面加一个":"号

except getopt.GetoptError:

usage()

sys.exit()

print(opts)

print(args)

for o, a in opts:

if o in ("-h", "--help"):

usage()

sys.exit()

elif o in ("-d","--days"):

day = a

elif o in ("-f", "--files"):

files = a

print(day)

print(files)

def usage():

print """

usage:xxxxxxx

"""

if __name__ == '__main__':

main()

如果要实现完整的功能，以上应该算一个基本的结构了，对于小功能的设置应该能满足了

但！整个过程还是相对零散，对于更加复杂的应用，编写的代码简直无法直视

下面介绍2款完善的类库，是官方推荐的

功能强大，易于使用，方便地生成标准的、符合Unix/posix规范的命令行说明

2.optparse

import optparse

def main():

parser = optparse.OptionParser(usage="%prog -f server.list -u root", version="%prog 1")

parser.add_option("-f", "--file", dest="File", action="store", help="Host list")

parser.add_option("-u", "--user",action="store", dest="User",help="Username")

parser.add_option('-v', '–version', action='store_true', dest='version', default=False, help='show the version of this command')

#store_true这个action的意思是只有在使用该参数的时候存储, default则可以给一个默认值。也就是你加了-v 显示版本信息，否则不显示。

(options, args) = parser.parse_args()

"""

1. options , 这是一个对象(optpars.Values)，保存命令行参数值。只要知道命令行参数名,如file，就可以访问其对应的值：options.file。

2. args , 一个由 positional arguments 组成的列表

"""

if __name__ == '__main__':

main()

已经能非常方便的进行参数解析，并且还提供自动help、参数分组、版本提示等

3.argparse

应该是对于optparse的一个升级，官方在2.7版本提议用argparse代替optparse

代码更易理解，功能基本和optparse一致

import argparse

def main():

parser = argparse.ArgumentParser(usage="%prog -f server.list -u root", version="%prog 1")

parser.add_argument("-f", "--file", dest="File", action="store", help="Host list")

parser.add_argument("-u", "--user",action="store", dest="User",help="Username")

parser.add_argument('-v', '–version', action='store_true', dest='version', default=False, help='show the version of this command')

options = parser.parse_args()

# 如果运行程序带了一个不认识的参数，就会报错，可以使用下面的接口做不同处理

options, args = parser.parse_known_args()

if __name__ == '__main__':

main()

以上2者使用基本相同，代码上基本使用基本只要替换接口名称即可

但argparse的自动help及usage能显示的更加专业，或许还有更深层次的不同