HabitTracker/models/Habit.py

import json
from dataclasses import dataclass
from datetime import datetime
from datetime import timedelta

from models.HabitTracking import HabitTracking
from db.SQLiteClient import (create_habit, get_habit, update_habit, delete_habit, get_next_slot, get_slots, update_slot,
                             get_habitTrackings, get_habitList, update_habit_statistics)


# unit will be represented by integers like this:
# 0: day
# 1: week (default)
# 2: month
# 3: year

@dataclass
class Habit:
    id: int
    list_id: int
    name: str
    note: str
    times: int
    unit: int
    slot: int
    checked: bool
    count: int
    streak: int
    percentage: int = 0

    def __post_init__(self):
        self.load_statistics()

    @staticmethod
    def create(list_id: int, name: str, times: int, note: str = None, unit: int = 1, checked: bool = False, count: int = 0, streak: int = 0):
        slot = get_next_slot(list_id)
        id = create_habit(list_id, name, note, times, unit, slot, checked, count, streak)
        return Habit(id, list_id, name, note, times, unit, slot, checked, count, streak)

    @staticmethod
    def get(id: int):
        habit = get_habit(id)
        return Habit(habit[0], habit[1], habit[2], habit[3], habit[4], habit[5], habit[6], habit[7], habit[8], habit[9]) if habit else None


    # Updates: name, note, times, unit
    def update(self):
        update_habit(self.id, self.name, self.note, self.times, self.unit)


    # Updates the slot and reorders the HabitList accordingly
    def update_slot(self, new_slot: int):
        # Fetches a list with the following structure [(id, slot), (id, slot), ...]
        slots = get_slots(self.list_id)

        # Splits the list depending on whether the new slot is higher or lower than the current one
        if new_slot > self.slot:                  # Example self.slot=1 new_slot=4
            slots = slots[self.slot:new_slot]     # Expected list: [(id, 2), (id, 3), (id, 4)]
            for slot in slots:
                update_slot(slot[0], slot[1]-1)
        if new_slot < self.slot:                  # Example self.slot=4 new_slot=1
            slots = slots[new_slot-1:self.slot-1] # Expected list: [(id, 1), (id, 2), (id, 3)]
            for slot in slots:
                update_slot(slot[0], slot[1]+1)

        # Update the slot of the current habit
        update_slot(self.id, new_slot)


    # Deletes the Habit
    def delete(self):
        # Reorders the slots
        slots = get_slots(self.list_id)[self.slot+1:]
        for slot in slots:
            update_slot(slot[0], slot[1] - 1)

        # Deletes all track-records associated with the Habit
        trackings = self.get_habitTrackings()
        for tracking in trackings:
            tracking.delete()

        # Deletes the current Habit
        delete_habit(self.id)


    # Returns all track-records for a Habit
    def get_habitTrackings(self) -> list:
        trackings = []
        for rawTracking in get_habitTrackings(self.id):
            trackings.append(HabitTracking(rawTracking[0], rawTracking[1],
                                            datetime.strptime(rawTracking[2], "%Y-%m-%dT%H:%M:%S.%f")))
        return trackings


    # Returns the HabitList in which the Habit is located
    def habit_list(self) -> list:
        from models.HabitList import HabitList
        raw_habitLists = get_habitList(self.list_id)
        return HabitList(raw_habitLists[0], raw_habitLists[1], raw_habitLists[2]) if raw_habitLists else None


    # Loads the progress and checks if the streak is not broken
    def load_statistics(self):
        print("loading")
        today_date = datetime.today().date()
        tracking_dates = [tracking.created_at.date() for tracking in self.get_habitTrackings()]
        yesterday = today_date - timedelta(days=1)
        if not self.checked and not yesterday in tracking_dates:
            self.streak = 0
            update_habit_statistics(self.id, self.count, self.count, self.streak)

        self.percentage = int(self.count / self.times * 100)

    # Saves the progress count and streak
    def fill_statistics(self):
        self.checked = True
        self.streak += 1
        self.count += 1
        update_habit_statistics(self.id, self.checked, self.count, self.streak)

    # Turns the statistics back to the unchecked state
    def reset_statistics(self):
        self.checked = False
        self.streak -= 1
        self.count -= 1
        update_habit_statistics(self.id, self.checked, self.count, self.streak)


    def to_json(self):
        return json.dumps(self, default=lambda o: o.__dict__, sort_keys=True, indent=4)