import sqlite3
import os
class Book:
"""Represents a book with a title, author, genre, and summary."""
def __init__(self, title, author, genre, summary):
"""Initialize the book with title, author, genre, and summary."""
self.title = title
self.author = author
self.genre = genre
self.summary = summary
def __str__(self):
"""Return a string representation of the book."""
return f"Title: {self.title}\nAuthor: {self.author}\nGenre: {self.genre}\nSummary: {self.summary}\n"
class BookDatabase:
"""Manages a database of books."""
def __init__(self, db_name="books.db"):
"""Initialize the database connection."""
self.db_name = db_name
self.connection = sqlite3.connect(self.db_name)
self.cursor = self.connection.cursor()
self.create_table_if_not_exists()
def create_table_if_not_exists(self):
"""Create the books table if it doesn't exist."""
self.cursor.execute('''
CREATE TABLE IF NOT EXISTS books (
title TEXT,
author TEXT,
genre TEXT,
summary TEXT
)
''')
self.connection.commit()
def add_book(self, book):
"""Add a book to the database."""
self.cursor.execute('''
INSERT INTO books (title, author, genre, summary)
VALUES (?, ?, ?, ?)
''', (book.title, book.author, book.genre, book.summary))
self.connection.commit()
def get_all_books(self):
"""Retrieve all books from the database."""
self.cursor.execute("SELECT title, author, genre, summary FROM books")
rows = self.cursor.fetchall()
return [Book(title, author, genre, summary) for title, author, genre, summary in rows]
def close(self):
"""Close the database connection."""
self.connection.close()
def display_books(db):
"""Display all books stored in the database."""
books = db.get_all_books()
if not books:
print("No books found in the database.")
else:
print("\nBooks stored in the database:\n")
for book in books:
print(book)
def add_book_to_db(db):
"""Allow the user to input information about a book and add it to the database."""
print("\nEnter the details of the new book:")
title = input("Title: ")
author = input("Author: ")
genre = input("Genre: ")
summary = input("Summary: ")
new_book = Book(title, author, genre, summary)
db.add_book(new_book)
print("\nBook added to the database!")
def main():
"""Main function to run the program."""
db = BookDatabase()
while True:
print("\nBook Database Utility")
print("1. Display all books")
print("2. Add a new book")
print("3. Exit")
choice = input("Enter your choice: ")
if choice == "1":
display_books(db)
elif choice == "2":
add_book_to_db(db)
elif choice == "3":
print("Exiting program.")
break
else:
print("Invalid choice. Please try again.")
db.close()
# Run the program
if __name__ == "__main__":
main()
Output
Book Database Utility
1. Display all books
2. Add a new book
3. Exit
Enter your choice: 1
Books stored in the database:
Title: The Great Gatsby
Author: F. Scott Fitzgerald
Genre: Fiction
Summary: A story about Jay Gatsby's obsession with Daisy Buchanan.
Código de ejemplo copiado