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)