Refactor database.py: Fix variable name and add debug statements

This commit is contained in:
WorldTeacher
2024-02-08 15:56:01 +01:00
parent 96d4dcf46a
commit a33e632e34

View File

@@ -93,7 +93,7 @@ class Database:
"SELECT deleted FROM media WHERE app_id=? AND prof_id=? AND bookdata=?"
)
params = (app_id, prof_id, pickle.dumps(bookdata))
result = self.cur.execute(query, params).fetchone()
result = cursor.execute(query, params).fetchone()
if result[0] == 1:
print("Book was deleted, updating bookdata")
query = "UPDATE media SET deleted=0 WHERE app_id=? AND prof_id=? AND bookdata=?"
@@ -124,7 +124,8 @@ class Database:
def getLastBookId(self)->int:
return self.query_db("SELECT id FROM media ORDER BY id DESC", one=True)[0]
def searchBook(self, data:dict[str, str])->list[tuple[BookData, int]]:
data = self.query_db("SELECT * FROM media WHERE deleted=0")
rdata = self.query_db("SELECT * FROM media WHERE deleted=0")
ic(rdata, len(rdata))
mode = 0
if len(data)== 1:
if "signature" in data.keys():
@@ -136,7 +137,7 @@ class Database:
else:
return None
ret = []
for book in data:
for book in rdata:
bookdata = load_pickle(book[1])
app_id = book[2]
prof_id = book[3]
@@ -149,6 +150,7 @@ class Database:
elif mode == 3:
if data["signature"] in bookdata.signature and data["title"] in bookdata.title:
ret.append((bookdata,app_id,prof_id))
ic(ret)
return ret
def setAvailability(self, book_id, available):
self.query_db("UPDATE media SET available=? WHERE id=?", (available,book_id))