python命令行参数传递_Python中 命令行参数传递 与 处理

英文原文： http://www.artima.com/weblogs/viewpost.jsp?thread=4829

很多人会在 if __name__ == "__main__": 中写N多行的代码。 一般情况这是没问题的。 但如果你需要以命令行的方式运行调试， 那么就不方便了。

而这个文章中像C一样定义了一个main()函数， 在 if __name__ == "__main__": 中只有一行 main() 调用， 很好的代码结构！

而且还有一个好处：你可以通过修改main()的调用参数，达到随意修改命令行参数的目的。

First, we change main() to take an optional 'argv' argument, which allows us to call it from the interactive Python prompt:

def main(argv=None):

if argv is None:

argv = sys.argv

# etc., replacing sys.argv with argv in the getopt() call.

首先， 让main()接受默认参数， 然后在main()里头判断， 如果没有给main参数的话，就是用sys.argv作为参数。

Note that we fill in the default for argv dynamically. This is more flexible than writing

def main(argv=sys.argv):

# etc.

because sys.argv might have been changed by the time the call is made; the default argument is calculated at the time the main() function is defined, for all times.

而这种方法更灵活！ 因为在main()内部引用 sys.argv参数时， sys.argv参数可能在调用main()前被我们自己的代码修改过了。 而main(argv=sys.argv)的定义方式， 保证了传给main（）的默认参数一定是执行该脚本文件时给的参数！！！ 因为Python函数的默认参数是在'编译'阶段就确定了，在运行过程中是不会被修改的。

====== 下面与译文无关， 只是一个练习： =====

下面是从 http://blog.sina.com.cn/s/blog_71c19c620100r7mk.html 上摘抄的：

使用getopt模块处理Unix模式的命令行选项：

getopt模块用于抽出命令行选项和参数，也就是sys.argv。命令行选项使得程序的参数更加灵活。支持短选项模式和长选项模式。

# scriptname.py

import getopt

shortargs = 'f:t'

longargs = ['directory-prefix=', 'format', '--f_long=']

opts, args = getopt.getopt( sys.argv[1:], shortargs, longargs )

getopt函数的格式是getopt.getopt ( [命令行参数列表], "短选项", [长选项列表] )

短选项名后的冒号(:)表示该选项必须有附加的参数。

长选项名后的等号(=)表示该选项必须有附加的参数。

返回opts和args。

opts是一个参数选项及其value的元组( ( '-f', 'hello'), ( '-t', '' ), ( '--format', '' ), ( '--directory-prefix', '/home' ) )

args是一个除去有用参数外其他的命令行输入 ( 'a', 'b' )

然后遍历opts便可以获取所有的命令行选项及其对应参数了。

for opt, val in opts:

if opt in ( '-f', '--f_long' ):

pass

对上边的片段， 我整理了一个例子：bell@bell-desktop:/tmp$ cat py_optParse.py

import sys

import getopt

shortargs = 'f:t'

longargs = ['directory-prefix=', 'format', '--f_long=']

# 这时要求长选项参数名字为： ----f_long

#如果执行python py_optParse.py --f_long a.txt 那么出错： option --f_long not recognized

opts, args = getopt.getopt( sys.argv[1:], shortargs, longargs )

print "opts:", opts

print "args:", args

class Usage(Exception):

def __init__(self, msg):

self.msg = msg

parameter_missing = True

for opt, val in opts:

#if opt in ( '-f', '--f_long' ):

if opt in ( '-f', '----f_long' ):

parameter_missing = False

if(parameter_missing):

raise Usage("missing argument!")

bell@bell-desktop:/tmp$ python py_optParse.py -f a.txt

opts: [('-f', 'a.txt')]

args: []

bell@bell-desktop:/tmp$ python py_optParse.py --f_long a.txt

Traceback (most recent call last):

File "py_optParse.py", line 6, in

opts, args = getopt.getopt( sys.argv[1:], shortargs, longargs )

File "/usr/lib/python2.6/getopt.py", line 89, in getopt

opts, args = do_longs(opts, args[0][2:], longopts, args[1:])

File "/usr/lib/python2.6/getopt.py", line 153, in do_longs

has_arg, opt = long_has_args(opt, longopts)

File "/usr/lib/python2.6/getopt.py", line 170, in long_has_args

raise GetoptError('option --%s not recognized' % opt, opt)

getopt.GetoptError: option --f_long not recognized

bell@bell-desktop:/tmp$ python py_optParse.py ----f_long a.txt

opts: [('----f_long', 'a.txt')]

args: []

Traceback (most recent call last):

File "py_optParse.py", line 24, in

raise Usage("missing argument!")

__main__.Usage

bell@bell-desktop:/tmp$