You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
990 B
37 lines
990 B
from be.model import store
|
|
|
|
|
|
class DBConn:
|
|
def __init__(self):
|
|
self.conn = store.get_db_conn()
|
|
|
|
def user_id_exist(self, user_id):
|
|
cursor = self.conn.execute(
|
|
"SELECT user_id FROM user WHERE user_id = ?;", (user_id,)
|
|
)
|
|
row = cursor.fetchone()
|
|
if row is None:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def book_id_exist(self, store_id, book_id):
|
|
cursor = self.conn.execute(
|
|
"SELECT book_id FROM store WHERE store_id = ? AND book_id = ?;",
|
|
(store_id, book_id),
|
|
)
|
|
row = cursor.fetchone()
|
|
if row is None:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def store_id_exist(self, store_id):
|
|
cursor = self.conn.execute(
|
|
"SELECT store_id FROM user_store WHERE store_id = ?;", (store_id,)
|
|
)
|
|
row = cursor.fetchone()
|
|
if row is None:
|
|
return False
|
|
else:
|
|
return True
|