Python - 命令行参数

你好，有抱负的程序员们！今天，我们将开始一段激动人心的旅程，探索 Python 中的命令行参数。如果你是编程新手，不用担心；我会像经验丰富的老师一样，耐心地指导你每一步。让我们开始吧！

Python - Command-Line Arguments

什么是命令行参数？

想象一下，你是一名厨师，不是总是做同样的菜，而是根据顾客的要求改变食材。在编程中，命令行参数就像顾客的特殊要求。它们允许我们在运行 Python 程序时给予不同的输入，使我们的代码更加灵活和强大。

Python 命令行参数

在 Python 中，我们可以使用 sys 模块轻松访问命令行参数。从一个简单的例子开始：

import sys

print("你好，我是你的第一个 Python 程序！")
print("参数数量：", len(sys.argv))
print("参数：", str(sys.argv))

将这段代码保存为 hello_args.py 并从命令行运行它：

python hello_args.py arg1 arg2 arg3

你会看到类似以下输出：

你好，我是你的第一个 Python 程序！
参数数量： 4
参数： ['hello_args.py', 'arg1', 'arg2', 'arg3']

让我们来解释一下：

sys.argv 是一个包含命令行参数的列表。
第一个元素（sys.argv[0]）总是脚本的名称本身。
其余的项是我们传递的参数。

访问特定参数

现在，让我们创建一个使用这些参数的程序：

import sys

if len(sys.argv) < 2:
print("请提供一个作为参数的名字！")
else:
name = sys.argv[1]
print(f"你好，{name}！欢迎来到命令行参数的世界！")

将这段代码保存为 greet.py 并运行它：

python greet.py Alice

输出：

你好，Alice！欢迎来到命令行参数的世界！

如果你不提供参数运行它：

python greet.py

输出：

请提供一个作为参数的名字！

这个例子展示了我们如何使我们的程序具有交互性和用户友好！

在执行时传递参数

让我们创建一个更实用的例子。我们将制作一个简单的计算器：

import sys

def calculator(operation, num1, num2):
if operation == 'add':
return num1 + num2
elif operation == 'subtract':
return num1 - num2
elif operation == 'multiply':
return num1 * num2
elif operation == 'divide':
return num1 / num2 if num2 != 0 else "错误：除以零"
else:
return "无效操作"

if len(sys.argv) != 4:
print("用法：python calculator.py <操作> <数字1> <数字2>")
else:
operation = sys.argv[1]
try:
num1 = float(sys.argv[2])
num2 = float(sys.argv[3])
result = calculator(operation, num1, num2)
print(f"结果：{result}")
except ValueError:
print("错误：请提供有效的数字")

将这段代码保存为 calculator.py 并像这样使用它：

python calculator.py add 5 3

输出：

结果： 8.0

这个计算器例子展示了我们如何创建强大的工具，这些工具可以接受不同的输入并执行各种操作。

Python getopt 模块

对于更复杂的命令行接口，Python 提供了 getopt 模块。它就像一个熟练的助手，帮助处理用户可能提供的所有不同选项和参数。

这里有一个例子：

import sys
import getopt

def main(argv):
inputfile = ''
outputfile = ''
try:
opts, args = getopt.getopt(argv,"hi:o:",["ifile=","ofile="])
except getopt.GetoptError:
print('test.py -i <输入文件> -o <输出文件>')
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
print('test.py -i <输入文件> -o <输出文件>')
sys.exit()
elif opt in ("-i", "--ifile"):
inputfile = arg
elif opt in ("-o", "--ofile"):
outputfile = arg
print('输入文件是：', inputfile)
print('输出文件是：', outputfile)

if __name__ == "__main__":
main(sys.argv[1:])

将这段代码保存为 file_processor.py 并运行它：

python file_processor.py -i input.txt -o output.txt

输出：

输入文件是： input.txt
输出文件是： output.txt

getopt 模块帮助我们处理短选项（-i, -o）和长选项（--ifile, --ofile），使我们的程序更加用户友好。

Python argparse 模块

对于更复杂的命令行接口，Python 提供了 argparse 模块。它就像为你的命令行需求提供全面服务的餐厅！

这里有一个例子：

import argparse

parser = argparse.ArgumentParser(description='处理一些整数。')
parser.add_argument('integers', metavar='N', type=int, nargs='+',
help='为累加器提供一个整数')
parser.add_argument('--sum', dest='accumulate', action='store_const',
const=sum, default=max,
help='求整数的和（默认：找出最大值）')

args = parser.parse_args()
print(args.accumulate(args.integers))

将这段代码保存为 integer_processor.py 并像这样使用它：

python integer_processor.py 1 2 3 4

输出：

或者使用 --sum 选项：

python integer_processor.py 1 2 3 4 --sum

输出：

argparse 模块会自动生成帮助信息，处理错误，并允许更复杂的参数结构。

方法比较

以下是我们可以讨论的方法的快速比较：

方法	复杂度	用例
sys.argv	简单	基本参数处理
getopt	中等	短选项和长选项
argparse	高级	复杂的命令行接口，自动帮助

记住，最好的方法取决于你的具体需求。从简单的 sys.argv 开始，随着你的程序变得更加复杂，考虑使用 getopt 或 argparse。

总之，命令行参数是 Python 工具箱中的一项强大工具。它们允许你创建灵活的、交互式的程序，这些程序可以根据不同的输入和用户需求进行调整。在你继续 Python 之旅的过程中，你会发现无数种使用这些技术构建惊人软件的方法。祝你编程愉快，愿你的命令行冒险永远充满激情！

Credits: Image by storyset