用Python处理Args的3种方法

欢迎关注 “小白玩转Python”，发现更多 “有趣”

1. sys 模块

Python 中的 sys 模块具有 argv 功能。当通过终端触发 main.py 的执行时，此功能将返回提供给 main.py 的所有命令行参数的列表。除了其他参数之外，返回列表中的第一个元素是 main.py 的路径。

考虑下面的 main.py 示例

import sys
list_of_arguments = sys.argv
print(list_of_args[0]) 
print(list_of_args[1]) 
print(list_of_args[2]) 
print(list_of_args[3])

用如下方式在命令行触发 main.py ：

python main.py first_arg "[second_arg]" "{\"arg\": 3}"

输出的结果如下所示：

test.py
first_arg
[second_arg]
{"arg": 3}

2.带有一个大参数的 sys 模块

这是一种为 Python 代码提供参数的简单而强大的方法。你不需要提供按空格分隔的大量参数，而是提供一个单一的“大”参数。这个大参数是一个字符串字典，其中 dict-keys 表示参数名称，dict-value 表示相应的值。

考虑下面的 main.py 示例

import sys
import ast


raw_arguments = sys.argv[1]


print(raw_arguments)
arguments = ast.literal_eval(raw_arguments)


print(arguments['name']) # John
print(arguments['surname']) # Doe
print(arguments['age']) # 22

用如下方式在命令行触发 main.py ：

python main.py "{\"name\": \"John\", \"surname\": \"Doe\", \"age\": 22}"

输出的结果如下所示：

{"name": "John", "surname": "Doe", "age": 22}
John
Doe
22

 3. argparse 模块

如果想为你的应用程序提供适当的命令行界面，则 argparse 是你需要的模块。

这是一个全面的模块，提供了现成的参数解析、帮助消息和误用参数时自动抛错。

要充分利用 argparse 提供的功能，需要花费一些时间来掌握。现在先然我们来考虑以下 main.py 示例：

import argparse


parser = argparse.ArgumentParser(description='Personal information')
parser.add_argument('--name', dest='name', type=str, help='Name of the candidate')
parser.add_argument('--surname', dest='surname', type=str, help='Surname of the candidate')
parser.add_argument('--age', dest='age', type=int, help='Age of the candidate')


args = parser.parse_args()
print(args.name)
print(args.surname)
print(args.age)

在初始化 ArgumentParses 的对象之后，我们使用 add_argument 函数添加所有预期的参数。该函数接收许多参数，其中包括参数名、目标变量、预期的数据类型、要显示的帮助消息等。

用如下方式在命令行触发 main.py ：

python main.py --name John --surname Doe --age 22

输出的结果如下所示：

John
Doe
22

我们这里只是简单介绍一下，要了解更多关于这个模块的信息，可以查看这个链接https://docs.python.org/2/library/argparse.html。

总结

很多时候，你需要将参数传递给 Python 脚本。Python通过 sys 模块提供对这些参数的访问。你可以直接访问argv功能并自己处理参数解析，也可以将其他模块用作argparse来完成。对于小编来说，通常习惯使用sys模块，适合你的才是最好的~

·  END  ·

HAPPY LIFE