用 Python 構(gòu)建一個簡單的 SQLite 庫管理器

高效管理數(shù)據(jù)是任何項(xiàng)目的關(guān)鍵部分，而 SQLite 使這項(xiàng)任務(wù)變得簡單而輕量。在本教程中，我們將構(gòu)建一個小型 Python 應(yīng)用程序來管理圖書館數(shù)據(jù)庫，使您可以輕松添加和檢索書籍。

讀完本文，您將了解如何：

• 創(chuàng)建 SQLite 數(shù)據(jù)庫和表。
• 插入記錄，同時防止重復(fù)。
• 根據(jù)特定條件檢索數(shù)據(jù)。

---

1. 創(chuàng)建數(shù)據(jù)庫和表

讓我們首先創(chuàng)建 SQLite 數(shù)據(jù)庫文件并定義 books 表。每本書都有標(biāo)題、作者、ISBN、出版日期和流派字段。

import sqlite3
import os

def create_library_database():
    """Creates the library database if it doesn't already exist."""
    db_name = "library.db"

    if not os.path.exists(db_name):
        print(f"Creating database: {db_name}")
        conn = sqlite3.connect(db_name)
        cursor = conn.cursor()
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS books (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                title TEXT,
                author TEXT,
                isbn TEXT UNIQUE,
                published_date DATE,
                genre TEXT
            )
        ''')
        conn.commit()
        conn.close()
    else:
        print(f"Database already exists: {db_name}")

運(yùn)行此函數(shù)來初始化數(shù)據(jù)庫：

create_library_database()

這將在您的項(xiàng)目目錄中創(chuàng)建一個library.db 文件，其中包含具有指定字段的書籍表。

---

1. 將書籍插入數(shù)據(jù)庫

要插入書籍，我們希望確保避免重復(fù)條目（基于 isbn 字段）。我們將使用 SQLite 的 INSERT OR IGNORE 語句，而不是手動檢查重復(fù)項(xiàng)。

這是添加書籍的功能：

def insert_book(book):
    """
    Inserts a book into the database. If a book with the same ISBN already exists,
    the insertion is ignored.
    """
    conn = sqlite3.connect("library.db")
    cursor = conn.cursor()

    try:
        # Insert the book. Ignore the insertion if the ISBN already exists.
        cursor.execute('''
            INSERT OR IGNORE INTO books (title, author, isbn, published_date, genre)
            VALUES (?, ?, ?, ?, ?)
        ''', (book["title"], book["author"], book["isbn"], book["published_date"], book["genre"]))
        conn.commit()

        if cursor.rowcount == 0:
            print(f"The book with ISBN '{book['isbn']}' already exists in the database.")
        else:
            print(f"Book inserted: {book['title']} by {book['author']}")

    except sqlite3.Error as e:
        print(f"Database error: {e}")
    finally:
        conn.close()

此函數(shù)使用 INSERT OR IGNORE SQL 語句來確保有效地跳過重復(fù)條目。

---

?3. 添加一些書籍

讓我們通過向我們的圖書館添加一些書籍來測試 insert_book 函數(shù)。

books = [
    {
        "title": "To Kill a Mockingbird",
        "author": "Harper Lee",
        "isbn": "9780061120084",
        "published_date": "1960-07-11",
        "genre": "Fiction"
    },
    {
        "title": "1984",
        "author": "George Orwell",
        "isbn": "9780451524935",
        "published_date": "1949-06-08",
        "genre": "Dystopian"
    },
    {
        "title": "Pride and Prejudice",
        "author": "Jane Austen",
        "isbn": "9781503290563",
        "published_date": "1813-01-28",
        "genre": "Romance"
    }
]

for book in books:
    insert_book(book)

---

當(dāng)您運(yùn)行上述代碼時，書籍將被添加到數(shù)據(jù)庫中。如果您再次運(yùn)行它，您將看到類似以下的消息：

The book with ISBN '9780061120084' already exists in the database.
The book with ISBN '9780451524935' already exists in the database.
The book with ISBN '9781503290563' already exists in the database.

---

4. 檢索書籍

您可以通過查詢數(shù)據(jù)庫輕松檢索數(shù)據(jù)。例如，要獲取圖書館中的所有書籍：

def fetch_all_books():
    conn = sqlite3.connect("library.db")
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM books")
    rows = cursor.fetchall()
    conn.close()
    return rows

books = fetch_all_books()
for book in books:
    print(book)

---

結(jié)論

只需幾行Python，您現(xiàn)在就擁有了一個功能齊全的圖書館管理器，可以插入書籍，同時防止重復(fù)并輕松檢索記錄。 SQLite 的 INSERT OR IGNORE 是一個強(qiáng)大的功能，可以簡化處理約束，使您的代碼更加簡潔和高效。

隨意擴(kuò)展此項(xiàng)目，具有以下功能：

• 按書名或作者搜索書籍。
• 更新圖書信息。
• 刪除書籍。

接下來你會建造什么？ ？

以上是用 Python 構(gòu)建一個簡單的 SQLite 庫管理器的詳細(xì)內(nèi)容。更多信息請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！