WMGZON/controllers/database/stats.py

from .database import DatabaseController
from models.stats import Stats
from datetime import datetime
from utils.general_utils import is_within_x_days


class StatsController(DatabaseController):
    FIELDS = ['id', 'userID', 'productID', 'viewDate']
    TYPE = Stats

    def __init__(self):
        super().__init__()

    def create(self, view: Stats):
        params = [
            view.userID,
            view.productID,
            view.viewDate
        ]
        query = """
            INSERT INTO Views
            (userID, productID, viewDate)
            VALUES (?, ?, ?)
        """

        self.do(query, params)

    def read(self) -> list[Stats] | None:
        query = "SELECT * FROM Views"
        self.get_many(query, [])

    def read_product(self, product_id: int = 0) -> list[Stats] | None:
        params = [
            product_id
        ]
        query = """
            SELECT * FROM Views
            WHERE productID = ?
        """

        return self.get_many(query, params)

    def read_user(self, user_id: int) -> list[Stats] | None:
        params = [
            user_id
        ]
        query = """
            SELECT * FROM Views
            WHERE userID = ?
        """

        return self.get_many(query, params)

    def read_days(self, product_id: int, prev_days: int = 7
                  ) -> dict[int, list[Stats]] | None:
        """ Returns data from within the given number of days """
        data = self.read_product(product_id)

        # Filter the values to only be within X prev days
        filtered_data = list()
        if data is not None:
            filtered_data = list(filter(
                lambda d: is_within_x_days(d.viewDate, prev_days),
                data
            ))

        day_views: dict[int, list[Stats]] = dict()

        for i in range(0, prev_days):
            day_views[i] = list()

        # Organise data into distinct
        for view in filtered_data:
            diff = datetime.now()-view.viewDate
            day_views[diff.days].append(view)

        return day_views

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

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