54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from dataclasses import dataclass
|
|
|
|
from models.Habit import Habit
|
|
from models.User import User
|
|
from db.SQLiteClient import (delete_habitList, create_habitList, get_habitList, get_habits,
|
|
get_users, add_user, remove_user)
|
|
|
|
|
|
@dataclass
|
|
class HabitList:
|
|
id: int
|
|
name: str
|
|
description: str
|
|
habits: list = None
|
|
|
|
@staticmethod
|
|
def create(user_id: int, name: str, description: str):
|
|
id = create_habitList(user_id, name, description)
|
|
return HabitList(id, name, description)
|
|
|
|
@staticmethod
|
|
def get(id: int):
|
|
habitList = get_habitList(id)
|
|
return HabitList(habitList[0], habitList[1], habitList[2]) if habitList else None
|
|
|
|
def get_habits(self) -> list:
|
|
raw_habits = get_habits(self.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
|
|
|
|
def get_users(self) -> list:
|
|
raw_users = get_users(self.id)
|
|
users = []
|
|
for user in raw_users:
|
|
user = User(user[0], user[1], user[2], user[3])
|
|
users.append(user)
|
|
|
|
return users
|
|
|
|
def add_user(self, email: str):
|
|
user = User.get_by_email(email)
|
|
add_user(self.id, user.id)
|
|
|
|
# The id of the current user is necessary
|
|
def delete(self, user_id):
|
|
if len(get_users) > 1:
|
|
remove_user(self.id, user_id)
|
|
else:
|
|
delete_habitList(self.id)
|