Merge pull request #40 from ternaustralia/edmond/add-timeout

Add timeout to requests
ternaustralia · Oct 6, 2022 · 4d96af5 · 4d96af5
2 parents cbea718 + 6e5fe54
commit 4d96af5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/linkeddata_api/data/sparql.py b/src/linkeddata_api/data/sparql.py
@@ -21,7 +21,7 @@ def post(
         "content-type": "application/sparql-query",
     }
 
-    response = requests.post(url=sparql_endpoint, headers=headers, data=query)
+    response = requests.post(url=sparql_endpoint, headers=headers, data=query, timeout=60)
 
     try:
         response.raise_for_status()
@@ -51,7 +51,7 @@ def get(
     }
     params = {"query": query}
 
-    response = requests.get(url=sparql_endpoint, headers=headers, params=params)
+    response = requests.get(url=sparql_endpoint, headers=headers, params=params, timeout=60)
 
     try:
         response.raise_for_status()

diff --git a/src/linkeddata_api/domain/curie.py b/src/linkeddata_api/domain/curie.py
@@ -62,7 +62,7 @@ def get(uri: str):
     base_uri = uri[:r_index]
 
     response = requests.post(
-        "https://prefix.zazuko.com/api/v1/shrink", params={"q": base_uri}
+        "https://prefix.zazuko.com/api/v1/shrink", params={"q": base_uri}, timeout=60
     )
 
     try:

diff --git a/src/linkeddata_api/views/api_v1/ontology_viewer/classes/flat/crud.py b/src/linkeddata_api/views/api_v1/ontology_viewer/classes/flat/crud.py
@@ -67,7 +67,9 @@ def get(ontology_id: str) -> List[schema.ClassItem]:
         "content-type": "application/sparql-query",
     }
 
-    r = requests.post(url=mapping["sparql_endpoint"], headers=headers, data=query)
+    r = requests.post(
+        url=mapping["sparql_endpoint"], headers=headers, data=query, timeout=60
+    )
 
     try:
         r.raise_for_status()