refactor: add error log if max polling duration reached

hotosm · Feb 7, 2024 · 12f0bee · 12f0bee
1 parent 1c288dc
commit 12f0bee
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/osm_rawdata/postgres.py b/osm_rawdata/postgres.py
@@ -501,6 +501,11 @@ def queryRemote(
             elif result_json.get("status") == "SUCCESS":
                 break
 
+        else:
+            # Maximum polling duration reached
+            log.error(f"{max_polling_duration} second elapsed. Aborting data extract.")
+            return None
+
         zip_url = result_json["result"]["download_url"]
         result = self.session.get(zip_url, headers=self.headers)
         fp = BytesIO(result.content)