import sqlite3
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 and create the table if necessary."""
self.db_name = db_name
self.connection = sqlite3.connect(self.db_name)
self.cursor = self.connection.cursor()
self._create_table()
def _create_table(self):
"""Create the books table if it does not exist."""
self.cursor.execute("""
CREATE TABLE IF NOT EXISTS books (
id INTEGER PRIMARY KEY AUTOINCREMENT,
title TEXT NOT NULL,
author TEXT NOT NULL,
genre TEXT NOT NULL,
summary TEXT NOT NULL
)
""")
self.connection.commit()
def add_book(self, book):
"""Add a new 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 get_book_data():
"""Prompt the user for book data."""
title = input("Enter book title: ")
author = input("Enter book author: ")
genre = input("Enter book genre: ")
summary = input("Enter book summary: ")
return Book(title, author, genre, summary)
# Test Program
if __name__ == "__main__":
# Initialize the database
db = BookDatabase()
# Ask the user to input a book's information
print("Please enter information for a new book:")
new_book = get_book_data()
# Add the new book to the database
db.add_book(new_book)
print("\nBook added successfully!\n")
# Ask the user if they want to add another book
while True:
add_another = input("Do you want to add another book? (y/n): ").lower()
if add_another == 'y':
new_book = get_book_data()
db.add_book(new_book)
print("\nBook added successfully!\n")
elif add_another == 'n':
break
else:
print("Invalid input. Please enter 'y' for yes or 'n' for no.")
# Retrieve and display all books in the database
print("\nBooks in the database:")
books = db.get_all_books()
for book in books:
print(book)
# Close the database connection
db.close()
Output
Please enter information for a new book:
Enter book title: The Great Gatsby
Enter book author: F. Scott Fitzgerald
Enter book genre: Fiction
Enter book summary: A story about Jay Gatsby's obsession with Daisy Buchanan.
Book added successfully!
Código de ejemplo copiado