Published on

Python获取脚本在命令行中运行参数的三种方法

Authors
  • Name
    Twitter

如果您在学习的过程中有任何疑问, 都是因为我没有表述清楚, 请在评论区告诉我 :)

方案一

pip install checkopt //终端中运行

import checkopt
print(checkopt.checkopt("help m|module="))

(venv) PS D:\python> python .\test.py -m module_name --help project_name
({'module': 'module_name', 'help': None}, ['project_name'])

运行checkopt.checkopt()后会返回一个字典选项和一个额外数据的数组 如果匹配的一个模式中同时给出了短选项和长选项(m|module), 运行py代码时如果给的是短选项(-m),会自动解析为长选项(module)

用法:字典变量,数组变量=checkopt.checkopt(要匹配的格式)

函数参数:匹配模式字符串 例子g help m|module=* 空格用来分割多个选项 一个字母如"g"在命令行中就是python a.py -g 多个字母如help在命令行中就是python a.py --help 竖杠 | 代表二者等价,竖杠前是短选项(-m),竖杠后是长选项(--module) 等号表示该标志后面必须跟一个数据 星号(*)表示允许接收多个参数

综上,以上匹配模式的一个实际运行演示:

options,arguments = checkopt.checkopt("g= help m|module=*")

(venv) PS D:\python> python .\test.py -m module_name1 --module module_name2 --help extra_data
options:
	{'module': ['module_name1', 'module_name2'], 'help': None}
arguments:
	['extra_data']

方案二 getopt

import sys
import getopt

opts, args = getopt.getopt(sys.argv[1:], 'm:', ['add='])

函数接收三个参数

  • 第一个是需要被解析的数组. sys.arg 第一个是python文件名, 所以从第二个开始取
  • 第二个是对单字符选项进行解析的字符串, 里面定义了哪些选项能被解析以及是否需要参数, 其中每个英文字母代表一个选项. 如果一个选项需要参数则后面跟一个冒号
  • 第三个是长选项(例如--help), 如果需要参数则后面跟一个等号

如果需要将短选项自动解析为长选项, (例如-m解析为--module,后续可以直接if "module" in options), 请使用方案一.

方案三 sys.argv

import sys
print(sys.argv)

运行后大概是这个样子 ['D:\demo.py', '-m', 'module_name1', '-m', 'module_name2', 'extra_data', '-f', '123'] 这种方案最大的缺点就是需要手动处理参数与顺序