161 lines
4.0 KiB
Python
161 lines
4.0 KiB
Python
from .database import DatabaseController
|
|
from models.products.product import Product
|
|
|
|
|
|
class ProductController(DatabaseController):
|
|
FIELDS = ['id', 'name', 'image', 'description', 'cost',
|
|
'sellerID', 'category', 'postedDate', 'quantityAvailable']
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def create(self, product: Product):
|
|
params = [
|
|
product.name,
|
|
product.image,
|
|
product.description,
|
|
product.cost,
|
|
product.category,
|
|
product.sellerID,
|
|
product.postedDate,
|
|
product.quantityAvailable
|
|
]
|
|
|
|
self._conn.execute(
|
|
"""
|
|
INSERT INTO Products
|
|
(name, image, description, cost, categoryID,
|
|
sellerID, postedDate, quantityAvailable)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)""",
|
|
params
|
|
)
|
|
self._conn.commit()
|
|
|
|
def read(self, name: str = "") -> list[Product] | None:
|
|
params = [
|
|
"%" + name + "%"
|
|
]
|
|
|
|
cursor = self._conn.execute(
|
|
"SELECT * FROM Products WHERE name like ?",
|
|
params
|
|
)
|
|
rows = cursor.fetchmany()
|
|
|
|
if rows is None:
|
|
return None
|
|
|
|
products = list()
|
|
|
|
# Create an object for each row
|
|
for product in rows:
|
|
params = dict(zip(self.FIELDS, product))
|
|
obj = self.new_instance(Product, params)
|
|
products.append(obj)
|
|
|
|
return products
|
|
|
|
def read_id(self, id: int) -> Product | None:
|
|
params = [
|
|
id
|
|
]
|
|
|
|
cursor = self._conn.execute(
|
|
"SELECT * FROM Products WHERE id == ?",
|
|
params
|
|
)
|
|
row = cursor.fetchone()
|
|
|
|
if row is None:
|
|
return None
|
|
|
|
# Create an object with the row
|
|
params = dict(zip(self.FIELDS, row))
|
|
obj = self.new_instance(Product, params)
|
|
|
|
return obj
|
|
|
|
def read_all(self, category: str = "",
|
|
search_term: str = "") -> list[Product] | None:
|
|
params = [
|
|
"%" + category + "%",
|
|
"%" + search_term + "%"
|
|
]
|
|
|
|
cursor = self._conn.execute(
|
|
"""SELECT * FROM Products
|
|
INNER JOIN Categories ON Products.categoryID = Categories.id
|
|
WHERE Categories.name LIKE ?
|
|
AND Products.name LIKE ?
|
|
""",
|
|
params
|
|
)
|
|
rows = cursor.fetchall()
|
|
|
|
if len(rows) == 0:
|
|
return None
|
|
|
|
products = list()
|
|
|
|
# Create an object for each row
|
|
for product in rows:
|
|
params = dict(zip(self.FIELDS, product))
|
|
obj = self.new_instance(Product, params)
|
|
products.append(obj)
|
|
|
|
return products
|
|
|
|
def read_user(self, user_id: int) -> list[Product] | None:
|
|
params = [
|
|
user_id
|
|
]
|
|
|
|
cursor = self._conn.execute(
|
|
"""SELECT * FROM Products
|
|
WHERE sellerID = ?
|
|
""",
|
|
params
|
|
)
|
|
rows = cursor.fetchall()
|
|
|
|
if len(rows) == 0:
|
|
return None
|
|
|
|
products = list()
|
|
|
|
# Create an object for each row
|
|
for product in rows:
|
|
params = dict(zip(self.FIELDS, product))
|
|
obj = self.new_instance(Product, params)
|
|
products.append(obj)
|
|
|
|
return products
|
|
|
|
def update(self, product: Product):
|
|
params = [
|
|
product.name,
|
|
product.description,
|
|
product.image,
|
|
product.cost,
|
|
product.quantityAvailable,
|
|
product.category,
|
|
product.id
|
|
]
|
|
|
|
cursor = self._conn.execute(
|
|
"""UPDATE Products
|
|
SET name = ?,
|
|
description = ?,
|
|
image = ?,
|
|
cost = ?,
|
|
quantityAvailable = ?,
|
|
categoryID = ?
|
|
WHERE id = ?
|
|
""",
|
|
params
|
|
)
|
|
self._conn.commit()
|
|
|
|
def delete(self):
|
|
print("Doing work")
|