Fix get_book

src/app/services/cache_updater.py

@@ -110,16 +110,14 @@ async def cache_file_by_book_id(
 ) -> Optional[CachedFile]:
     r_client: aioredis.Redis = ctx["redis"]
 
-    try:
+    get_book_retry = 3 if by_request else 1
-        book = await get_book(book_id)
+    book = await get_book(book_id, get_book_retry)
-    except httpx.ConnectError:
+
+    if book is None:
         if by_request:
             return None
         raise Retry(defer=15)
 
-    if book is None:
-        return None
-
     if file_type not in book.available_types:
         return None
 

src/app/services/library_client.py

@@ -48,16 +48,22 @@ class BookDetail(Book):
 AUTH_HEADERS = {"Authorization": env_config.LIBRARY_API_KEY}
 
 
-async def get_book(book_id: int) -> Optional[BookDetail]:
+async def get_book(book_id: int, retry: int = 3) -> Optional[BookDetail]:
-    async with httpx.AsyncClient(timeout=2 * 60) as client:
+    if retry == 0:
-        response = await client.get(
+        return None
-            f"{env_config.LIBRARY_URL}/api/v1/books/{book_id}", headers=AUTH_HEADERS
-        )
 
-        if response.status_code != 200:
+    try:
-            return None
+        async with httpx.AsyncClient(timeout=2 * 60) as client:
+            response = await client.get(
+                f"{env_config.LIBRARY_URL}/api/v1/books/{book_id}", headers=AUTH_HEADERS
+            )
 
-        return BookDetail.parse_obj(response.json())
+            if response.status_code != 200:
+                return None
+
+            return BookDetail.parse_obj(response.json())
+    except (httpx.ConnectError, httpx.ReadError, httpx.ReadTimeout):
+        return await get_book(book_id, retry=retry - 1)
 
 
 async def get_books(page: int, page_size: int) -> Page[Book]: