Python에서 간단한 SQLite 라이브러리 관리자 구축-파이썬 튜토리얼-php.cn

Python에서 간단한 SQLite 라이브러리 관리자 구축

Patricia Arquette

풀어 주다： 2024-12-10 14:29:08

원래의

700명이 탐색했습니다.

Building a Simple SQLite Library Manager in Python

Python에서 간단한 SQLite 라이브러리 관리자 구축

데이터를 효율적으로 관리하는 것은 모든 프로젝트의 핵심 부분이며 SQLite는 이 작업을 간단하고 가볍게 만듭니다. 이 튜토리얼에서는 최소한의 노력으로 책을 추가하고 검색할 수 있도록 도서관 데이터베이스를 관리하는 작은 Python 애플리케이션을 구축합니다.

이 기사가 끝나면 다음 방법을 알게 될 것입니다.

SQLite 데이터베이스와 테이블을 생성하세요.
중복을 방지하면서 기록을 삽입하세요.
특정 기준에 따라 데이터를 검색합니다.

1. 데이터베이스 및 테이블 생성

SQLite 데이터베이스 파일을 생성하고 books 테이블을 정의하는 것부터 시작해 보겠습니다. 각 책에는 제목, 저자, 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}")

로그인 후 복사

데이터베이스를 초기화하려면 다음 함수를 실행하세요.

create_library_database()

로그인 후 복사

이렇게 하면 지정된 필드가 있는 책 테이블이 포함된 library.db 파일이 프로젝트 디렉토리에 생성됩니다.

데이터베이스에 도서 삽입

도서를 삽입할 때 isbn 필드를 기준으로 중복 항목이 발생하지 않도록 하려고 합니다. 중복을 수동으로 확인하는 대신 SQLite의 INSERT OR IGNORE 문을 사용하겠습니다.

도서 추가 기능은 다음과 같습니다.

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()

로그인 후 복사

이 함수는 INSERT OR IGNORE SQL 문을 사용하여 중복 항목을 효율적으로 건너뛰도록 합니다.

3. 일부 도서 추가

라이브러리에 책을 추가하여 insert_book 기능을 테스트해 보겠습니다.

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)

로그인 후 복사

위 코드를 실행하면 해당 도서가 데이터베이스에 추가됩니다. 다시 실행하면 다음과 같은 메시지가 표시됩니다.

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. 도서 검색

데이터베이스를 쿼리하여 쉽게 데이터를 검색할 수 있습니다. 예를 들어, 도서관에 있는 모든 책을 가져오려면:

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)

로그인 후 복사

결론

이제 Python 몇 줄만 있으면 중복을 방지하면서 책을 삽입하고 손쉽게 레코드를 검색할 수 있는 기능적인 라이브러리 관리자가 있습니다. SQLite의 INSERT OR IGNORE는 제약 조건 처리를 단순화하여 코드를 더욱 간결하고 효율적으로 만드는 강력한 기능입니다.

다음과 같은 기능으로 이 프로젝트를 자유롭게 확장하세요.