##
## Programa que cria uma interface de fácil
## uso para manejo da tabela usuarios
##

import mysql.connector

def insert_bd(colunas,valores): 
    
    # função para inserserção de dados no bd que recebe colunas utilizadas e valores
    
    cur.execute("INSERT INTO usuarios ({}) VALUES({})".format(colunas,valores)) #Salva comando no cursor
    db_connection.commit()


# CONEXÃO COM SQL

user_db = '[usuario_sql]'
password_db = '[senha_sql]'
db_connection = mysql.connector.connect(host='localhost',user = user_db, passwd = password_db, auth_plugin='mysql_native_password')

cur = db_connection.cursor() #cursor que recebe comandos mysql

# cur.execute("USE COSMICPAMPA") #escolhe database

# Começo

print("PROGRAMA DE USUARIOS (crtrl+c para sair) \n Opções:")
print("1. Adicionar usuario \n2. Retirar usuario \n3. Checar usuarios\n\n")

opcoes=[1,2,3] #opções de escolhas

resposta="0"

while resposta not in opcoes: #pergunta até que a resposta esteja entre 1-3
    
    resposta = input("O que quer fazer? ")
    
    try: # para que a resposta seja um número 
        resposta = int(resposta)
    except ValueError:
        print("Responda com um número\n")

if resposta == 1:
    print("User, Nome e Pais\n")
    
    user = input("Usuario: ")
    nome = input("Nome: ")
    pais = input("País: ")
    
    valores =f"'{user}','{nome}','{pais}'"
    col1 = "user, nome, pais"
    print((valores,col1))
    insert_bd(col1, valores)
    

elif resposta == 2:
    cur.execute(f"select user FROM usuarios")
    cadastrados= str(cur.fetchall())
    print("opções: "+cadastrados)
    
    user= 'ze$%!' # o loop só irá parar quando vier uma opção válida    
    while user not in cadastrados:
        user = input("qual usuario deve ser retirado? ")    
        
    cur.execute(f"DELETE FROM usuarios WHERE user = '{user}'") #Deleta usuario
    db_connection.commit()
else:
    cur.execute(f"select * FROM usuarios")
    colunas = cur.column_names
    dados = cur.fetchall()
    
    print(colunas[0]+"\t|\t"+colunas[1]+"\t|\t"+colunas[2])
    print("--------------------------------------------------")
    
    for us,nom,pai in dados:
        print(str(us)+"\t|\t"+str(nom)+"\t|\t"+str(pai))