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 14:45:36 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-05-22 12:23:31 +02:00
parent cd6f28cdbc
commit 688e673364
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/app/utils.py
View File

@@ -1,11 +1,13 @@
from fastapi import HTTPException, Request, status from fastapi import HTTPException, status
from redis.asyncio import ConnectionPool
from app.models import CachedFile as CachedFileDB from app.models import CachedFile as CachedFileDB
from app.services.cache_updater import cache_file_by_book_id from app.services.cache_updater import cache_file_by_book_id
async def get_cached_file_or_cache( async def get_cached_file_or_cache(
request: Request, object_id: int, object_type: str object_id: int, object_type: str, connection_pool: ConnectionPool
) -> CachedFileDB: ) -> CachedFileDB:
cached_file = await CachedFileDB.objects.get_or_none( cached_file = await CachedFileDB.objects.get_or_none(
object_id=object_id, object_type=object_type object_id=object_id, object_type=object_type
@@ -13,7 +15,7 @@ async def get_cached_file_or_cache(
if not cached_file: if not cached_file:
cached_file = await cache_file_by_book_id( cached_file = await cache_file_by_book_id(
{"redis": request.app.state.redis_client}, object_id, object_type object_id, object_type, redis_pool=connection_pool
) )
if not cached_file: if not cached_file:

8
src/app/views.py
View File

@@ -40,7 +40,9 @@ async def get_cached_file(request: Request, object_id: int, object_type: str):
@router.get("/download/{object_id}/{object_type}") @router.get("/download/{object_id}/{object_type}")
async def download_cached_file(request: Request, object_id: int, object_type: str): async def download_cached_file(request: Request, object_id: int, object_type: str):
cached_file = await get_cached_file_or_cache(request, object_id, object_type) cached_file = await get_cached_file_or_cache(
object_id, object_type, request.app.state.redis_pool
)
cache_data: dict = cached_file.data # type: ignore cache_data: dict = cached_file.data # type: ignore
data = await download_file_from_cache( data = await download_file_from_cache(
@@ -49,7 +51,9 @@ async def download_cached_file(request: Request, object_id: int, object_type: st
if data is None: if data is None:
await CachedFileDB.objects.filter(id=cached_file.id).delete() await CachedFileDB.objects.filter(id=cached_file.id).delete()
cached_file = await get_cached_file_or_cache(request, object_id, object_type) cached_file = await get_cached_file_or_cache(
object_id, object_type, request.app.state.redis_pool
)
cache_data: dict = cached_file.data # type: ignore cache_data: dict = cached_file.data # type: ignore
data = await download_file_from_cache( data = await download_file_from_cache(