python获取命令行参数实例方法讲解

下面是Python获取命令行参数的实例方法攻略：

目录

• 为什么需要命令行参数？
• 获取命令行参数的三种方法
• 方法一：sys.argv
• 方法二：argparse
• 方法三：click
• 示例1：sys.argv的使用
• 示例2：argparse的使用

为什么需要命令行参数？

命令行参数指的是在命令行中输入程序执行命令时，传递给程序的参数。比如在命令行执行 python test.py arg1 arg2 命令时，arg1 和 arg2 就是命令行参数。

在程序设计中，命令行参数非常重要。它使得程序可以根据用户传递的参数，动态改变行为。比如一个命令行工具，可以通过命令行参数控制输出内容、输出路径、输出格式等等。

获取命令行参数的三种方法

Python中获取命令行参数主要有三种方法：sys.argv、argparse和click。下面分别进行介绍：

方法一：sys.argv

sys.argv是Python内置的标准库，用于获取命令行参数。它是一个列表，包含了命令行中所有传入的参数。其中，sys.argv[0] 表示程序本身，sys.argv[1:] 表示命令行参数。

下面是sys.argv的使用示例：

import sys

def main():
    print("程序名称:", sys.argv[0])
    print("命令行参数为:")
    for arg in sys.argv[1:]:
        print(arg)

if __name__ == '__main__':
    main()

在命令行执行 python test.py arg1 arg2 后，输出结果如下：

程序名称: test.py
命令行参数为:
arg1
arg2

方法二：argparse

argparse是Python标准库，用于解析命令行参数。相比于sys.argv，argparse提供了更加丰富的功能，支持自定义解析规则、提供帮助文档等等。

下面是argparse的使用示例：

import argparse

def main():
    parser = argparse.ArgumentParser(description='这是一个命令行参数解析程序')
    parser.add_argument('arg1', help='第一个参数')
    parser.add_argument('arg2', help='第二个参数')
    parser.add_argument('--arg3', '-a', help='第三个参数')

    args = parser.parse_args()

    print("arg1:", args.arg1)
    print("arg2:", args.arg2)
    print("arg3:", args.arg3)

if __name__ == '__main__':
    main()

在命令行执行 python test.py arg1 arg2 -a arg3_value 后，输出结果如下：

arg1: arg1
arg2: arg2
arg3: arg3_value

方法三：click

click是一个Python第三方库，用于构建命令行程序。相比于argparse，click更加方便和易用，同时提供了更多的功能。

下面是click的使用示例：

import click

@click.command()
@click.argument('arg1')
@click.argument('arg2')
@click.option('--arg3', '-a', help='第三个参数')
def main(arg1, arg2, arg3):
    print("arg1:", arg1)
    print("arg2:", arg2)
    print("arg3:", arg3)

if __name__ == '__main__':
    main()

在命令行执行 python test.py arg1 arg2 -a arg3_value 后，输出结果如下：

arg1: arg1
arg2: arg2
arg3: arg3_value

示例1：sys.argv的使用

假设有一个脚本 read_file.py，用于读取文件中的内容并打印出来。文件的路径是通过命令行参数传递的。

import sys

def read_file(file_path):
    with open(file_path, 'r') as f:
        contents = f.read()
        print(contents)

def main():
    if len(sys.argv) < 2:
        print("请传入文件路径")
        return

    file_path = sys.argv[1]
    read_file(file_path)

if __name__ == '__main__':
    main()

在命令行执行 python read_file.py /path/to/file.txt 后，会输出文件的内容。

示例2：argparse的使用

假设有一个脚本 download_file.py，用于从某个URL下载文件到本地。文件的URL地址和本地路径是通过命令行参数传递的。

import argparse
import urllib.request

def download_file(url, file_path):
    urllib.request.urlretrieve(url, file_path)

def main():
    parser = argparse.ArgumentParser(description='这是一个下载文件程序')
    parser.add_argument('url', help='文件的URL地址')
    parser.add_argument('file_path', help='本地文件路径')

    args = parser.parse_args()

    url = args.url
    file_path = args.file_path

    download_file(url, file_path)

if __name__ == '__main__':
    main()

在命令行执行 python download_file.py http://example.com/file.txt /path/to/file.txt 后，会下载文件并保存到本地路径。