使用 argparse 调用函数并在函数中使用参数 - Python

Python中的 argparse 模块可以轻松解析命令行参数，并将它们传递给你的Python脚本中的函数。

安装 argparse

argparse 可以在 Python 中的任何版本中使用，因为它是标准库的一部分。因此，你不需要额外安装任何东西。

使用 argparse

要使用 argparse，请按照以下步骤进行：

步骤 1：导入 argparse

import argparse

步骤 2：创建 ArgumentParser

ArgumentParser 是 argparse 的中心类。它可以处理解析器的所有操作。

parser = argparse.ArgumentParser()

步骤 3：添加参数

要添加参数，请使用 add_argument 方法。它接受多个参数，包括参数名称，帮助文本和默认值。例如，下面的代码添加了一个名为 name 的字符串类型参数。

parser.add_argument('--name', type=str, help='enter your name', default='World')

步骤 4：解析参数

args = parser.parse_args()

步骤 5：使用参数

现在你可以像使用任何其他变量一样使用它们。

print(f'Hello, {args.name}!')

完整代码

import argparse

def hello(name):
    print(f'Hello, {name}!')

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--name', type=str, help='enter your name', default='World')
    args = parser.parse_args()
    hello(args.name)

if __name__ == '__main__':
    main()

运行程序

python hello.py --name Alice

输出：

Hello, Alice!

结论

argparse 模块是一个强大的工具，可以在捕获命令行参数时提供帮助，并提供每个参数的默认值，这使得编写交互式Python脚本更加容易。