python命令行输入文件名参数_Python自定义命令行参数

用过别人的脚本命令参数后，如何自己也自定义命令行参数呢，下面教给大家这些纸老虎：

命令行模块getopt:

getopt函数中一共有三个参数：

getopt.getopt(sys.argv[1:], "ho:i:", ["help", "output="])

sys.argv[1:] 和C语言中的argv相似，我们一般写sys.argv[1:]来过滤第一个参数，因为第一个参数是python脚本文件名，一般对我们没有实际用处，过滤掉即可。

"ho:i:":刚看的同学肯定有点懵*，可以称作为短格式分析串，是否曾经用过类似 XXX -i XXX -o XXX这样的命令，这对帮助你理解短格式分析串很有帮助，h不加":"，表示不带参数。即-h，o后面有":"代表需要加参数-o "your file path",那么短格式，输入一个"-"即可。

["help", "output="]:python列表形式表示，称为长格式串，无"="不加参数，带"="需要加参数类似短格式串。用法为--help --output "your file path"。

以上分析相信你已经了解了这些参数作用,那么上一段简短的代码：

if __name__ == '__main__':

# for item in allfile:

# print item

# print os.path.splitext(item)[0]

#getopt()

try:

options, args = getopt.getopt(sys.argv[1:], "ho:i:", ["help", "ip=", "port="])

except getopt.GetoptError:

print 'hi Sam getopt error!Please input -h or --help'

sys.exit(1)

for name, value in options:

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

usage()

print ""

elif name in("-o","-i"):

print "short parameter %s" % (value)

elif name in("--ip","--port"):

print "long parameter %s"%(value)

for item in args:

print item

python环境下我们输入命令测试我们的脚本是否正确：

$ python getoptTest.py -o sam -i tom

short parameter sam

short parameter tom

$ python getoptTest.py --port 80

long parameter 80

python FileTest.py --port 80 -o Sam 100 120

long parameter 80

short parameter Sam

100

120

如果我们估计输入错误：

$ python FileTest.py -i

异常捕获结果：hi Sam getopt error!Please input -h or --help

自定义就是这么简单。