Calculadora com e sem função

1_calc_sem_funcao.py

#!/usr/bin/python3.6

# Pega a Operação
operacao = input("""
Bem vindo a calculadora de dois valores.
Opções:
1 para soma.
2 para subtração.
3 para multiplicação.
4 para divisão.
Digite a Operação:""")

# Pega os Valores
primeiro_valor = float(input("Digite o primeiro valor:"))
segundo_valor = float(input("Digite o segundo valor:"))

# Testa divisão por zero e fecha o programa se for
if segundo_valor == 0:
    print("Não dividimos por zero aqui...")
    exit()

# Executa a operação conforme solicitado
if operacao == "1":
    valor_final = primeiro_valor + segundo_valor
elif operacao == "2":
    valor_final = primeiro_valor - segundo_valor
elif operacao == "3":
    valor_final = primeiro_valor * segundo_valor
elif operacao == "4":
    valor_final = primeiro_valor / segundo_valor

# Imprime o resultado
print(f"O valor da operacao é: {valor_final}")

2_calc_com_funcao.py

#!/usr/bin/python3.6


def soma(val1, val2):
    """
    Retorna a soma de dois números:
    """
    return val1 + val2


def subtracao(val1, val2):
    """
    Retorna a subtracao de dois números:
    """
    return val1 - val2


def multiplicacao(val1, val2):
    """
    Retorna a multiplicacao de dois números:
    """
    return val1 * val2


def divisao(val1, val2):
    """
    Retorna a divisao de dois números:
    """
    return val1 / val2


def calcula_dois_numeros(operacao, primeiro_valor, segundo_valor):
    """
    Retorna o valor de uma operacao de dois números:
    """
    if segundo_valor == 0:
        print("Não dividimos por zero aqui...")
        exit()

    if operacao == "1":
        valor_final = soma(primeiro_valor, segundo_valor)
    elif operacao == "2":
        valor_final = subtracao(primeiro_valor, segundo_valor)
    elif operacao == "3":
        valor_final = multiplicacao(primeiro_valor, segundo_valor)
    elif operacao == "4":
        valor_final = divisao(primeiro_valor, segundo_valor)

    return valor_final


opcao = input("""
Bem vindo a calculadora de dois valores.
Opções:
1 para soma.
2 para subtração.
3 para multiplicação.
4 para divisão.
Digite a Operação:""")

valor_um = float(input("Digite o primeiro valor:"))
valor_dois = float(input("Digite o segundo valor:"))

resultado = calcula_dois_numeros(opcao, valor_um, valor_dois)

print(f"O valor da operacao é: {resultado}")