116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
import json
|
|
from dataclasses import dataclass
|
|
from datetime import datetime
|
|
|
|
from models.HabitTrackings import HabitTrackings
|
|
from db.SQLiteClient import (create_habit, get_habit, update_habit, delete_habit, get_next_slot, get_slots, update_slot,
|
|
get_habitTrackings_by_habit_id, get_habitList)
|
|
|
|
|
|
# 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
|
|
percentage: int = 0
|
|
|
|
def __post_init__(self):
|
|
self.fill_statistics()
|
|
|
|
@staticmethod
|
|
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, note, times, unit, slot)
|
|
return Habit(id, list_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):
|
|
update_habit(self.id, self.name, self.note, self.times, self.unit)
|
|
|
|
|
|
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)
|
|
|
|
|
|
def delete(self):
|
|
# Deletes the current habit
|
|
slots = get_slots(self.list_id)[self.slot+1:]
|
|
for slot in slots:
|
|
update_slot(slot[0], slot[1] - 1)
|
|
|
|
# Deletes the current habit
|
|
delete_habit(self.id)
|
|
|
|
|
|
def get_habitTrackings(self) -> list:
|
|
trackings = []
|
|
for rawTracking in get_habitTrackings_by_habit_id(self.id):
|
|
trackings.append(HabitTrackings(rawTracking[0], rawTracking[1]))
|
|
return trackings
|
|
|
|
|
|
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
|
|
|
|
|
|
def fill_statistics(self):
|
|
count = 0
|
|
self.checked = False
|
|
for tracking in self.get_habitTrackings():
|
|
if tracking.created_at.date() == datetime.today().date():
|
|
self.checked = True
|
|
|
|
# day
|
|
if self.unit == 0:
|
|
if tracking.created_at.date() == datetime.today().date():
|
|
count += 1
|
|
# week
|
|
elif self.unit == 1:
|
|
if tracking.created_at.isocalendar()[1] == datetime.today().isocalendar()[1]:
|
|
count += 1
|
|
# month
|
|
elif self.unit == 2:
|
|
if tracking.created_at.month == datetime.today().month:
|
|
count += 1
|
|
# year
|
|
elif self.unit == 3:
|
|
if tracking.created_at.year == datetime.today().year:
|
|
count += 1
|
|
|
|
self.percentage = int(count / self.times * 100)
|
|
|
|
|
|
def to_json(self):
|
|
return json.dumps(self, default=lambda o: o.__dict__, sort_keys=True, indent=4)
|