前言
平常我们在用别人写好的python包的时候,在cmd输入xx -h就能查看到帮助信息,输入xx -p 8080就能把参数传入程序里,看起来非常酷。
本篇就来讲下如何在python代码里加入命令行参数,并且其它功能,能调用这个参数。
argv获取参数
Python 中也可以所用 sys 的 sys.argv 来获取命令行参数:
sys.argv 是命令行参数列表。
len(sys.argv) 是命令行参数个数。
注:sys.argv[0] 表示脚本名。
test.py代码如下
执行以上代码,输出结果为:
getopt模块
getopt模块是专门处理命令行参数的模块,用于获取命令行选项和参数,也就是sys.argv。命令行选项使得程序的参数更加灵活。支持短选项模式(-)和长选项模式(—)。
该模块提供了两个方法及一个异常处理(Exception getopt.GetoptError)来解析命令行参数。
getopt.getopt 方法用于解析命令行参数列表,语法格式如下:
getopt.getopt(args, options[, long_options])
参数说明:
args: 要解析的命令行参数列表。
options: 以列表的格式定义,options后的冒号(:)表示该选项必须有附加的参数,不带冒号表示该选项不附加参数。
long_options: 以字符串的格式定义,long_options 后的等号(=)表示如果设置该选项,必须有附加的参数,否则就不附加参数。
该方法返回值由两个元素组成: 第一个是 (option, value) 元组的列表。 第二个是参数列表,包含那些没有’-‘或’—‘的参数。
实例
假定我们创建这样一个脚本,可以通过命令行向脚本文件传递两个文件名,同时我们通过另外一个选项查看脚本的使用。脚本使用方法如下:
$ test.py -i-o
test.py 文件代码如下所示:
执行以上代码,输出结果为:
实际场景运用
结合selenium测试,比如我想测试chrome浏览器,那就在命令行输入“chrome”参数,想测试firefox浏览器的时候,就在命令行输入“firefox”参数,这样就能灵活切换不同浏览器之间的测试了
cmd执行情况
备注:python2在cmd执行时,中文会显示乱码,用python3就不会有乱码了
领取专属 10元无门槛券
私享最新 技术干货