crud-e-menu-inicial

import os
os.system("cls")

biblioteca = []

# Função para abrir o arquivo CSV
def abrir_arquivo():
    return open("bestsellers with categories.csv", "a+", encoding="utf-8")

# abaixo, seguem as funções CRUD

# (C - CREATE) Função para adicionar um livro
def adicionar_livro():
    titulo = input("Digite o título do livro: ")
    autor = input("Digite o autor do livro: ")

    # validando o ano
    while True:
        ano = input("Digite o ano de publicação do livro: ")
        if not ano.isdigit() or len(ano) != 4:
            print("Ano inválido. Digite um ano válido com 4 dígitos.")
        else:
            break

    # Validando o preço
    while True:
        preco = input("Digite o preço do livro: ")
        try:
            float(preco)
            break
        except ValueError:
            print("Preço inválido. Digite um valor numérico.")

    categorias = input("Digite as categorias do livro (separadas por vírgula): ").split(',')
    categorias = [categoria.strip() for categoria in categorias]

    # Abre o arquivo no modo de leitura e escrita
    with abrir_arquivo() as arquivo:
        # Escreve os dados do livro no arquivo
        arquivo.write(f"{titulo},{autor},{ano},{','.join(categorias)},{preco}\n")

    print("Livro adicionado com sucesso!")

# (R - READ) Função para listar os livros
def listar_livros():
    if not biblioteca:
        print("A biblioteca está vazia.")
    else:
        for i, livro in enumerate(biblioteca):
            print(f"ID: {i}, Título: {livro['Título']}, Autor: {livro['Autor']}, Ano: {livro['Ano']}, Categorias: {', '.join(livro['Categorias'])}, Preço: {livro['Preço']}")
            print("-" * 50)  # Separador

# (U - UPDATE) Função para atualizar um livro
def atualizar_livro():
    listar_livros()
    id_livro = int(input("Digite o ID do livro que deseja atualizar: "))
    if id_livro < 0 or id_livro >= len(biblioteca):
        print("ID inválido. Tente novamente.")
        return

    livro = biblioteca[id_livro]
    print("Informações atuais do livro:")
    print(f"Título: {livro['Título']}, Autor: {livro['Autor']}, Ano: {livro['Ano']}, Categorias: {', '.join(livro['Categorias'])}, Preço: {livro['Preço']}")

    novo_titulo = input("Digite o novo título (ou deixe em branco para manter o mesmo): ")
    novo_autor = input("Digite o novo autor (ou deixe em branco para manter o mesmo): ")
    novo_ano = input("Digite o novo ano (ou deixe em branco para manter o mesmo): ")
    novas_categorias = input("Digite as novas categorias (separadas por vírgula) ou deixe em branco: ").split(',')
    novas_categorias = [categoria.strip() for categoria in novas_categorias]
    novo_preco = input("Digite o novo preço (ou deixe em branco para manter o mesmo): ")

    if novo_titulo:
        livro["Título"] = novo_titulo
    if novo_autor:
        livro["Autor"] = novo_autor
    if novo_ano:
        livro["Ano"] = novo_ano
    if novas_categorias:
        livro["Categorias"] = novas_categorias
    if novo_preco:
        livro["Preço"] = novo_preco
    print("Livro atualizado com sucesso!")

# (D - DELETE) Função para excluir um livro
def excluir_livro():
    listar_livros()
    id_livro = int(input("Digite o ID do livro que deseja excluir: "))
    if id_livro < 0 or id_livro >= len(biblioteca):
        print("ID inválido. Tente novamente.")
        return

    livro_excluido = biblioteca.pop(id_livro)
    print(f"Livro '{livro_excluido['Título']}' excluído com sucesso!")

# Menu principal
while True:
    print("1.Adicionar\t2.Visualizar\t3.Atualizar\t4.Excluir\t5.Sair")
    escolha = int(input("Digite a opção escolhida: "))
    
    limpar_tela()

    if escolha == 1:
        adicionar_livro()
    elif escolha == 2:
        listar_livros()
    elif escolha == 3:
        atualizar_livro()
    elif escolha == 4:
        excluir_livro()
    elif escolha == 5:
        print("Até mais!")
        break
    else:
        print("Opção inválida. Tente novamente.")