X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=plomtask%2Fdays.py;h=78340e2d13000e7ef49bff1bdc24f27e561993c1;hb=8570f4ce4d44b813a1f02b72c5c45a57d2003bae;hp=258d38dbbf1d7f920f36ca06c9f8292506d16807;hpb=4815fe5c7be508e67ceec144968a81bdd6a923d4;p=plomtask diff --git a/plomtask/days.py b/plomtask/days.py index 258d38d..78340e2 100644 --- a/plomtask/days.py +++ b/plomtask/days.py @@ -1,7 +1,7 @@ """Collecting Day and date-related items.""" from __future__ import annotations from datetime import datetime, timedelta -from plomtask.exceptions import BadFormatException, NotFoundException +from plomtask.exceptions import BadFormatException from plomtask.db import DatabaseConnection, BaseModel DATE_FORMAT = '%Y-%m-%d' @@ -30,6 +30,7 @@ class Day(BaseModel[str]): to_save = ['comment'] def __init__(self, date: str, comment: str = '') -> None: + super().__init__(date) self.id_: str = valid_date(date) self.datetime = datetime.strptime(self.date, DATE_FORMAT) self.comment = comment @@ -65,19 +66,6 @@ class Day(BaseModel[str]): days = gapless_days return days - @classmethod - def by_id(cls, db_conn: DatabaseConnection, - date: str, create: bool = False) -> Day: - """Retrieve Day by date if in DB (prefer cache), else return None.""" - day, _ = super()._by_id(db_conn, date) - if day: - return day - if not create: - raise NotFoundException(f'Day not found for date: {date}') - day = cls(date) - day.cache() - return day - @property def date(self) -> str: """Return self.id_ under the assumption it's a date string."""