每天一个 Python 库:argparse 参数解析,轻松搞定脚本配置!

在开发 Python 脚本时,命令行参数的解析是一个常见而又重要的操作。无论是处理用户输入的文件路径,还是接受参数配置,argparse 都能轻松帮你搞定。

如果你还不熟悉 argparse,本文将带你一步步掌握!


argparse 简介

argparse 是 Python 标准库中的模块,专门用于解析命令行参数,它让你可以轻松处理命令行输入,并根据用户输入执行不同的操作。

常见应用场景:

  • 配置脚本:接收不同的参数来执行不同的操作。
  • 程序优化:通过命令行传递配置参数,避免硬编码。


基本用法:命令行参数的定义与解析


学习本来就不是一蹴而就的事,不过只要你肯练、敢用,坚持一阵子,你一定能看到变化!


首先,我们来看看最简单的命令行参数解析。我们定义了一个参数 --name 和 --age,来接收用户输入的名字和年龄。

示例代码:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project :Fish
@File    :D12.py
@Date    :2025/4/24
@Author  : malijie
"""
import argparse

# 创建解析器
parser = argparse.ArgumentParser(description="输入你的个人信息")
parser.add_argument('--name', type=str, help="输入你的名字")
parser.add_argument('--age', type=int, help="输入你的年龄")

# 解析命令行参数
args = parser.parse_args()

# 打印结果
print(f"你好,{args.name},你的年龄是 {args.age}")

执行示例:

$ python your_script.py --name "Jay" --age 25
你好,Jay,你的年龄是 25

高级用法:必选参数与默认值

除了简单的命令行参数,argparse 还支持必选参数、默认值和不同类型的参数。

示例 1:必选参数

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project :Fish
@File    :D12.py
@Date    :2025/4/24
@Author  : malijie
"""
import argparse

# 创建解析器
parser = argparse.ArgumentParser(description="计算两个数字的和")
parser.add_argument('num1', type=int, help="第一个数字")
parser.add_argument('num2', type=int, help="第二个数字")

# 解析命令行参数
args = parser.parse_args()

# 计算和并输出
print(f"{parser.description}:{args.num1} + {args.num2} = {args.num1 + args.num2}")

执行示例:

我们少传个参数看看:


常用技巧:添加帮助信息和参数类型

argparse 不仅能帮你解析参数,还能自动生成帮助信息,指导用户如何使用你的脚本。

示例 2:自动生成帮助信息

import argparse

# 创建解析器
parser = argparse.ArgumentParser(description="计算两数之和")
parser.add_argument('num1', type=int, help="第一个数字")
parser.add_argument('num2', type=int, help="第二个数字")

# 解析命令行参数
args = parser.parse_args()

# 计算和并输出
print(f"{args.num1} + {args.num2} = {args.num1 + args.num2}")

执行 python D12.py --help 后,自动显示帮助信息:



argparse 进阶技巧:互斥参数与可选参数

有时你可能需要设置互斥的参数,确保同一时间只能传递其中一个参数。例如,我们不希望用户同时传递 --name 和 --age。

示例 3:互斥参数

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project :Fish
@File    :D12.py
@Date    :2025/4/24
@Author  : malijie
"""
import argparse

# 创建解析器
parser = argparse.ArgumentParser(description="输入你的名字或者年龄")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--name', type=str, help="输入你的名字")
group.add_argument('--age', type=int, help="输入你的年龄")

# 解析命令行参数
args = parser.parse_args()

# 输出结果
if args.name:
    print(f"你的名字是 {args.name}")
else:
    print(f"你的年龄是 {args.age}")

执行示例:



总结:argparse 模块的核心能力

  1. 基本参数解析:支持必选参数和可选参数的解析。
  2. 帮助信息:自动生成帮助信息,提升用户体验。
  3. 互斥组:确保在命令行中使用互斥的参数。
  4. 默认值:为参数指定默认值。

使用场景:什么时候用 argparse?

  • 自动化脚本:你可以通过命令行传递参数控制脚本行为。
  • 配置管理:传递配置文件路径、设置调试级别等。
  • 数据处理:为数据分析脚本提供输入参数。

下期预告:numpy 数值运算,超高性能的数据处理工具!


点关注点赞不迷路,后续更新更多自动化测试经验,我们一起成长!


欢迎评论区留言你最常用的 argparse 场景或踩过的坑~

点击头像查看往期精彩文章!

相关文章

一文掌握怎么利用Python+Shell构建一个自动化的数据处理系统

简介:本文系统讲解了如何利用Python与Shell脚本相结合,构建一个高效、自动化的数据处理系统。文章从项目结构设计、核心功能实现到Web管理界面的搭建,详细介绍了数据采集、清洗、分析、入库和报表展...

Python脚本保护:如何守护你的代码安全

在当今数字化时代,Python因其简洁易用的特性,成为众多开发者首选的编程语言。然而,随着Python应用的广泛普及,代码安全问题也日益凸显。如何保护Python脚本,防止源代码被轻易窃取或逆向工程,...

自动化文件管理的6个Python脚本模板

文件管理是我们日常工作中常见的任务之一,Python作为一种功能强大的编程语言,可以帮助我们轻松完成这些任务。今天我们来学习6个实用的Python脚本模板,让自动化文件管理工作变得简单又高效。1. 获...

Python脚本处理Info.plist文件

打包或生成工程项目的时候,我们常常需要给一些插件或者接入的SDK修改Info.plist文件,一些权限又或者设置等等,为了避免忘记和节省时间,我们可以结合python脚本执行处理。1、首先安装一个pl...

使用Python编写手机短信压测脚本

在现代应用中,短信验证是常见的功能之一,尤其在用户注册、密码重置等场景下。然而,随着用户量的增加,短信服务的稳定性和性能也变得至关重要。因此,对短信发送服务进行压力测试是非常必要的。本文将介绍如何使用...