nXX29c1u

1import sqlite3
2import os.path
3
4BASE_DIR = os.path.dirname(os.path.abspath(__file__))
5db_path = os.path.join(BASE_DIR, "books.db")
6
7
8def connect():
9    conn = sqlite3.connect("books.db")
10    cur = conn.cursor()
11    cur.execute(
12        "CREATE TABLE IF NOT EXISTS books (id INTEGER PRIMARY KEY, title TEXT, author TEXT, year INTEGER, isbn INTEGER)")
13    conn.commit()
14    conn.close()
15
16
17def insert(title, author, year, isbn):
18    conn = sqlite3.connect("books.db")
19    cur = conn.cursor()
20    cur.execute("INSERT INTO books VALUES (NULL, ?, ?, ?, ?)", (title, author, year, isbn))
21    conn.commit()
22    conn.close()
23
24
25def view():
26    conn = sqlite3.connect("books.db")
27    cur = conn.cursor()
28    cur.execute("SELECT FROM * books")
29    rows = cur.fetchall()
30    conn.close()
31
32
33def search(title="", author="", year="", isbn=""):
34    conn = sqlite3.connect("books.db")
35    cur = conn.cursor()
36    cur.execute("SELECT * FROM books WHERE title=? OR author=? OR year=? OR isbn=?", (title, author, year, isbn))
37    rows = cur.fetchall()
38    conn.close()
39    return rows
40
41
42def delete(id):
43    conn = sqlite3.connect("books.db")
44    cur = conn.cursor()
45    cur.execute("DELETE FROM books WHERE id=?, (id,)")
46    conn.close()
47    # return rows
48
49
50def update(id, title, author, year, isbn):
51    conn = sqlite3.connect("book.db")
52    cur = conn.cursor()
53    cur.execute("UPDATE books SET title=?, author=?, year=?, isbn=? WHERE id=?", (title, author, year, isbn, id))
54    conn.commit()
55    conn.close()
56
57
58connect()