python 脚本获取传参

原创

mob649e815b5994 2023-12-22 07:38:52 ©著作权

文章标签 Python python 命令行 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e815b5994的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python脚本获取传参

在Python开发中，我们经常需要通过命令行为脚本传递参数。这些参数可以用于控制脚本的行为，使脚本更加灵活和可配置。本文将介绍如何使用Python脚本获取传参的方法，并附带代码示例。

为什么需要获取传参

脚本的参数可以用于以下几个方面：

配置参数：通过命令行传参可以设置脚本的配置参数，如文件路径、数据库连接等。
控制选项：参数可以用于控制脚本的执行选项，如是否启用某个功能、是否输出调试信息等。
数据输入：通过参数可以将数据输入脚本，而不需要手动输入或者通过文件读取。

获取传参的方法对于脚本的可扩展性和重用性非常重要。

获取传参的方法

Python提供了多种方法来获取传参，我们将介绍以下几种常用的方法：

位置参数：位置参数是最常见的获取传参的方式。位置参数是指按照参数的顺序传递值给脚本，脚本可以通过位置来确定每个参数的含义。以下是一个简单的示例：

import sys

if len(sys.argv) != 3:
    print("Usage: python script.py <arg1> <arg2>")
    sys.exit(1)

arg1 = sys.argv[1]
arg2 = sys.argv[2]

print(f"arg1: {arg1}")
print(f"arg2: {arg2}")

在这个示例中，我们使用了sys.argv来获取命令行参数列表。sys.argv[0]是脚本本身的名称，从sys.argv[1]开始才是传递给脚本的参数。在这个示例中，我们期望传递两个参数给脚本。

可选参数：有时候我们希望某些参数是可选的，可以通过设置默认值来实现。Python的argparse模块提供了强大的命令行参数解析功能。以下是一个使用argparse的示例：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--arg1', type=int, default=100, help='Argument 1')
parser.add_argument('--arg2', type=float, default=3.14, help='Argument 2')

args = parser.parse_args()

print(f"arg1: {args.arg1}")
print(f"arg2: {args.arg2}")

在这个示例中，我们使用了argparse.ArgumentParser创建一个参数解析器。然后使用add_argument方法来定义参数的名称、类型、默认值和帮助信息。通过parse_args方法来解析传递给脚本的参数。

键值对参数：有时候我们希望传递一组键值对参数，可以使用--key=value的形式传递参数。Python的argparse模块也可以解析这种参数。以下是一个使用argparse解析键值对参数的示例：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--config', type=str, required=True, help='Config file')
parser.add_argument('--output', type=str, default='output.txt', help='Output file')

args = parser.parse_args()

print(f"config: {args.config}")
print(f"output: {args.output}")

在这个示例中，我们定义了--config和--output两个参数。--config参数是必须的，而--output参数有一个默认值。通过--config=config.ini和--output=result.txt的形式来传递参数。