您的位置:

Python sys.argv: 命令行参数解析

在进行Python脚本开发时,我们经常需要在终端中向程序传递一些参数。Python的sys模块提供了sys.argv方法来获取命令行参数。sys.argv是一个字符串列表,包含了命令行中所有的参数,其中第一个参数是程序本身的名称。本文将从多个方面解析sys.argv的用法和相关注意事项。

一、sys.argv的用法

1、获取命令行参数

import sys

print(sys.argv)

在终端中执行Python脚本时,可以通过以下方式传递参数:

$ python test.py arg1 arg2 arg3

运行结果:

['test.py', 'arg1', 'arg2', 'arg3']

其中,test.py是程序本身的名称,arg1、arg2、arg3是传递的参数。

2、获取指定位置的参数

可以通过sys.argv[index]的方式获取指定位置的参数值,其中index表示参数的索引值,从0开始计数。例如,获取第一个参数的值:

import sys

print(sys.argv[1])

当运行以下命令时:

$ python test.py arg1 arg2 arg3

运行结果:

arg1

二、sys.argv的注意事项

1、sys.argv的参数都是字符串类型

无论传递的参数是数字、布尔值还是其他类型,sys.argv都会将其转换为字符串类型。因此,如果需要使用传递的值进行数值计算或比较操作,需要使用int()、float()等方法进行转换。

例如,传递一个数字参数:

import sys

print(type(sys.argv[1]), sys.argv[1]+1)

运行结果:

   arg11

  

此时的输出结果是字符串类型,因此需要进行类型转换才能进行计算。

import sys

print(type(int(sys.argv[1])), int(sys.argv[1])+1)

运行结果:

   2

  

2、程序本身的名称也算作参数

sys.argv列表中第一个参数是程序本身的名称,如果需要获取传递的参数值,需要从第二个元素开始获取。如果使用sys.argv[0]获取第一个元素,得到的将是程序本身的名称。

例如:

import sys

print(sys.argv[0]) # 输出:test.py
print(sys.argv[1]) # 输出:arg1

3、传递多个参数时的注意事项

如果传递的参数中有空格或其他特殊字符,需要用引号将其括起来。例如,传递包含空格的参数:

$ python test.py "arg with space" arg2 arg3

运行结果:

['test.py', 'arg with space', 'arg2', 'arg3']

如果不使用引号将含有空格的参数括起来,程序将会把其解析成多个参数。

三、sys.argv的示例

以下是一个使用sys.argv的示例,根据传递的参数来执行不同的操作。运行此脚本需要传递一个参数,该参数为“start”或“stop”,start表示启动服务,stop表示停止服务。

import sys

if len(sys.argv) == 2:
    if sys.argv[1] == "start":
        # 启动服务
        print("start service")
    elif sys.argv[1] == "stop":
        # 停止服务
        print("stop service")
    else:
        print("invalid parameter")
else:
    print("usage: python service.py [start|stop]")

当运行以下命令时:

$ python service.py start

运行结果:

start service

当运行以下命令时:

$ python service.py test

运行结果:

invalid parameter

当运行以下命令时:

$ python service.py

运行结果:

usage: python service.py [start|stop]