Python 命令行选项

Python 命令行选项是指在使用Python解释器执行代码时，可以通过命令行参数来传递额外的选项，以定制程序的行为。本文将从多个方面详细阐述Python命令行选项的使用方法和常见的应用场景。

一、解析命令行选项

1、使用argparse模块

import argparse

parser = argparse.ArgumentParser(description='This is a program to demonstrate command-line options in Python.')

parser.add_argument('-f', '--file', type=str, help='Specify the input file.')
parser.add_argument('-l', '--limit', type=int, default=10, help='Specify the limit on the number of items.')

args = parser.parse_args()

if args.file:
    print(f"Input file: {args.file}")
else:
    print("No input file specified.")

print(f"Limit: {args.limit}")

上述代码演示了如何使用argparse模块解析命令行选项。其中-f或--file指定了一个字符串类型的选项，-l或--limit指定了一个整数类型的选项。通过args.xxx的方式可以获取相应的选项值。

2、手动解析命令行参数

import sys

args = sys.argv[1:]  # 排除脚本名称本身

file = None
limit = 10

for i in range(len(args)):
    if args[i] == '-f' and i + 1 < len(args):
        file = args[i + 1]
    elif args[i] == '-l' and i + 1 < len(args):
        limit = int(args[i + 1])

if file:
    print(f"Input file: {file}")
else:
    print("No input file specified.")

print(f"Limit: {limit}")

上述代码演示了手动解析命令行选项的方法。通过遍历命令行参数，找到相应的选项和参数，然后进行相应的处理。

二、常见命令行选项

1、传递布尔值

命令行选项中通常可以传递布尔值（开关型选项），表示选项是否开启。在argparse模块中，可以使用store_true和store_false这两个特殊的action参数实现。

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('-v', '--verbose', action='store_true', help='Enable verbose mode.')

args = parser.parse_args()

if args.verbose:
    print("Verbose mode enabled.")
else:
    print("Verbose mode disabled.")

2、传递列表

有时候需要从命令行传递一个列表作为选项值，可以使用nargs参数设置'*'表示接受任意多个参数，并将它们存储为列表。

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('-i', '--items', type=int, nargs='*', help='Specify a list of items.')

args = parser.parse_args()

if args.items:
    print(f"Items: {args.items}")
else:
    print("No items specified.")

3、传递字典

类似于传递列表，也可以从命令行传递一个字典作为选项值，可以使用'*'和'='将键值对一起传递，并解析为字典。

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('-d', '--dict', nargs='*', type=lambda s: dict(map(str, s.split('='))), help='Specify a dictionary.')

args = parser.parse_args()

if args.dict:
    print(f"Dictionary: {args.dict}")
else:
    print("No dictionary specified.")

三、实际应用场景

Python命令行选项在实际应用中具有广泛的应用场景。以下是几个常见的实际案例：

1、脚本参数配置

使用命令行选项可以方便地进行脚本参数配置。例如，可以通过命令行选项设置输入文件、输出文件、日志级别等参数，使脚本具有更强的灵活性。

2、批量处理文件

对于需要批量处理一系列文件的脚本，可以使用命令行选项指定输入文件夹和输出文件夹，从而一次性处理多个文件。

3、调试和测试

在开发调试和单元测试过程中，命令行选项可以用于指定特定的测试用例或调试模式，方便进行问题定位和测试覆盖。

总之，Python命令行选项为开发者提供了一种灵活且便捷的定制程序行为的方式，可以根据需要来选择使用何种选项以及如何解析和处理。通过合理使用命令行选项，可以使Python程序更加易于使用和维护。