用Argparse让代码命令行化

2023-01-21 11:20:19

探索 Python 中 Argparse：命令行参数解析利器

在 Python 中驾驭命令行参数从来都不是一件容易的事，直到 Argparse 的出现。这个标准库提供了处理命令行参数的强大工具集，让编写复杂命令行工具变得轻而易举。

Argparse 的魅力

Argparse 的魅力在于它将命令行参数无缝解析为 Python 变量，赋予您对应用程序参数的全面控制。它不仅使用简单，而且功能强大，足以应对各种复杂的参数场景。

使用方法

使用 Argparse 是直截了当的。首先，创建一个 ArgumentParser 对象，它将负责解析参数。然后，使用 add_argument() 方法逐个添加参数，指定名称、类型和帮助信息。

import argparse

# 创建ArgumentParser对象
parser = argparse.ArgumentParser()

# 添加参数
parser.add_argument("-f", "--file", help="The input file")
parser.add_argument("-o", "--output", help="The output file")
parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbose output")

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

通过调用 parse_args() 方法，您可以访问命令行参数并将其存储在 args 对象中。

技巧

Argparse 提供了一些实用技巧，让您进一步定制参数处理：

nargs 参数： 指定参数可以接受多个值。
choices 参数： 限制参数值范围。
metavar 参数： 在帮助信息中指定参数的显示名称。

示例

让我们看看一个更高级的示例：

import argparse

# 创建ArgumentParser对象
parser = argparse.ArgumentParser()

# 添加参数
parser.add_argument("-f", "--file", nargs="+", help="The input files")
parser.add_argument("-o", "--output", help="The output file")
parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbose output")
parser.add_argument("-c", "--choices", choices=["a", "b", "c"], help="The choice")

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