在使用- argparse指定参数时运行特定函数

import argparse

def add(x, y):
    return x + y

def subtract(x, y):
    return x - y

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="A simple calculator")

    parser.add_argument("num1", type=int, help="first number")
    parser.add_argument("num2", type=int, help="second number")
    parser.add_argument("-a", "--add", action="store_true", help="add the two numbers")
    parser.add_argument("-s", "--subtract", action="store_true", help="subtract the two numbers")

    args = parser.parse_args()

    if args.add:
        result = add(args.num1, args.num2)
        print("Result:", result)

    if args.subtract:
        result = subtract(args.num1, args.num2)
        print("Result:", result)

python calculator.py 5 3 -a

Result: 8