104 lines
2.3 KiB
Python
104 lines
2.3 KiB
Python
|
from .database import DatabaseController
|
||
|
from models.stats import Stats
|
||
|
|
||
|
|
||
|
class StatsController(DatabaseController):
|
||
|
FIELDS = ['id', 'userID', 'productID', 'viewDate']
|
||
|
|
||
|
def __init__(self):
|
||
|
super().__init__()
|
||
|
|
||
|
def create(self, view: Stats):
|
||
|
params = [
|
||
|
view.userID,
|
||
|
view.productID,
|
||
|
view.viewDate
|
||
|
]
|
||
|
|
||
|
self._conn.execute(
|
||
|
"""
|
||
|
INSERT INTO Views
|
||
|
(userID, productID, viewDate)
|
||
|
VALUES (?, ?, ?)""",
|
||
|
params
|
||
|
)
|
||
|
self._conn.commit()
|
||
|
|
||
|
def read(self) -> list[Stats] | None:
|
||
|
|
||
|
cursor = self._conn.execute(
|
||
|
"SELECT * FROM Views",
|
||
|
)
|
||
|
rows = cursor.fetchmany()
|
||
|
|
||
|
if rows is None:
|
||
|
return None
|
||
|
|
||
|
views = list()
|
||
|
|
||
|
# Create an object for each row
|
||
|
for view in rows:
|
||
|
params = dict(zip(self.FIELDS, view))
|
||
|
obj = self.new_instance(Stats, params)
|
||
|
views.append(obj)
|
||
|
|
||
|
return views
|
||
|
|
||
|
def read_product(self, product_id: int = 0) -> list[Stats] | None:
|
||
|
params = [
|
||
|
product_id
|
||
|
]
|
||
|
|
||
|
cursor = self._conn.execute(
|
||
|
"""SELECT * FROM Stats
|
||
|
WHERE productID = ?
|
||
|
""",
|
||
|
params
|
||
|
)
|
||
|
rows = cursor.fetchall()
|
||
|
|
||
|
if len(rows) == 0:
|
||
|
return None
|
||
|
|
||
|
views = list()
|
||
|
|
||
|
# Create an object for each row
|
||
|
for view in rows:
|
||
|
params = dict(zip(self.FIELDS, view))
|
||
|
obj = self.new_instance(Stats, params)
|
||
|
views.append(obj)
|
||
|
|
||
|
return views
|
||
|
|
||
|
def read_user(self, user_id: int) -> list[Stats] | None:
|
||
|
params = [
|
||
|
user_id
|
||
|
]
|
||
|
|
||
|
cursor = self._conn.execute(
|
||
|
"""SELECT * FROM Views
|
||
|
WHERE userID = ?
|
||
|
""",
|
||
|
params
|
||
|
)
|
||
|
rows = cursor.fetchall()
|
||
|
|
||
|
if len(rows) == 0:
|
||
|
return None
|
||
|
|
||
|
views = list()
|
||
|
|
||
|
# Create an object for each row
|
||
|
for view in rows:
|
||
|
params = dict(zip(self.FIELDS, view))
|
||
|
obj = self.new_instance(Stats, params)
|
||
|
views.append(obj)
|
||
|
|
||
|
return views
|
||
|
|
||
|
def update(self):
|
||
|
print("Doing work")
|
||
|
|
||
|
def delete(self):
|
||
|
print("Doing work")
|