Update getting books when caching

This commit is contained in:
2023-02-01 01:36:25 +01:00
parent 20b0c32a3c
commit 3680b370c2
2 changed files with 17 additions and 7 deletions

View File

@@ -16,7 +16,7 @@ from app.models import CachedFile
from app.services.caption_getter import get_caption
from app.services.downloader import download
from app.services.files_client import upload_file
from app.services.library_client import Book, get_book, get_books
from app.services.library_client import Book, get_book, get_books, get_last_book_id
logger = logging.getLogger("telegram_channel_files_manager")
@@ -56,12 +56,10 @@ async def check_books_page(ctx, page_number: int) -> None:
async def check_books(ctx: dict, *args, **kwargs) -> None: # NOSONAR
arq_pool: ArqRedis = ctx["arc_pool"]
try:
books_page = await get_books(1, PAGE_SIZE)
except httpx.ConnectError:
raise Retry(defer=15) # noqa: B904
for page_number in range(books_page.total_pages, 0, -1):
last_book_id = await get_last_book_id()
for page_number in range(0, last_book_id // 100 + 1):
await arq_pool.enqueue_job(
"check_books_page",
page_number,

View File

@@ -72,11 +72,14 @@ async def get_book(
async def get_books(page: int, page_size: int) -> Page[Book]:
id_lte = page * page_size
id_gte = (page + 1) * page_size - 1
async with httpx.AsyncClient(timeout=5 * 60) as client:
response = await client.get(
(
f"{env_config.LIBRARY_URL}/api/v1/books/"
f"?page={page}&size={page_size}&is_deleted=false"
f"?is_deleted=false&id_gte={id_gte}&id_lte={id_lte}"
),
headers=AUTH_HEADERS,
)
@@ -87,3 +90,12 @@ async def get_books(page: int, page_size: int) -> Page[Book]:
page_data.items = [Book.parse_obj(item) for item in page_data.items]
return page_data
async def get_last_book_id() -> int:
async with httpx.AsyncClient() as client:
response = await client.get(
f"{env_config.LIBRARY_URL}/api/v1/books/last", headers=AUTH_HEADERS
)
return int(response.text)