calculadora_tratamento_erros.py
import operator
def calculadora(numero1, numero2, operador):
operacao = {
"+": operator.add,
"-": operator.sub,
"*": operator.mul,
"/": operator.truediv
}
if operador not in operacao:
return "Opção inválida"
try:
return f"Resultado: {operacao[operador](int(numero1), int(numero2))}"
except ValueError:
return "Erro: Entrada inválida. Digite apenas números."
except ZeroDivisionError:
return "Erro: Divisão por zero não é permitida."
main.py
from calculadora_tratamento_erros import calculadora
numero1 = input("Digite o primeiro número: ")
operador = input("Escolha a operação (+, -, *, /): ")
numero2 = input("Digite o segundo número: ")
print(calculadora(numero1, numero2, operador))