# -*- coding: utf-8 -*
#import time
import math

def welcome():
    print('''
Bem vindo
''')
def calculate():
    op = input('''
Selecione a operação
+ para adição
- para subtração
* para multiplicação
/ para divisão
** para potencia
r para raiz quadrada
''')

    if op == 'r':
        n1 = float(input('Digite a raiz: '))
        raiz = math.sqrt(n1)
        print(raiz)

    else:

        n1 = int(input('Digite o primeiro numero: '))
        n2 = int(input('Digite o segundo numero: '))

        if op == '+':
            print('{} + {} = '.format(n1, n2))
            print(n1 + n2)

        elif op == '-':
            print('{} - {} = '.format(n1, n2))
            print(n1 - n2)

        elif op == '*':
            print('{} * {} = '.format(n1, n2))
            print(n1 * n2)

        elif op == '/':
            print('{} / {} = '.format(n1, n2))
            print(n1 / n2)

        elif op == '**':
            print('{} ** {} = '.format(n1, n2))
            print(n1 ** n2)

        else:
            print('Operação invalida, execute novamente o programa.')

            # Add again() na função calculate() function

    again()
                    #time.sleep(3000)

def again():
    calc_again = input('''
Precisa resolver mais calculos?
Digite S para SIM ou N para nao.
''')

    if calc_again.upper() == 'S':
        calculate()
    elif calc_again.upper() == 'N':
        print('Ate mais.')
    else:
        again()

calculate()