详解python中的三种命令行模块(sys.argv,argparse,click)


Posted in Python onDecember 15, 2020

Python作为一门脚本语言,经常作为脚本接受命令行传入参数,Python接受命令行参数大概有三种方式。因为在日常工作场景会经常使用到,这里对这几种方式进行总结。

命令行参数模块

这里命令行参数模块平时工作中用到最多就是这三种模块:sys.argv,argparse,click。sys.argv和argparse都是内置模块,click则是第三方模块。

sys.argv模块(内置模块)

先看一个简单的示例:

#!/usr/bin/python
import sys

def hello(name, age, sex, *args):
  print("Hello, My name is {name}.".format(name=name))
  print("I'm {age} years old.".format(age=age))
  print("I'm a {sex}".format(sex=sex))

  print("Other word:\n{args}".format(args="\n".join(args)))


if __name__ == "__main__":
  file_name = sys.argv[0]
  name = sys.argv[1]
  age = sys.argv[2]
  sex = sys.argv[3]
  other = sys.argv[4:]
  hello(name, age, sex, *other)

调用脚本:

python test_sysargv.py zhangsan 13 man nibi ss

脚本输出:

Hello, My name is zhangsan.
I'm 13 years old.
I'm a man
Other word:
nibi
ss

sys.argv模块不难理解,命令参数作为列表传入Python脚本中,argv[0]是脚本的名字,argv[1]则是第一个参数,后面以此类推。所以在脚本中只需要提取列表中的参数即可使用。上面演示的是正确调用Python脚本的情况,下面则是调用失败的情况。

错误调用脚本:

python test_sysargv.py zhangsan 13

错误输出:

Traceback (most recent call last):
 File "test_sysargv.py", line 16, in <module>
  sex = sys.argv[3]
IndexError: list index out of range

关于错误也很好理解,经典的列表索引超出范围,之所列表索引超出范围,没有传入足够的参数。当然你可以使用try...except捕获错误。但是这种做法太过死板,因为在命令行中必须按照脚本规定的参数顺序输入参数,所以这种模块使用一般是针对一些需要的参数比较少并且固定的脚本。

argparse模块(内置模块)

同样的先看一个简单的示例:

#!/usr/bin/python
import argparse

parser = argparse.ArgumentParser(description='Test for argparse module.')  # 构建命令参数实例
parser.add_argument("--name", "-n", help="name attribute: 非必要属性")
parser.add_argument("--age", "-a", help="age attribute: 非必要属性")
parser.add_argument("--sex", "-s", help="sex attribute: 非必要属性")
parser.add_argument("--type", "-t", help="type attribute: 非必要属性", required=True)
args = parser.parse_args()


def hello(name, age, sex, *args):
  print("Hello, My name is {name}.".format(name=name))
  print("I'm {age} years old.".format(age=age))
  print("I'm a {sex}".format(sex=sex))

  print("Other word:\n{args}".format(args="\n".join(args)))


if __name__ == "__main__":
  print("Format of transfer file: {type}".format(type=args.type))
  if args.name and args.age and args.sex:
    hello(args.name, args.age, args.sex)

执行脚本:

python3 test_argparse.py -t json -n zhangsan -a 13 -s man

脚本成功输出:

Format of transfer file: json
Hello, My name is zhangsan.
I'm 13 years old.
I'm a man
Other word:

关于argparse模块的使用,首先需要生成一个命令行参数的实例,然后通过对这个对象添加属性,添加需要从命令行获取的参数,包括哪些是必要参数(required=True),哪些非必要参数,同时也可以对每个参数进行帮助提示(help="")。

而上面示例中分别添加了四个属性,--name和-n同时可以再命令行中使用,都表示了参数name。ArgumentParser通过parse_ags()方法解析参数,检查命令行,将每个参数转换为适当的类型,所以在脚本中同时也可以使用args.n和args.name获取到参数,相对应的如果没有传入该参数,脚本中则获取到None。

查看命令行参数之后脚本的帮助提示:

python3 test_argparse.py -h
usage: test_argparse.py [-h] [--name NAME] [--age AGE] [--sex SEX] --type TYPE

Test for argparse module.

optional arguments:
 -h, --help      show this help message and exit
 --name NAME, -n NAME name attribute: 非必要属性
 --age AGE, -a AGE   age attribute: 非必要属性
 --sex SEX, -s SEX   sex attribute: 非必要属性
 --type TYPE, -t TYPE type attribute: 非必要属性

另外在添加命令行参数的属性中,还有更多的设置,多余的可以参考Python官方文档,里面都有详细的标注,这里就不展开来讲,总结的就是关于argparse模块的使用非常简便,同时十分人性化,也很符合日常工作的需要。

click模块

先开一个简单的使用示例:

#!/usr/bin/python
import click

@click.command()
@click.option("--name", default="zhangsan", help="name attribute: 非必要属性")
@click.option("--age", help="age attribute", type=int)
@click.option("--sex", help="sex attribute")
@click.option("-t", help="type attribute: 必要属性", required=True)
def hello(t, name, age, sex, *args):
  print("Format of transfer file: {type}".format(type=t))
  print("Hello, My name is {name}.".format(name=name))
  print("I'm {age} years old.".format(age=age))
  print("I'm a {sex}".format(sex=sex))

  print("Other word:\n{args}".format(args="\n".join(args)))


if __name__ == "__main__":
  hello()

执行脚本:

python3 test_click.py -t 1 --age 13 --sex man

脚本输出:

Format of transfer file: 1
Hello, My name is zhangsan.
I'm 13 years old.
I'm a man
Other word:

click模块是Flask团队优秀的开源项目,使用方法和argparse模块很相似,同样为命令行封装了大量的方法,使用者只需要专注代码功能的实现。
click模块和argparse模块不同的地方就是,click模块使用装饰器的方式给函数添加命令行属性,关于装饰器简单来讲就是能够在不修改原有函数的基础上添加功能。虽然使用装饰器但是添加命令行属性的方式和argparse模块很相似,包括options中常用的参数含义也有很多类似的地方。值得注意的就是一开始需要通过command()将函数成为命令行的接口。
关于的click模块的就大致讲到这里,其余有兴趣的可以再去了解一下。

总结

关于这三个模块值得注意的是,尽量贴近自己应用场景去选择,真正的做到自己能够方便使用才是你去使用这些模块的原因。

以上就是详解python中的三种命令行模块(sys.argv,argparse,click)的详细内容,更多关于python 命令行模块的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
Python通过websocket与js客户端通信示例分析
Jun 25 Python
在Gnumeric下使用Python脚本操作表格的教程
Apr 14 Python
Python使用BeautifulSoup库解析HTML基本使用教程
Mar 31 Python
Python Paramiko模块的安装与使用详解
Nov 18 Python
Python实现PS图像明亮度调整效果示例
Jan 23 Python
python实现银行管理系统
Oct 25 Python
Python使用Opencv实现图像特征检测与匹配的方法
Oct 30 Python
pytorch 模型的train模式与eval模式实例
Feb 20 Python
selenium+python配置chrome浏览器的选项的实现
Mar 18 Python
python中append函数用法讲解
Dec 11 Python
python实现简单的学生管理系统
Feb 22 Python
Python绘制K线图之可视化神器pyecharts的使用
Mar 02 Python
python 基于Apscheduler实现定时任务
Dec 15 #Python
next在python中返回迭代器的实例方法
Dec 15 #Python
python中slice参数过长的处理方法及实例
Dec 15 #Python
关于django python manage.py startapp 应用名出错异常原因解析
Dec 15 #Python
Window10上Tensorflow的安装(CPU和GPU版本)
Dec 15 #Python
详解python中的异常捕获
Dec 15 #Python
python 6种方法实现单例模式
Dec 15 #Python
You might like
PHP 分页原理分析,大家可以看看
2009/12/21 PHP
细谈php中SQL注入攻击与XSS攻击
2012/06/10 PHP
自己开发Dojo的建议框架
2008/09/24 Javascript
Jquery颜色选择器ColorPicker实现代码
2012/11/14 Javascript
Javascript 数组排序详解
2014/10/22 Javascript
JS基础随笔(菜鸟必看篇)
2016/07/13 Javascript
使用plupload自定义参数实现多文件上传
2016/07/19 Javascript
原生js实现旋转木马轮播图效果
2017/02/27 Javascript
Bootstrap组合上、下拉框简单实现代码
2017/03/06 Javascript
vue.js的安装方法
2017/05/12 Javascript
AngularJS实现表格的增删改查(仅限前端)
2017/07/04 Javascript
React入门教程之Hello World以及环境搭建详解
2017/07/11 Javascript
jQuery利用FormData上传文件实现批量上传
2018/12/04 jQuery
vue 地图可视化 maptalks 篇实例代码详解
2019/05/21 Javascript
小程序选项卡以及swiper套用(跨页面)
2020/06/19 Javascript
Vue多选列表组件深入详解
2021/03/02 Vue.js
一则python3的简单爬虫代码
2014/05/26 Python
一个小示例告诉你Python语言的优雅之处
2014/07/04 Python
Python中用Descriptor实现类级属性(Property)详解
2014/09/18 Python
Python基础教程之正则表达式基本语法以及re模块
2016/03/25 Python
python实现12306抢票及自动邮件发送提醒付款功能
2018/03/08 Python
Python 字符串换行的多种方式
2018/09/06 Python
python随机在一张图像上截取任意大小图片的方法
2019/01/24 Python
selenium获取当前页面的url、源码、title的方法
2019/06/12 Python
简单了解python gevent 协程使用及作用
2019/07/22 Python
对python 树状嵌套结构的实现思路详解
2019/08/09 Python
Python 获取numpy.array索引值的实例
2019/12/06 Python
Python简单实现区域生长方式
2020/01/16 Python
pip 20.3 新版本发布!即将抛弃 Python 2.x(推荐)
2020/12/16 Python
python利用opencv实现颜色检测
2021/02/23 Python
基于HTML5的WebGL经典3D虚拟机房漫游动画
2017/11/15 HTML / CSS
铁路个人事迹材料
2014/01/30 职场文书
优秀女职工事迹材料
2014/02/06 职场文书
激情洋溢的毕业生就业求职信
2014/03/15 职场文书
离婚协议书怎样才有法律效力
2014/10/10 职场文书
详解Nginx启动失败的几种错误处理
2021/04/01 Servers