HabitTracker/models/User.py

from datetime import datetime

from flask_login import UserMixin
from db.SQLiteClient import create_user, get_user, get_user_by_email, delete_user, update_user, \
    get_habitLists, get_heatmap_value


class User(UserMixin):
    def __init__(self, id: int, name: str, email: str, password: str | None = None):
        self.id = id
        self.name = name
        self.email = email
        self.password = password

    @staticmethod
    def create(name: str, email: str, password: str):
        id = create_user(name, email, password)
        return User(id, name, email)

    @staticmethod
    def get(id: int):
        user = get_user(id)
        return User(user[0], user[1], user[2], user[3]) if user else None

    @staticmethod
    def get_by_email(email: str):
        user = get_user_by_email(email)
        return User(user[0], user[1], user[2], user[3]) if user else None

    def update(self):
        update_user(self.id, self.name, self.email, self.password if self.password else None)

    def delete(self):
        delete_user(self.id)

    def get_habitLists(self):
        from models.HabitList import HabitList

        raw_habitLists = get_habitLists(self.id)
        habitLists = []
        for habitList in raw_habitLists:
            habitList = HabitList(habitList[0], habitList[1], habitList[2], datetime.strptime(habitList[3], "%Y-%m-%dT%H:%M:%S.%f"), datetime.strptime(habitList[4], "%Y-%m-%dT%H:%M:%S.%f"))
            habitLists.append(habitList)

        return habitLists

    def get_heatmap(self):
        heatmap = []
        for day in range(0, 28):
            value = get_heatmap_value(self.id, day)
            heatmap.append(value)
        return heatmap