X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;ds=sidebyside;f=plomtask%2Fdays.py;h=5e975602d112a81f71b7bafb47b2939f9084d365;hb=ce9b74158c87750ccf1c8f363a59c0465e4b1277;hp=9f3aa697a831758d0635109446a35e73428b7181;hpb=704c00a101fc1922158a3a71a043562d34b696ad;p=plomtask diff --git a/plomtask/days.py b/plomtask/days.py index 9f3aa69..5e97560 100644 --- a/plomtask/days.py +++ b/plomtask/days.py @@ -5,6 +5,8 @@ from plomtask.exceptions import BadFormatException from plomtask.db import DatabaseConnection, BaseModel DATE_FORMAT = '%Y-%m-%d' +MIN_RANGE_DATE = '2024-01-01' +MAX_RANGE_DATE = '2030-12-31' def valid_date(date_str: str) -> str: @@ -42,7 +44,14 @@ class Day(BaseModel[str]): def all(cls, db_conn: DatabaseConnection, date_range: tuple[str, str] = ('', ''), fill_gaps: bool = False) -> list[Day]: - """Return list of Days in database within date_range.""" + """Return list of Days in database within (open) date_range interval. + + If no range values provided, defaults them to MIN_RANGE_DATE and + MAX_RANGE_DATE. Also knows to properly interpret 'today' as value. + + On fill_gaps=True, will instantiate (without saving) Days of all dates + within the date range that don't exist yet. + """ min_date = '2024-01-01' max_date = '2030-12-31' start_date = valid_date(date_range[0] if date_range[0] else min_date) @@ -69,6 +78,17 @@ class Day(BaseModel[str]): assert isinstance(self.id_, str) return self.id_ + @property + def first_of_month(self) -> bool: + """Return what month self.date is part of.""" + assert isinstance(self.id_, str) + return self.id_[-2:] == '01' + + @property + def month_name(self) -> str: + """Return what month self.date is part of.""" + return self.datetime.strftime('%B') + @property def weekday(self) -> str: """Return what weekday matches self.date.""" @@ -85,7 +105,3 @@ class Day(BaseModel[str]): """Return date succeeding date of this Day.""" next_datetime = self.datetime + timedelta(days=1) return next_datetime.strftime(DATE_FORMAT) - - def save(self, db_conn: DatabaseConnection) -> None: - """Add (or re-write) self to DB and cache.""" - self.save_core(db_conn)