Python教程——为脚本添加运行参数

在日常工作中可以看到所有命令都是可以使用不同的参数来实现不同的功能，而自己编写的Python脚本同样也支持在运行的始终使用定义参数，比如

python3 --help
python3 --noargs
python3 a.py b.py c.py

在Python中可以通过argparse这个函数库实现命令行选项解释器，该函数库需要Python 3.2以上才支持

import argparse
parser = argparse.ArgumentParser(description="该程序用来演示参数处理")  #使用--help的时候就会显示出这里的信息以及每个参数的描述

#使用"--"表示增加一个可选参数，不加--表示必选参数
parser.add_argument("number1",help="请输入一个数字")
parser.add_argument("--number2",help="请输入一个数字")

#强制参数类型和设置默认值
parser.add_argument("--number3",type=int,default=0,help="请输入一个数字")

args = parser.parse_args()
print(f'你输入的数字是{args.number}')

执行方式

cat test.py
import argparse
parser = argparse.ArgumentParser(description="这是一个测试程序")
parser.add_argument("--number1",type=int,default=1,help="请输入一个数字")
parser.add_argument("--number2",type=int,default=2,help="请输入一个数字")
args = parser.parse_args()
print(f"你输入的两数相加结果为{args.number1+args.number2}")

python3 test.py 3 5 #输出8

查看帮助信息

argsument.py --help 
usage: argsument.py [-h] [--number1 NUMBER1] [--number2 NUMBER2]

这是一个测试程序

optional arguments:
  -h, --help         show this help message and exit
  --number1 NUMBER1  请输入一个数字
  --number2 NUMBER2  请输入一个数字

进程已结束,退出代码0