python3如何查看命令行参数

在Python3中查看命令行参数的方法有多种，其中最常用的方法包括使用sys模块、argparse模块和click模块。sys模块简单易用、argparse模块功能强大、click模块适用于构建复杂的命令行工具。接下来，我们将详细介绍这几种方法及其使用场景。

一、使用sys模块

sys模块是Python的标准库之一，提供了一些与Python解释器交互的功能。通过sys.argv，我们可以获取命令行参数。

1.1 获取命令行参数

sys.argv是一个列表，包含了所有的命令行参数。第一个元素是脚本的名称，后面的元素是传递给脚本的参数。

import sys
def main():
    # 获取所有命令行参数
    args = sys.argv
    # 打印脚本名称
    print(f"脚本名称: {args[0]}")
    # 打印所有参数
    for i, arg in enumerate(args[1:], start=1):
        print(f"参数{i}: {arg}")
if __name__ == "__main__":
    main()

1.2 解析命令行参数

sys.argv只提供了基本的功能，通常需要手动解析参数。对于简单的参数解析，可以使用列表操作。

import sys
def main():
    args = sys.argv
    if len(args) < 3:
        print("Usage: script.py <arg1> <arg2>")
        sys.exit(1)
    arg1 = args[1]
    arg2 = args[2]
    print(f"参数1: {arg1}")
    print(f"参数2: {arg2}")
if __name__ == "__main__":
    main()

二、使用argparse模块

argparse模块是Python内置的命令行解析模块，适用于处理复杂的命令行参数。它可以自动生成帮助信息，并支持多种类型的参数。

2.1 基本用法

argparse模块使用ArgumentParser类来解析命令行参数。

import argparse
def main():
    parser = argparse.ArgumentParser(description="命令行参数解析示例")
    # 添加参数
    parser.add_argument("arg1", type=str, help="第一个参数")
    parser.add_argument("arg2", type=int, help="第二个参数")
    # 解析参数
    args = parser.parse_args()
    print(f"参数1: {args.arg1}")
    print(f"参数2: {args.arg2}")
if __name__ == "__main__":
    main()

2.2 可选参数

除了位置参数，argparse还支持可选参数。

import argparse
def main():
    parser = argparse.ArgumentParser(description="命令行参数解析示例")
    # 添加位置参数
    parser.add_argument("arg1", type=str, help="第一个参数")
    # 添加可选参数
    parser.add_argument("-o", "--option", type=str, default="default_value", help="可选参数")
    # 解析参数
    args = parser.parse_args()
    print(f"参数1: {args.arg1}")
    print(f"可选参数: {args.option}")
if __name__ == "__main__":
    main()

三、使用click模块

click模块是一个用于创建命令行界面的第三方库，具有更简洁的语法和更强大的功能。它适用于构建复杂的命令行工具。

3.1 基本用法

click模块使用装饰器来定义命令行参数。

import click
@click.command()
@click.argument("arg1")
@click.argument("arg2")
def main(arg1, arg2):
    print(f"参数1: {arg1}")
    print(f"参数2: {arg2}")
if __name__ == "__main__":
    main()

3.2 可选参数

click模块也支持可选参数。

import click
@click.command()
@click.argument("arg1")
@click.option("--option", default="default_value", help="可选参数")
def main(arg1, option):
    print(f"参数1: {arg1}")
    print(f"可选参数: {option}")
if __name__ == "__main__":
    main()