X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=plomtask%2Fdays.py;h=78340e2d13000e7ef49bff1bdc24f27e561993c1;hb=de2aaabbc463da2f1f55fe6279c4d0410d5b894b;hp=5fe984b7c49596121246bb7e822c8b4bb1536139;hpb=ac5a85f6d0186d714415ce7e2b51597bf5dca248;p=plomtask diff --git a/plomtask/days.py b/plomtask/days.py index 5fe984b..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' @@ -28,9 +28,9 @@ class Day(BaseModel[str]): """Individual days defined by their dates.""" table_name = 'days' to_save = ['comment'] - id_type = str 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 @@ -66,21 +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: - assert isinstance(day, Day) - return day - if not create: - raise NotFoundException(f'Day not found for date: {date}') - day = cls(date) - day.cache() - assert isinstance(day, Day) - return day - @property def date(self) -> str: """Return self.id_ under the assumption it's a date string."""