flibusta-apps/telegram_files_cache_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/telegram_files_cache_server.git synced 2025-12-06 06:35:38 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix book getting

Browse Source
This commit is contained in:
Bulat Kurbanov
2022-12-07 20:51:52 +01:00
parent 6c77f0ced2
commit 6dcf78736a
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/app/services/cache_updater.py
View File

@@ -117,6 +117,9 @@ async def cache_file_by_book_id(
return None return None
raise Retry(defer=15) raise Retry(defer=15)
if book is None:
return None
if file_type not in book.available_types: if file_type not in book.available_types:
raise FileTypeNotAllowed(f"{file_type} not in {book.available_types}!") raise FileTypeNotAllowed(f"{file_type} not in {book.available_types}!")

7
src/app/services/library_client.py
View File

@@ -1,5 +1,5 @@
from datetime import date from datetime import date
from typing import Generic, TypeVar from typing import Generic, TypeVar, Optional
import httpx import httpx
from pydantic import BaseModel from pydantic import BaseModel
@@ -48,12 +48,15 @@ class BookDetail(Book):
AUTH_HEADERS = {"Authorization": env_config.LIBRARY_API_KEY} 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: async with httpx.AsyncClient(timeout=2 * 60) as client:
response = await client.get( response = await client.get(
f"{env_config.LIBRARY_URL}/api/v1/books/{book_id}", headers=AUTH_HEADERS 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()) return BookDetail.parse_obj(response.json())