crud.py

from crud_modules import firebase_settings, add, delete, register_store, read

auth = firebase_settings.auth
database = firebase_settings.database
storage = firebase_settings.storage
admin_auth = firebase_settings.admin_auth
admin_db = firebase_settings.admin_db


def testar_conexao():
    ref = admin_db.reference('/')
    #return ref.listen(callback)
    return print(ref.get())
    

if __name__ == '__main__':
    keep = True
    
    while keep:
        try:
            escolha = int(input("1. Apagar todas as contas\n2. Registrar um novo item\n3. Definir stand-by em um item existente\n" + 
                                "4. Excluir um item existente\n5. Listar itens registrados\n6. Registrar nova loja\n\n> Digite a opção desejada: "))

            # apagar todas as contas (apenas para testes de registro de usuario)
            if escolha == 1:
                userlist = admin_auth.list_users().iterate_all()
                for user in userlist:
                    admin_auth.delete_user(user.uid)
                    print("deletado!")
                    
            # registrar um novo item
            elif escolha == 2:
                item = add.criar_novo_item()
                if item:
                    print("[-] Item criado com sucesso. Enviando para o banco de dados...")
                    print(add.adicionar_novo_item(item))

            # Colocar item em stand-by (falta de estoque)
            elif escolha == 3:
                testar_conexao()
            
            # Remover completamente um item
            elif escolha == 4: 
                print(delete.delete(delete.selecionar_item()))
            
            elif escolha == 5:
                print('-'*15 + " Itens Registrados " + '-'*15)
                read.ler_todos_itens()
                print ('-'*49 +'\n')
            
            elif escolha == 6:
                print(register_store.adicionar_nova_loja(register_store.criar_nova_loja()))
            else:
                keep = False
        
        except Exception as erro:
            print(f"[X] Erro: {erro}\n")