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

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-01-01 21:24:54 +01:00
parent 3ed43b002e
commit 25ba409be7
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -72,7 +72,9 @@ async def check_books(ctx: dict, *args, **kwargs) -> None: # NOSONAR
) )
async def cache_file(book: Book, file_type: str) -> Optional[CachedFile]: async def cache_file(
book: Book, file_type: str, by_request: bool = True
) -> Optional[CachedFile]:
if await CachedFile.objects.filter( if await CachedFile.objects.filter(
object_id=book.id, object_type=file_type object_id=book.id, object_type=file_type
).exists(): ).exists():
@@ -106,13 +108,16 @@ async def cache_file(book: Book, file_type: str) -> Optional[CachedFile]:
if upload_data is None: if upload_data is None:
return None return None
return await CachedFile.objects.create( cached_file = await CachedFile.objects.create(
object_id=book.id, object_id=book.id,
object_type=file_type, object_type=file_type,
message_id=upload_data.data["message_id"], message_id=upload_data.data["message_id"],
chat_id=upload_data.data["chat_id"], chat_id=upload_data.data["chat_id"],
) )
if by_request:
return cached_file
async def cache_file_by_book_id( async def cache_file_by_book_id(
ctx: dict, # NOSONAR ctx: dict, # NOSONAR
@@ -138,7 +143,7 @@ async def cache_file_by_book_id(
try: try:
try: try:
async with lock: async with lock:
return await cache_file(book, file_type) return await cache_file(book, file_type, by_request)
except LockError: except LockError:
raise Retry(defer=timedelta(minutes=15).seconds * random.random()) raise Retry(defer=timedelta(minutes=15).seconds * random.random())
except Retry as e: except Retry as e: