WMGZON/controllers/database/product.py

from .database import DatabaseController
from models.products.product import Product


class ProductController(DatabaseController):
    FIELDS = ['id', 'name', 'image', 'description', 'cost',
              'category', 'sellerID', '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 == 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_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 update(self):
        print("Doing work")

    def delete(self):
        print("Doing work")