una calculadora

def calculator(a, b, operation = "add"):
  if operation == "add":
    return a + b
  elif operation == "sub":
    return a - b
  elif operation == "mul":
    return a * b
  elif operation == "div":
    return a / b
  else:
    return "Unsupported operation"

calculator(1, 2, "add")