DevOps | Python 命令行交互自动化

2023-11-12 08:37:23

使用 sys 模块

sys 模块提供了一种简单的方法来获取命令行参数。它包含了一个名为 argv 的列表，其中包含了脚本的名称以及所有命令行参数。例如，如果我们运行以下脚本：

import sys

print(sys.argv)

那么输出结果将是：

['script.py', 'arg1', 'arg2']

sys 模块还提供了一些其他有用的函数，比如 sys.exit() 函数，它可以用来退出脚本。

2. 使用 argparse 模块

argparse 模块提供了一种更强大的方式来解析命令行参数。它允许您定义参数的类型、默认值和帮助信息。例如，我们可以使用以下代码来定义一个名为 "name" 的参数，它是一个字符串类型的参数，默认值为 "World"，帮助信息为 "要问候的人的名字"：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("name", type=str, default="World", help="要问候的人的名字")

args = parser.parse_args()

print("Hello, {}!".format(args.name))

运行这个脚本，我们可以得到以下输出：

Hello, World!

如果我们想向脚本传递一个不同的名字，我们可以使用以下命令：

python script.py Bob

那么输出结果将是：

Hello, Bob!

3. 使用 getopt 模块

getopt 模块提供了一种更低级别的命令行参数解析方式。它允许您直接访问原始的命令行参数。例如，我们可以使用以下代码来获取脚本的名称和所有命令行参数：

import getopt

opts, args = getopt.getopt(sys.argv[1:], "hn:")

for opt, arg in opts:
    if opt == "-h":
        print("Usage: script.py [-h] [-n NAME]")
        sys.exit(0)
    elif opt == "-n":
        name = arg

print("Hello, {}!".format(name))