X-Git-Url: https://plomlompom.com/repos/%7B%7B%20web_path%20%7D%7D/decks/%7B%7Bdeck_id%7D%7D/cards/%7B%7Bcard_id%7D%7D/form?a=blobdiff_plain;f=plomtask%2Fdays.py;h=9f3aa697a831758d0635109446a35e73428b7181;hb=704c00a101fc1922158a3a71a043562d34b696ad;hp=78340e2d13000e7ef49bff1bdc24f27e561993c1;hpb=c6ad5c1fe2811225648a1d1c74c667ed38b7671e;p=plomtask diff --git a/plomtask/days.py b/plomtask/days.py index 78340e2..9f3aa69 100644 --- a/plomtask/days.py +++ b/plomtask/days.py @@ -30,14 +30,11 @@ class Day(BaseModel[str]): to_save = ['comment'] def __init__(self, date: str, comment: str = '') -> None: - super().__init__(date) - self.id_: str = valid_date(date) + id_ = valid_date(date) + super().__init__(id_) self.datetime = datetime.strptime(self.date, DATE_FORMAT) self.comment = comment - def __eq__(self, other: object) -> bool: - return isinstance(other, self.__class__) and self.date == other.date - def __lt__(self, other: Day) -> bool: return self.date < other.date @@ -69,6 +66,7 @@ class Day(BaseModel[str]): @property def date(self) -> str: """Return self.id_ under the assumption it's a date string.""" + assert isinstance(self.id_, str) return self.id_ @property @@ -90,4 +88,4 @@ class Day(BaseModel[str]): def save(self, db_conn: DatabaseConnection) -> None: """Add (or re-write) self to DB and cache.""" - self.save_core(db_conn, update_with_lastrowid=False) + self.save_core(db_conn)