Python 获取命令行参数:getopt模块

引用

Python 命令行参数,修改并增加了部分内容,原文链接地址ttp://www.runoob.com/python/python-command-line-arguments.html,如涉及版权问题请及时联系我!

Python 提供了 getopt 模块来获取命令行参数,getopt模块是专门处理命令行参数的模块,用于获取命令行选项和参数。下面通过实际例子来详细解释如何操作。

类似于下面包含参数的命令行

例1- $ python test.py -i input.txt -o output.txt middle.txt final.txt
例2- $ python test.py --ifile=input.txt --ofile=output.txt middle.txt 

getopt.getopt 方法函数原型:

getopt.getopt(args, shortopts, longopts=[])

方法参数说明:

  • ** args:args为需要解析的参数列表。一般使用sys.argv[1:]**,这样可以过滤掉第一个参数(ps:第一个参数是脚本的名称,即例子中的test.py文件名,它不应该作为参数进行解析).
  • ** shortopts:**简写参数列表,短格式 (-) ,冒号(:)表示该选项必须有附加的参数,不带冒号表示该选项不附加参数.
  • ** longopts:**长参数列表,长格式(--) ,等号(=)表示如果设置该选项,必须有附加的参数,否则就不附加参数。
方法返回值:
  • opts:分析出的(option, value)列表对。
  • args:不属于格式信息的剩余命令行参数列表。
实例

假定我们创建这样一个脚本,可以通过命令行向脚本文件传递两个文件名,同时我们通过另外一个选项查看脚本的使用。脚本使用方法如下:

usage: test.py -i <inputfile> -o <outputfile>

test.py 文件代码如下所示:

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import sys, getopt
def main(argv): 
    inputfile = '' 
    outputfile = '' 
    try: 
        opts, args = getopt.getopt(argv,"hi:o:",["help","ifile=","ofile="])       
    except getopt.GetoptError: 
        sys.exit(2) 
    for opt, arg in opts: 
        if opt in ("-h","--help"): 
            print 'test.py -i <inputfile> -o <outputfile>' 
            sys.exit() 
        elif opt in ("-i", "--ifile"): 
            inputfile = arg 
        elif opt in ("-o", "--ofile"): 
            outputfile = arg 
    print '输入的文件为:', inputfile 
    print '输出的文件为:', outputfile

if __name__ == "__main__": 
    main(sys.argv[1:])
执行以上代码,输出结果为:
$ python test.py -h
usage: test.py -i <inputfile> -o <outputfile>

$ python test.py -i input.txt -o output.txt middle.txt final.txt
输入的文件为: input.txt
输出的文件为: output.txt

$ python test.py --ifile=input.txt --ofile=output.txt middle.txt final.txt
输入的文件为: input.txt
输出的文件为: output.txt
对代码的理解:
  • main(sys.argv[1:])
    调用main函数,getopt.getopt获取命令行参数-i input.txt -o output.txt middle.txt final.txt,

  • 返回值opts是个包含元祖的列表,每个元祖是分析出来的格式信息,比如 [('-i','input.txt'),('-o','output.txt')] ;

  • 返回值args是个列表,包含那些没有‘-’或‘--’的参数,比如:['middle.txt','final.txt']

  • "hi:o:"
    短格式 h 后面没有冒号:表示后面不带参数,i和 o后面有冒号表示后面需要参数

  • ["help","ifile=","ofile="]
    长格式 help后面没有等号=,表示后面不带参数,其他两个后面有=,表示后面需要参数

注意:定义命令行参数时,要先定义带'-'选项的参数,再定义没有‘-’的参数

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • linux资料总章2.1 1.0写的不好抱歉 但是2.0已经改了很多 但是错误还是无法避免 以后资料会慢慢更新 大...
    数据革命阅读 12,203评论 2 33
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,854评论 18 139
  • getopt:和C中的getopt()等价。optparse:2.7后已不推荐使用。argparse:基于optp...
    顾慎为阅读 8,729评论 1 10
  • 哪里有它哪里就有欢乐熊本县de吉祥物Kumamon靠着蠢萌搞怪的个性俘获了全人类这大概是本世纪最蠢萌蠢萌的吉祥物了...
    WankaTravel阅读 567评论 0 0
  • 时针指在四点钟方向,对面楼梯间的灯光都亮着。外面却是漆黑一片。城里的人都还在睡梦中,也许有些早起的环卫工人拿着扫帚...
    雪过天晴_99a2阅读 360评论 0 0