from datetime import date, datetime
from dataclasses import dataclass
from db.SQLiteClient import create_habitTracking, get_habitTracking, delete_habitTracking


@dataclass
class HabitTracking:
    id: int
    habit_id: int
    created_at: date

    @staticmethod
    def create(habit_id: int):
        id = create_habitTracking(habit_id)
        return HabitTracking(id, habit_id, datetime.now())

    @staticmethod
    def get(id: int):
        habitTrackings = get_habitTracking(id)
        return HabitTracking(habitTrackings[0], habitTrackings[1],
                              datetime.strptime(habitTrackings[2], "%Y-%m-%dT%H:%M:%S.%f")) \
                              if habitTrackings else None

    # Deletes the HabitTracking
    def delete(self):
        delete_habitTracking(self.id)