python main 入参

原创

mob649e8153b214 2023-11-01 04:31:41 ©著作权

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

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

如何实现"python main 入参"

流程图

flowchart TD
    A[定义命令行参数] --> B[解析命令行参数]
    B --> C[使用命令行参数]

表格展示步骤

步骤	描述
1	定义命令行参数
2	解析命令行参数
3	使用命令行参数

步骤详解

步骤1：定义命令行参数

在Python中，我们可以使用argparse模块来定义和解析命令行参数。首先，我们需要导入argparse模块，并创建一个ArgumentParser对象。

import argparse

parser = argparse.ArgumentParser()

步骤2：解析命令行参数

接下来，我们需要定义我们的命令行参数，并解析它们。我们可以使用add_argument方法来定义命令行参数，该方法接受多个参数，包括参数名、参数类型和帮助信息等。

parser.add_argument("name", help="The name of the person")
parser.add_argument("age", type=int, help="The age of the person")

add_argument方法还可以接受一些可选参数，例如default、choices和required等。这些参数可以根据实际需求进行配置。

parser.add_argument("-g", "--gender", choices=["male", "female"], default="male", help="The gender of the person")

然后，我们需要调用parse_args方法来解析命令行参数，并将其保存在一个变量中。

args = parser.parse_args()

步骤3：使用命令行参数

现在，我们已经成功解析了命令行参数，我们可以在我们的代码中使用它们了。args对象是一个命名空间，我们可以通过.运算符来访问其中的参数。

print("Name:", args.name)
print("Age:", args.age)
print("Gender:", args.gender)

完整代码

import argparse

def main():
    # 定义命令行参数
    parser = argparse.ArgumentParser()
    parser.add_argument("name", help="The name of the person")
    parser.add_argument("age", type=int, help="The age of the person")
    parser.add_argument("-g", "--gender", choices=["male", "female"], default="male", help="The gender of the person")
  
    # 解析命令行参数
    args = parser.parse_args()
  
    # 使用命令行参数
    print("Name:", args.name)
    print("Age:", args.age)
    print("Gender:", args.gender)

if __name__ == "__main__":
    main()

代码解析

第3行：导入argparse模块。
第5行：创建一个ArgumentParser对象。
第7-9行：使用add_argument方法定义命令行参数。
- name参数表示参数的名称。
- help参数表示参数的帮助信息。
- type参数表示参数的类型。
- choices参数表示参数的可选值。
- default参数表示参数的默认值。
第14行：调用parse_args方法解析命令行参数，并将其保存在args变量中。
第17-19行：使用命令行参数。
- 通过args.name、args.age和args.gender访问命令行参数的值。
第22行：调用main函数，执行程序。