37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from dataclasses import dataclass
|
|
from db.SQLiteClient import create_habit, get_habits, get_habit
|
|
|
|
# Unit wird als Integers wie folgt gemessen:
|
|
# 0: Tag
|
|
# 1: Woche
|
|
# 2: Monal
|
|
# 3: Jahr
|
|
|
|
@dataclass
|
|
class Habit:
|
|
id: int
|
|
user_id: int
|
|
name: str
|
|
note: str
|
|
times: int
|
|
unit: int
|
|
list_index: int
|
|
|
|
@staticmethod
|
|
def create(user_id: int, name: str, times: int, list_index: int, note: str | None=None, unit: int | None=1):
|
|
id = create_habit(user_id, name, note, times, unit, list_index)
|
|
return Habit(id, user_id, name, note, times, unit, list_index)
|
|
|
|
@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
|
|
|
|
@staticmethod
|
|
def get_all(user_id):
|
|
raw_habits = get_habits(user_id)
|
|
habits = []
|
|
for habit in raw_habits:
|
|
habit = Habit(habit[0], habit[1], habit[2], habit[3], habit[4], habit[5], habit[6])
|
|
habits.append(habit)
|
|
return habits if habits else None |