X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=plomtask%2Fdays.py;h=0936d047aa3eda7c4ce9374c90ca3f9ad3362ad5;hb=12bbfea600184c9e2657cc9a0c705848a4dae29a;hp=776658e14aa6f7f488eeda7546eb8612c2f50ee0;hpb=af769db012fc0cf590b565c3368102b2f6099394;p=plomtask diff --git a/plomtask/days.py b/plomtask/days.py index 776658e..0936d04 100644 --- a/plomtask/days.py +++ b/plomtask/days.py @@ -1,5 +1,5 @@ """Collecting Day and date-related items.""" -from datetime import datetime +from datetime import datetime, timedelta from sqlite3 import Row from plomtask.misc import HandledException from plomtask.db import DatabaseConnection @@ -46,7 +46,7 @@ class Day: def all(cls, db_conn: DatabaseConnection, date_range: tuple[str, str] = ('', '')): """Return list of Days in database within date_range.""" - def date_from_range_element(date_str: str, default: str): + def date_from_range_str(date_str: str, default: str): if date_str == '': date_str = default elif date_str == 'today': @@ -54,8 +54,8 @@ class Day: elif not date_valid(date_str): raise HandledException(f'Bad date: {date_str}') return date_str - start_date = date_from_range_element(date_range[0], '2024-01-01') - end_date = date_from_range_element(date_range[1], '2030-01-01') + start_date = date_from_range_str(date_range[0], '2024-01-01') + end_date = date_from_range_str(date_range[1], '2030-01-01') days = [] sql = 'SELECT * FROM days WHERE date >= ? AND date <= ?' for row in db_conn.exec(sql, (start_date, end_date)): @@ -78,6 +78,18 @@ class Day: """Return what weekday matches self.date.""" return self.datetime.strftime('%A') + @property + def prev_date(self): + """Return date preceding date of this Day.""" + prev_datetime = self.datetime - timedelta(days=1) + return prev_datetime.strftime(DATE_FORMAT) + + @property + def next_date(self): + """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): """Add (or re-write) self to database.""" db_conn.exec('REPLACE INTO days VALUES (?, ?)',