main.py

import sqlite3
from typing import List, Tuple, Optional

class BibliotecaDB:
    def __init__(self, nome_db: str = "biblioteca.db"):
        self.nome_db = nome_db
        self.inicializar_db()

    def inicializar_db(self):
        """Inicializa o banco de dados e cria a tabela de livros se ela não existir."""
        with sqlite3.connect(self.nome_db) as conn:
            cursor = conn.cursor()
            cursor.execute('''
                CREATE TABLE IF NOT EXISTS livros (
                    id INTEGER PRIMARY KEY AUTOINCREMENT,
                    titulo TEXT NOT NULL,
                    autor TEXT NOT NULL,
                    isbn TEXT UNIQUE,
                    ano_publicacao INTEGER,
                    status TEXT DEFAULT 'disponivel'
                )
            ''')
            conn.commit()

    def adicionar_livro(self, titulo: str, autor: str, isbn: str, ano_publicacao: int) -> bool:
        """Adiciona um novo livro à biblioteca."""
        try:
            with sqlite3.connect(self.nome_db) as conn:
                cursor = conn.cursor()
                cursor.execute('''
                    INSERT INTO livros (titulo, autor, isbn, ano_publicacao)
                    VALUES (?, ?, ?, ?)
                ''', (titulo, autor, isbn, ano_publicacao))
                conn.commit()
                return True
        except sqlite3.IntegrityError:
            print("Erro: ISBN já existe no banco de dados.")
            return False

    def obter_todos_livros(self) -> List[Tuple]:
        """Recupera todos os livros da biblioteca."""
        with sqlite3.connect(self.nome_db) as conn:
            cursor = conn.cursor()
            cursor.execute('SELECT * FROM livros')
            return cursor.fetchall()

    def buscar_livros(self, termo: str) -> List[Tuple]:
        """Busca livros por título ou autor."""
        with sqlite3.connect(self.nome_db) as conn:
            cursor = conn.cursor()
            cursor.execute('''
                SELECT * FROM livros 
                WHERE titulo LIKE ? OR autor LIKE ?
            ''', (f'%{termo}%', f'%{termo}%'))
            return cursor.fetchall()

    def atualizar_livro(self, id_livro: int, titulo: Optional[str] = None, 
                       autor: Optional[str] = None, isbn: Optional[str] = None,
                       ano_publicacao: Optional[int] = None, status: Optional[str] = None) -> bool:
        """Atualiza informações do livro."""
        try:
            with sqlite3.connect(self.nome_db) as conn:
                cursor = conn.cursor()
                atualizacoes = []
                valores = []
                
                if titulo is not None:
                    atualizacoes.append('titulo = ?')
                    valores.append(titulo)
                if autor is not None:
                    atualizacoes.append('autor = ?')
                    valores.append(autor)
                if isbn is not None:
                    atualizacoes.append('isbn = ?')
                    valores.append(isbn)
                if ano_publicacao is not None:
                    atualizacoes.append('ano_publicacao = ?')
                    valores.append(ano_publicacao)
                if status is not None:
                    atualizacoes.append('status = ?')
                    valores.append(status)

                if not atualizacoes:
                    return False

                valores.append(id_livro)
                query = f'''
                    UPDATE livros 
                    SET {', '.join(atualizacoes)}
                    WHERE id = ?
                '''
                cursor.execute(query, valores)
                conn.commit()
                return True
        except sqlite3.IntegrityError:
            print("Erro: ISBN já existe no banco de dados.")
            return False

    def deletar_livro(self, id_livro: int) -> bool:
        """Remove um livro da biblioteca."""
        with sqlite3.connect(self.nome_db) as conn:
            cursor = conn.cursor()
            cursor.execute('DELETE FROM livros WHERE id = ?', (id_livro,))
            conn.commit()
            return cursor.rowcount > 0

def main():
    # Cria uma nova biblioteca
    biblioteca = BibliotecaDB()

    while True:
        print("\n=== Sistema de Gerenciamento de Biblioteca ===")
        print("1. Adicionar novo livro")
        print("2. Ver todos os livros")
        print("3. Buscar livro")
        print("4. Atualizar livro")
        print("5. Deletar livro")
        print("6. Sair")
        
        escolha = input("\nDigite sua escolha (1-6): ")

        if escolha == "1":
            titulo = input("Digite o título do livro: ")
            autor = input("Digite o nome do autor: ")
            isbn = input("Digite o ISBN: ")
            try:
                ano = int(input("Digite o ano de publicação: "))
                if biblioteca.adicionar_livro(titulo, autor, isbn, ano):
                    print("Livro adicionado com sucesso!")
                else:
                    print("Falha ao adicionar livro.")
            except ValueError:
                print("Formato de ano inválido. Por favor, digite um número.")

        elif escolha == "2":
            livros = biblioteca.obter_todos_livros()
            if livros:
                print("\nTodos os Livros:")
                for livro in livros:
                    print(f"ID: {livro[0]}, Título: {livro[1]}, Autor: {livro[2]}, "
                          f"ISBN: {livro[3]}, Ano: {livro[4]}, Status: {livro[5]}")
            else:
                print("Nenhum livro encontrado na biblioteca.")

        elif escolha == "3":
            termo = input("Digite o termo de busca (título ou autor): ")
            livros = biblioteca.buscar_livros(termo)
            if livros:
                print("\nResultados da Busca:")
                for livro in livros:
                    print(f"ID: {livro[0]}, Título: {livro[1]}, Autor: {livro[2]}, "
                          f"ISBN: {livro[3]}, Ano: {livro[4]}, Status: {livro[5]}")
            else:
                print("Nenhum livro encontrado com esse termo.")

        elif escolha == "4":
            try:
                id_livro = int(input("Digite o ID do livro para atualizar: "))
                print("\nDeixe em branco para manter o valor atual")
                titulo = input("Digite o novo título (ou pressione Enter para pular): ").strip()
                autor = input("Digite o novo autor (ou pressione Enter para pular): ").strip()
                isbn = input("Digite o novo ISBN (ou pressione Enter para pular): ").strip()
                ano_str = input("Digite o novo ano de publicação (ou pressione Enter para pular): ").strip()
                status = input("Digite o novo status (ou pressione Enter para pular): ").strip()

                # Converte strings vazias para None
                titulo = titulo if titulo else None
                autor = autor if autor else None
                isbn = isbn if isbn else None
                ano = int(ano_str) if ano_str else None
                status = status if status else None

                if biblioteca.atualizar_livro(id_livro, titulo, autor, isbn, ano, status):
                    print("Livro atualizado com sucesso!")
                else:
                    print("Falha ao atualizar livro.")
            except ValueError:
                print("Entrada inválida. Por favor, digite um ID de livro e ano válidos.")

        elif escolha == "5":
            try:
                id_livro = int(input("Digite o ID do livro para deletar: "))
                if biblioteca.deletar_livro(id_livro):
                    print("Livro deletado com sucesso!")
                else:
                    print("Livro não encontrado.")
            except ValueError:
                print("Entrada inválida. Por favor, digite um ID de livro válido.")

        elif escolha == "6":
            print("Obrigado por usar o Sistema de Gerenciamento de Biblioteca!")
            break

        else:
            print("Escolha inválida. Por favor, tente novamente.")

if __name__ == "__main__":
    main()