使用Python的cmdline()函数处理命令行输入的示例代码
发布时间:2023-12-16 12:15:40
Python的cmdline()函数是一个用于处理命令行输入的内建函数。它允许我们从命令行接收输入参数,并根据参数执行相应的操作。下面是一个简单的示例代码,演示了cmdline()函数的使用:
import sys
def cmdline():
args = sys.argv[1:] # 获取命令行参数,排除脚本名称
if len(args) == 0:
print("请输入参数!")
else:
action = args[0]
if action == "add":
if len(args) == 3:
num1 = int(args[1])
num2 = int(args[2])
result = num1 + num2
print("结果:", result)
else:
print("参数错误!请提供两个数字作为参数。")
elif action == "sub":
if len(args) == 3:
num1 = int(args[1])
num2 = int(args[2])
result = num1 - num2
print("结果:", result)
else:
print("参数错误!请提供两个数字作为参数。")
else:
print("未知操作!")
# 调用cmdline()函数
cmdline()
在这个示例代码中,我们定义了一个cmdline()函数,用于处理命令行参数。首先,我们使用sys.argv[1:]获取除脚本名称外的所有命令行参数,存储在args列表中。然后,我们检查args列表的长度,如果长度为0,则提示用户输入参数。否则,我们提取args列表中的 个参数作为操作符。
如果操作符是"add",则我们检查args列表的长度是否为3,如果是,则将前两个参数转换为整数并执行加法运算,将结果打印出来。如果参数个数不正确,则提示用户参数错误。
如果操作符是"sub",则我们执行类似的操作,但是是进行减法运算。
如果操作符不是"add"或"sub",则输出未知操作的提示。
下面是一些使用这个示例代码的实例:
1. 从命令行执行python script.py add 5 3,将输出结果结果: 8。
2. 从命令行执行python script.py sub 10 5,将输出结果结果: 5。
3. 从命令行执行python script.py add 10,将输出结果参数错误!请提供两个数字作为参数。。
4. 从命令行执行python script.py mul 2 3,将输出结果未知操作!。
通过这个示例代码,我们可以看到如何使用Python的cmdline()函数来处理命令行输入,并根据输入执行相应的操作。我们可以根据实际需求扩展这个示例,添加更多的操作和参数处理逻辑。
