HabitTracker/models/Habit.py

from dataclasses import dataclass
from models.HabitTrackings import HabitTrackings
from db.SQLiteClient import create_habit, get_habit,update_habit, delete_habit, get_next_slot, \
    get_habitTrackings_by_habit_id, get_slots, update_slot


# Unit wird als Integers wie folgt gemessen:
# 0: Tag
# 1: Woche (Default)
# 2: Monat
# 3: Jahr

@dataclass
class Habit:
    id: int
    user_id: int
    name: str
    note: str
    times: int
    unit: int
    slot: int

    @staticmethod
    def create(user_id: int, name: str, times: int, note: str | None = None, unit: int | None = 1):
        slot = get_next_slot(user_id)
        id = create_habit(user_id, name, times, unit, slot, note)
        return Habit(id, user_id, name, note, times, unit, slot)

    @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]) if habit else None

    def update(self, name: str=None, note: str=None, times: int=None, unit: int=None):
        update_habit(self.id, name, note, times, unit)
        if name is not None:
            self.name = name
        if note is not None:
            self.note = note
        if times is not None:
            self.times = times
        if unit is not None:
            self.unit = unit

    # So sollte die Slots Liste aussehen damit es funktioniert
    #[(id, 1), (id, 2), (id, 3), (id, 4), (id, 5)]
    def update_slot(self, new_slot: int):
        slots = get_slots(self.user_id)
        if new_slot > self.slot:
            slots = slots[self.slot:new_slot]
            for slot in slots:
                update_slot(slot[0], slot[1]-1)
        if new_slot < self.slot:
            slots = slots[new_slot-1:self.slot-1]
            for slot in slots:
                update_slot(slot[0], slot[1]+1)
        self.slot = new_slot


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


    def get_habitTrackings(self) -> list[HabitTrackings]:
        return get_habitTrackings_by_habit_id(self.id)