diff --git a/src/auto_archiver/modules/atlos_feeder_db_storage/atlos_feeder_db_storage.py b/src/auto_archiver/modules/atlos_feeder_db_storage/atlos_feeder_db_storage.py index ef82238..87b4f82 100644 --- a/src/auto_archiver/modules/atlos_feeder_db_storage/atlos_feeder_db_storage.py +++ b/src/auto_archiver/modules/atlos_feeder_db_storage/atlos_feeder_db_storage.py @@ -43,6 +43,8 @@ class AtlosFeederDbStorage(Feeder, Database, Storage): response.raise_for_status() return response.json() + # ! Atlos Module - Feeder Methods + def __iter__(self) -> Iterator[Metadata]: """Iterate over unprocessed, visible source materials from Atlos.""" cursor = None @@ -61,6 +63,8 @@ class AtlosFeederDbStorage(Feeder, Database, Storage): if not results or cursor is None: break + # ! Atlos Module - Database Methods + def failed(self, item: Metadata, reason: str) -> None: """Mark an item as failed in Atlos, if the ID exists.""" atlos_id = item.metadata.get("atlos_id") @@ -104,6 +108,8 @@ class AtlosFeederDbStorage(Feeder, Database, Storage): ) logger.info(f"Stored success for {item.get_url()} (ID {atlos_id}) on Atlos") + # ! Atlos Module - Storage Methods + def get_cdn_url(self, _media: Media) -> str: """Return the base Atlos URL as the CDN URL.""" return self.atlos_url