26 lines
774 B
Python
26 lines
774 B
Python
from dataclasses import dataclass
|
|
from datetime import date, datetime
|
|
|
|
from db.SQLiteClient import create_habitTrackings, get_habitTrackings, delete_habitTrackings
|
|
|
|
|
|
@dataclass
|
|
class HabitTrackings:
|
|
id: int
|
|
habit_id: int
|
|
times: int
|
|
created_at: date
|
|
|
|
@staticmethod
|
|
def create(habit_id: int, times: int):
|
|
id = create_habitTrackings(habit_id, times)
|
|
return HabitTrackings(id, habit_id, times, datetime.now())
|
|
|
|
@staticmethod
|
|
def get(id: int):
|
|
habitTrackings = get_habitTrackings(id)
|
|
return HabitTrackings(habitTrackings[0], habitTrackings[1], habitTrackings[2], datetime.strptime(habitTrackings[3], "%Y-%m-%dT%H:%M:%S.%f")) if habitTrackings else None
|
|
|
|
def delete(self):
|
|
delete_habitTrackings(self.id)
|