使用Python编写一个简单的命令行工具
发布时间:2023-12-04 13:17:50
下面是一个简单的命令行工具的Python代码示例:
import argparse
def add_numbers(num1, num2):
return num1 + num2
def multiply_numbers(num1, num2):
return num1 * num2
def main():
parser = argparse.ArgumentParser(description='A simple command-line calculator')
group = parser.add_mutually_exclusive_group()
group.add_argument('-a', '--add', nargs=2, type=float, help='add two numbers')
group.add_argument('-m', '--multiply', nargs=2, type=float, help='multiply two numbers')
args = parser.parse_args()
if args.add:
result = add_numbers(args.add[0], args.add[1])
print(f"The sum of {args.add[0]} and {args.add[1]} is {result}")
elif args.multiply:
result = multiply_numbers(args.multiply[0], args.multiply[1])
print(f"The product of {args.multiply[0]} and {args.multiply[1]} is {result}")
else:
parser.print_help()
if __name__ == '__main__':
main()
使用该命令行工具,可以执行简单的加法或乘法运算。以下是使用例子:
1. 加法运算:
$ python calculator.py -a 2 3 The sum of 2.0 and 3.0 is 5.0
2. 乘法运算:
$ python calculator.py -m 4 5 The product of 4.0 and 5.0 is 20.0
3. 查看帮助信息:
$ python calculator.py -h
usage: calculator.py [-h] [-a ADD ADD] [-m MULTIPLY MULTIPLY]
A simple command-line calculator
optional arguments:
-h, --help show this help message and exit
-a ADD ADD, --add ADD ADD
add two numbers
-m MULTIPLY MULTIPLY, --multiply MULTIPLY MULTIPLY
multiply two numbers
该命令行工具使用argparse模块解析命令行参数,并根据参数执行相应的操作。add_numbers()和multiply_numbers()函数分别实现加法和乘法运算。
