Python添加命令行参数步骤

第Python添加命令行参数步骤目录前言添加命令行参数的一般步骤命令行参数示例

前言

许多任务程序如果为其构造为一个命令行界面，就可以通过接受不同的参数来改变它的工作方式。例如，在爬虫程序中，不同URL通常可以作为命令行参数传递给任务程序，从而可以爬取不同网页中的数据。在Python标准库中包含一个强大的argparse模块，可以轻松创建丰富的命令行参数解析。

添加命令行参数的一般步骤

在程序脚本中，argparse的基本使用方式可以分三个步骤显示：

定义脚本要接受的参数，生成新的参数解析器调用定义的解析器，返回一个包含所有结果参数的对象使用带有参数的命令调用脚本，使得定义的参数行为生效

一般而言，使用命令行参数的脚本遵循以下结构：

导入相关库

定义函数

定义参数解析器

对参数进行操作

使用参数调用函数，以执行不同操作

其中，一般使用main函数来显式的声明代码的执行入口，简单情况下，我们可以直接使用命令行传入的参数而无需进行复杂处理。

命令行参数示例

接下来，我们通过一个示例来了解如何添加并使用命令行参数。首先创建一个脚本argparse_example.py，该脚本接受单个整数作为位置参数，并打印执行次数的Helloworld.。脚本代码如下，我们遵循上述结构，其中定义的main仅仅执行打印功能：

importargparse

defmain(num):

print('Helloworld.\n'*num)

if__name__=='__main__':

parse=argparse.ArgumentParser()

parse.add_argument('number',type=int,help='Anumber')

args=parse.parse_args()

main(args.number)

在脚本中，参数通过add_arguments添加到解析器。定义了所有参数之后，调用parse_args()将返回一个包含参数解析结果的对象，如果出现错误则退出执行。

调用脚本并检查参数的使用方式，不带参数调用脚本时会自动显示帮助信息，使用参数-h可以显示扩展帮助信息：

$pythonargparse_example.py

usage:argparse_example.py[-h]number

argparse_example.py:error:thefollowingargumentsarerequired:number

$pythonargparse_example.py-h

usage:argparse_example.py[-h]number

positionalarguments:

number

Anumber

optionalarguments:

-h,--help

showthishelpmessageandexit

使用额外参数调用脚本，可以按程序预期执行脚本：

$pythonargparse_example.py3

Helloworld.

Helloworld.

Helloworld.

$pythonargparse_example.pythree

usage:argparse_example.py[-h]number

argparse_example.py:error:argumentnumber:invalidintvalue:'three'

更改脚本，添加可选参数用于接受所需打印的字符串，且默认值为Helloworld，编写argparse_example_2.py脚本如下所示：

importargparse

defmain(string,num):

print(string*num)

if__name__=='__main__':

parser=argparse.ArgumentParser()

parser.add_argument('num',type=int,help='Anumber')

parser.add_argument('-s',type=str,help='Stringtoprint',default='Helloworld.\n')

args=parser.parse_args()

main(args.s,args.num)

再次执行脚本，可以看到帮助已经更新，使用-s选项可以打印不同的字符串：

$pythonargparse_example_2.py-h

usage:argparse_example_2.py[-h][-sS]num

positionalarguments:

num

Anumber

optionalarguments:

-h,--help

showthishelpmessageandexit

-sS

Stringtoprint

$pythonargparse_example_2.py3Helloworld.

Helloworld.

Helloworld.

$pythonargparse_example_2.py3-s'Ilovemovie'

IlovemovieIlovemovieIlovemovie

在实际场景中，我们应当为每个参数都添加一个帮助描述信息，参数可以分为位置参数和可选参数，它们的使用方式具有较大差别：

如果参数以-开头，则将其视为可选参数，例如脚本中的-s参数。否则，该参数为位置参数，如脚本中的数字参数为了清楚起见