Add timeout to IA availability requests to avoid localhost hanging (#11247)

Author: RayBB

openlibrary/core/lending.py

@@ -4,11 +4,13 @@
 
 import datetime
 import logging
+import os
 import time
 import uuid
 from typing import TYPE_CHECKING, Literal, TypedDict, cast
 
 import eventer
+import httpx
 import requests
 import web
 from simplejson.errors import JSONDecodeError
@@ -202,10 +204,22 @@ def get_groundtruth_availability(ocaid, s3_keys=None):
     including 1-hour borrows"""
     params = '?action=availability&identifier=' + ocaid
     url = S3_LOAN_URL % config_bookreader_host
+    timeout = 2 if os.getenv('LOCAL_DEV') else 5
     try:
-        response = requests.post(url + params, data=s3_keys)
+        response = httpx.post(url + params, data=s3_keys, timeout=timeout)
         response.raise_for_status()
-    except requests.HTTPError:
+    except httpx.TimeoutException:
+        if os.getenv('LOCAL_DEV'):
+            logger.warning(
+                "Availability request timed out in LOCAL_DEV environment. Returning empty dictionary."
+            )
+            return {}
+        else:
+            logger.error(
+                "Availability request timed out in non-LOCAL_DEV environment. Re-raising the exception."
+            )
+            raise  # Re-raise the timeout exception if not in LOCAL_DEV
+    except httpx.HTTPError:
         pass  # TODO: Handle unexpected responses from the availability server.
     try:
         data = response.json().get('lending_status', {})