diff --git a/backend/pashub_fetcher/handler/handler.py b/backend/pashub_fetcher/handler/handler.py index 971cd6cd..2a39db60 100644 --- a/backend/pashub_fetcher/handler/handler.py +++ b/backend/pashub_fetcher/handler/handler.py @@ -1,3 +1,4 @@ +import time from typing import Any, List, Mapping from backend.pashub_fetcher.cotality_client import CotalityClient @@ -20,7 +21,15 @@ def handler(event: Mapping[str, Any], context: Any) -> None: raise client = CotalityClient(token=token) - job_id = "5abf6e27-e4c4-4ba8-b69d-9e34939e0002" # TODO: get from request body - saved_files: List[str] = client.get_core_envidence_files_by_job_id(job_id) + jobs = [ + "5abf6e27-e4c4-4ba8-b69d-9e34939e0002", + "047f4455-85e2-4293-97b1-6b460137d33e", + ] # TODO: get these from request body + + saved_files: List[str] = [] + for job_id in jobs: + saved_files.extend(client.get_core_envidence_files_by_job_id(job_id)) + time.sleep(10) # Simulate manual download + print(f"saved {len(saved_files)} files")