如何编写一个带命令行参数的Python文件

Posted 554303896

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何编写一个带命令行参数的Python文件相关的知识,希望对你有一定的参考价值。

看到别人执行一个支持命令行参数的python文件,瞬间觉得高大上起来、牛逼起来,那么如何编写一个带命令行参数的python脚本呢?不用紧张,下面将简单易懂地让你学会如何让自己的python脚本,支持命令行参数。

首先你要知道python中的sys模块的一些功能:

import sys

print "the number of python program‘s argument:",len(sys.argv)

print "the value of every argument is ",str(sys.argv)

#上述程序的文件名sysargv.py
python sysargv.py argv1 argv2 argv3 argv4
the number of python programs argument: 5
the value of every argument is  [sysargv.py, argv1, argv2, argv3, argv4]

其次,python程序使用命令行参数,必不可少的模块,就是getopt 模块,先看看一段代码

getopt.getopt(args, options[, long_options])
import getopt
args = -a -b -cfoo -d bar a1 a2.split()
args
[-a, -b, -cfoo, -d, bar, a1, a2]
optlist, args = getopt.getopt(args, abc:d:)
optlist
[(-a, ‘‘), (-b, ‘‘), (-c, foo), (-d, bar)]
args
[a1, a2]

使用long_options

s = --condition=foo --testing --output-file abc.def -x a1 a2
args = s.split()
args
[--condition=foo, --testing, --output-file, abc.def, -x, a1, a2]
optlist, args = getopt.getopt(args, x, [condition=, output-file=, testing])
optlist
[(--condition, foo), (--testing, ‘‘), (--output-file, abc.def), (-x, ‘‘)]
args
[a1, a2]

最后实战一个例子吧!

import getopt,sys

def main():
    try:
        opts,args=getopt.getopt(sys.argv[1:],"hi:o:v",["help","infile=","outfile="])
    except getopt.GetoptError as error:
        print str(error)
        usage()
        sys.exit(2)
    infile=None
    output=None
    verbose=False
    for key,value in opts:
        if key=="-v":
            verbose=True
        elif key in ("-h","--help"):
            print "sysargv.py -i <inputfile> -o <outputfile>"
            print "or sysargv.py --infile <inputfile> --outfile <outputfile>"

        elif key in ("-i","--infile"):
            infile = value
        elif key in ("-o","--outfile"):
            output= value
    print "inputfile:", infile
    print  "outputfile:", output
    print  verbose
if __name__=="__main__":
    main()

测试结果:

C:\Python27>python sysargv.py --help
sysargv.py -i <inputfile> -o <outputfile>
or sysargv.py --infile <inputfile> --outfile <outputfile>
inputfile: None
outputfile: None
False

C:\Python27>python sysargv.py -h
sysargv.py -i <inputfile> -o <outputfile>
or sysargv.py --infile <inputfile> --outfile <outputfile>
inputfile: None
outputfile: None
False

C:\Python27>python sysargv.py -i "inputfile1" -o "ouputfile2"
inputfile: inputfile1
outputfile: ouputfile2
False

C:\Python27>python sysargv.py -i "inputfile1"
inputfile: inputfile1
outputfile: None
False


C:\Python27>python sysargv.py -o "outputfile1"
inputfile: None
outputfile: outputfile1
False

C:\Python27>python sysargv.py -o "outputfile1" -v
inputfile: None
outputfile: outputfile1
True



C:\Python27>python sysargv.py --infile "inputfile" --outfile "outputfile1" -v
inputfile: inputfile
outputfile: outputfile1
True

 

以上是关于如何编写一个带命令行参数的Python文件的主要内容,如果未能解决你的问题,请参考以下文章

如何在命令行脚本中启动带参数的Windows服务

WPF如何获取命令行参数

如何将字符串语句作为命令行参数传递

如何运行文本文档编程命令

安装了python2跟python3命令行怎么区分

如何用Python编写代码在Word中实现带公式计算过程的计算书?