用 Python 構(gòu)建一個(gè)簡(jiǎn)單的 SQLite 庫(kù)管理器

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

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

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

---

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

讓我們首先創(chuàng)建 SQLite 數(shù)據(jù)庫(kù)文件并定義 books 表。每本書(shū)都有標(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ù)來(lái)初始化數(shù)據(jù)庫(kù)：

create_library_database()

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

---

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

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

這是添加書(shū)籍的功能：

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 語(yǔ)句來(lái)確保有效地跳過(guò)重復(fù)條目。

---

?3. 添加一些書(shū)籍

讓我們通過(guò)向我們的圖書(shū)館添加一些書(shū)籍來(lái)測(cè)試 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í)，書(shū)籍將被添加到數(shù)據(jù)庫(kù)中。如果您再次運(yùn)行它，您將看到類(lèi)似以下的消息：

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ū)籍

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

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)在就擁有了一個(gè)功能齊全的圖書(shū)館管理器，可以插入書(shū)籍，同時(shí)防止重復(fù)并輕松檢索記錄。 SQLite 的 INSERT OR IGNORE 是一個(gè)強(qiáng)大的功能，可以簡(jiǎn)化處理約束，使您的代碼更加簡(jiǎn)潔和高效。

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

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

接下來(lái)你會(huì)建造什么？ ？

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