diff --git a/models/Habit.py b/models/Habit.py index b462a5c..e2d46a5 100644 --- a/models/Habit.py +++ b/models/Habit.py @@ -2,16 +2,16 @@ import json from dataclasses import dataclass from datetime import datetime -from models.HabitTrackings import HabitTrackings -from db.SQLiteClient import update_slot, create_habit, get_habit, delete_habit, get_next_slot, \ - get_habitTrackings_by_habit_id, get_slots, update_habit, get_habitList +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) -# Unit wird als Integers wie folgt gemessen: -# 0: Tag -# 1: Woche (Default) -# 2: Monat -# 3: Jahr +# unit will be represented by integers like this: +# 0: day +# 1: week (default) +# 2: month +# 3: year @dataclass class Habit: @@ -28,83 +28,74 @@ class Habit: self.fill_statistics() @staticmethod - def create(list_id: int, name: str, times: int, note: str | None = None, unit: int | None = 1): + def create(list_id: int, name: str, times: int, note: str = None, unit: int = 1): slot = get_next_slot(list_id) - id = create_habit(list_id, name, times, unit, slot, note) + id = create_habit(list_id, name, note, times, unit, slot) return Habit(id, list_id, name, note, times, unit, slot) @staticmethod def get(id: int): habit = get_habit(id) - habit = Habit(habit[0], habit[1], habit[2], habit[3], habit[4], habit[5], habit[6]) if habit else None + return Habit(habit[0], habit[1], habit[2], habit[3], habit[4], habit[5], habit[6]) if habit else None - return habit - 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 + # 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) - if new_slot > self.slot: - slots = slots[self.slot:new_slot] + + # 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: - slots = slots[new_slot-1:self.slot-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) - def get_habitTrackings(self) -> list[HabitTrackings]: + + # Returns all track-records for a Habit + def get_habitTrackings(self) -> list: trackings = [] - for rawTracking in get_habitTrackings_by_habit_id(self.id): - trackings.append(HabitTrackings(rawTracking[0], rawTracking[1], + 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 - def getStreak(self): - streak = 0 - trackings = [] - - for rawTracking in get_habitTrackings_by_habit_id(self.id): - trackings.append(HabitTrackings(datetime.strptime(rawTracking[2], "%Y-%m-%d"))) - - trackings.sort(reverse=True) - if current_date == tracking[0]: - streak += 1 - for habitdate in trackings[1:]: - future_date = habitdate - relativedelta(day=1) - if future_date - habitdate == 0-0-1: - streak + 1 - - - future_date = habitdate - relativedelta(day=1) - - for habitdate in trackings: - current_date = date.today() - if current_date == habitdate or current_date == habitdate - relativedelta(day=1): - streak += 1 - - future_date = habitdate - relativedelta(day=1) - - return streak + # 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 + # Saves the progress of the Habit in the attribute percentage def fill_statistics(self): count = 0 self.checked = False