mirror of
https://github.com/flibusta-apps/telegram_files_cache_server.git
synced 2025-12-06 06:35:38 +01:00
Fix book getting
This commit is contained in:
@@ -117,6 +117,9 @@ async def cache_file_by_book_id(
|
||||
return None
|
||||
raise Retry(defer=15)
|
||||
|
||||
if book is None:
|
||||
return None
|
||||
|
||||
if file_type not in book.available_types:
|
||||
raise FileTypeNotAllowed(f"{file_type} not in {book.available_types}!")
|
||||
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
from datetime import date
|
||||
from typing import Generic, TypeVar
|
||||
from typing import Generic, TypeVar, Optional
|
||||
|
||||
import httpx
|
||||
from pydantic import BaseModel
|
||||
@@ -48,12 +48,15 @@ class BookDetail(Book):
|
||||
AUTH_HEADERS = {"Authorization": env_config.LIBRARY_API_KEY}
|
||||
|
||||
|
||||
async def get_book(book_id: int) -> BookDetail:
|
||||
async def get_book(book_id: int) -> Optional[BookDetail]:
|
||||
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
|
||||
)
|
||||
|
||||
if response.status_code != 200:
|
||||
return None
|
||||
|
||||
return BookDetail.parse_obj(response.json())
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user