欢迎访问宙启技术站
智能推送

使用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()函数分别实现加法和乘法运算。