X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=plomtask%2Fdays.py;h=afdea33760b41df3149c857b306c6e2b447b7df8;hb=37ad3f2181d8bacd2a9d48e8fca4ede1e24de6b6;hp=3b81a7f3bf526fd7bc56849558fbfaa451cdbe88;hpb=3558a14701955de18ae7adbda0e93aaee7710a92;p=plomtask diff --git a/plomtask/days.py b/plomtask/days.py index 3b81a7f..afdea33 100644 --- a/plomtask/days.py +++ b/plomtask/days.py @@ -2,7 +2,7 @@ from __future__ import annotations from datetime import datetime, timedelta from sqlite3 import Row -from plomtask.misc import HandledException +from plomtask.exceptions import BadFormatException, NotFoundException from plomtask.db import DatabaseConnection DATE_FORMAT = '%Y-%m-%d' @@ -16,7 +16,7 @@ def valid_date(date_str: str) -> str: dt = datetime.strptime(date_str, DATE_FORMAT) except (ValueError, TypeError) as e: msg = f'Given date of wrong format: {date_str}' - raise HandledException(msg) from e + raise BadFormatException(msg) from e return dt.strftime(DATE_FORMAT) @@ -76,7 +76,7 @@ class Day: for row in db_conn.exec('SELECT * FROM days WHERE date = ?', (date,)): return cls.from_table_row(row) if not create: - raise HandledException(f'Day not found for date: {date}') + raise NotFoundException(f'Day not found for date: {date}') return cls(date) @property