Untitled

import psycopg2


def connect():
    try:
        conn = psycopg2.connect(
            host="localhost",
            database="postgres",
            user="postgres",
            password="77DntG4k",
        )
        return conn

    except Exception as e:
        print(f"Erro conectando a base de dados: {e}")


def cria_tabela():
    conn = connect()
    if conn is not None:
        try:
            with conn:
                with conn.cursor() as sql:
                    sql.execute('''
                            CREATE TABLE IF NOT EXISTS banda (
                                id SERIAL PRIMARY KEY,
                                nome VARCHAR(50),
                                estilo VARCHAR(50),
                                membros INTEGER
                            );
                        ''')
                    conn.commit()
                    print("Tabela criada com sucesso")
        except Exception as e:
            print(f"Erro ao criar tabela: {e}")
        finally:
            conn.close()
    else:
        print("Erro na conexão com a base de dados.")


def insere_dados(nome, estilo, membros):
    conn = connect()
    if conn is not None:
        try:
            with conn:
                with conn.cursor() as sql:
                    sql.execute(
                        "INSERT INTO banda (nome, estilo, membros) VALUES (%s, %s, %s)",
                        (nome, estilo, membros),
                    )
                    conn.commit()
                    print("Dados inseridos com sucesso.")
        except Exception as e:
            print(f"Erro ao inserir dados: {e}")
        finally:
            conn.close()
    else:
        print("Erro na conexão com a base de dados.")


def checa_tabela():
    conn = connect()
    if conn is not None:
        try:
            with conn:
                with conn.cursor() as sql:
                    sql.execute("SELECT * FROM banda")
                    print(sql.fetchall())
        except Exception as e:
            print(f"Erro ao checar tabela: {e}")
        finally:
            conn.close()
    else:
        print("Erro na conexão com a base de dados.")


if __name__ == "__main__":
    cria_tabela()

while True:
    nome = input('Digite o nome da banda: ')
    estilo = input('Digite o estilo da banda: ')
    membros = int(input('Digite o número de membros da banda: '))

    insere_dados(nome, estilo, membros)

    continuar = input('Deseja inserir mais dados? (s/n): ')
    if continuar.lower() == 'n':
        break