Fix celery gets stuck (#220)

* Add timeout to fetching archive or else it hangs indefinitely * Add timeout to fetching swm or else it hangs indefinitely
canonical · Oct 10, 2024 · 7402e4e · 7402e4e
1 parent 8f49818
commit 7402e4e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/test_observer/external_apis/archive.py b/backend/test_observer/external_apis/archive.py
@@ -115,7 +115,7 @@ def _create_download_and_extract_filepaths(self) -> None:
 
     def _download_data(self) -> None:
         """Download Packages.gz file from archive"""
-        response = requests.get(self.url, stream=True)
+        response = requests.get(self.url, stream=True, timeout=30)
         if not response.ok:
             response.raise_for_status()
 

diff --git a/backend/test_observer/kernel_swm_integration/swm_reader.py b/backend/test_observer/kernel_swm_integration/swm_reader.py
@@ -20,7 +20,7 @@ def get_artefacts_swm_info(db: Session) -> dict[int, ArtefactTrackerInfo]:
 
 def _fetch_stable_workflow_manager_status() -> dict:
     url = "https://kernel.ubuntu.com/swm/status.json"
-    return requests.get(url).json()
+    return requests.get(url, timeout=30).json()
 
 
 def _extract_artefact_bug_info_from_swm(