X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=plomtask%2Fdb.py;fp=plomtask%2Fdb.py;h=8e5529062ae18c60a2ed222deb27305f9c18e54c;hb=3a74c67ef298d16e7a7e5dd030e1fd813f42f0b4;hp=34b5887c0204532ea6a32ade0eed6ccea92fbbaa;hpb=704c00a101fc1922158a3a71a043562d34b696ad;p=plomtask diff --git a/plomtask/db.py b/plomtask/db.py index 34b5887..8e55290 100644 --- a/plomtask/db.py +++ b/plomtask/db.py @@ -179,27 +179,26 @@ class BaseModel(Generic[BaseModelId]): obj.cache() return obj - @classmethod - def _by_id(cls, db_conn: DatabaseConnection, - id_: BaseModelId) -> Self | None: - """Return instance found by ID, or None, and if from cache or not.""" - obj = cls.get_cached(id_) - if not obj: - for row in db_conn.row_where(cls.table_name, 'id', id_): - obj = cls.from_table_row(db_conn, row) - obj.cache() - break - return obj - @classmethod def by_id(cls, db_conn: DatabaseConnection, id_: BaseModelId | None, # pylint: disable=unused-argument create: bool = False) -> Self: - """Retrieve by id_, on failure throw NotFoundException.""" + """Retrieve by id_, on failure throw NotFoundException. + + First try to get from cls.cache_, only then check DB; if found, + put into cache. + + If create=True, make anew (but do not cache yet). + """ obj = None if id_ is not None: - obj = cls._by_id(db_conn, id_) + obj = cls.get_cached(id_) + if not obj: + for row in db_conn.row_where(cls.table_name, 'id', id_): + obj = cls.from_table_row(db_conn, row) + obj.cache() + break if obj: return obj if create: