fiap_fase2_cap6/src/database/login/iniciar_database.py at master · Hinten/fiap_fase2_cap6 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import os

from src.logger.loggers import log_info, log_warning, log_error, log_success
from src.database.login.senha import salvar_senha_arquivo_base64, carregar_senha_arquivo_base64
from src.database.tipos_base.database import Database
from src.database.utils import input_str, input_bool


def iniciar_database():

    log_info("Iniciando o banco de dados...")
    user = os.environ.get('user')
    senha = os.environ.get('senha')
    dsn = os.environ.get('dsn')

    pegou_ambiente = False
    pegou_b64 = False

    if user is not None and senha is not None:
        pegou_ambiente = True
    else:
        user, senha, dsn = carregar_senha_arquivo_base64()

        if user is not None and senha is not None:
            pegou_b64 = True


    iniciou = False

    while not iniciou:

        if user is None:
            log_warning("--- Usuário não encontrado ---", write=False)
            user = input_str('user', message_override="Digite o usuário do banco de dados: ")

        if senha is None:
            log_warning("--- Senha não encontrada ---", write=False)
            senha = input_str('senha', message_override="Digite a senha do banco de dados: ")

        if dsn is None:
            log_warning("--- DSN não encontrado ---", write=False)
            dsn = input_str('dsn', message_override="Digite o DSN do banco de dados\n(Pressione enter para o valor oracle.fiap.com.br:1521/ORCL): ", old_value='oracle.fiap.com.br:1521/ORCL')

        iniciou = Database.init_oracledb(user=user, password=senha, dsn=dsn)

        if not iniciou:
            user = None
            senha = None
            dsn = None
            log_error("--- Erro ao conectar ao banco de dados, tentando novamente. ---")

    log_success("--- Banco de dados conectado com sucesso! ---")

    if not pegou_ambiente and not pegou_b64:
        print("Deseja salvar o usuário e senha?")
        salvar = input_bool('Salvar Senha', modo='S')

        if salvar:
            salvar_senha_arquivo_base64(user, senha, dsn)
            log_success("--- Senha salva com sucesso! ---", write=False)


if __name__ == "__main__":
    iniciar_database()