app.py

from utils import database

USER_CHOICES = """
Enter:
- 'a' to add a new book 
- 'l' to list all books
- 'r' to mark a book as read
- 'd' to delete a book
- 'q' to quit

Your choice: """


def menu():
    database.create_book_table()
    user_input = input(USER_CHOICES)

    while user_input != 'q':
        if user_input == 'a':
            prompt_add_book()

        elif user_input == 'l':
            list_books()

        elif user_input == 'r':
            prompt_read_book()

        elif user_input == 'd':
            prompt_delete_book()

        else:
            print("Unknown command. Please try again.")

        user_input = input(USER_CHOICES)


def prompt_add_book():
    name = input('Enter the title of the book: ')
    author = input('Enter the author: ')

    database.add_book(name, author)


def list_books():
    books = database.get_all_books()
    for i, book in enumerate(books):
        read = 'YES' if book['read'] else 'NO'
        print(f"{str(i + 1).zfill(2)} - {book['name']} by {book['author']}, read: {read}")


def prompt_read_book():
    name = input("Enter the name of the book you just finished reading: ")

    database.mark_book_as_read(name)


def prompt_delete_book():
    name = input("Enter the name of the book you wish to delete: ")

    database.delete_book(name)


menu()