From 04c85439a6155fb32fea998a1d73ae9e7b926e46 Mon Sep 17 00:00:00 2001 From: Kurbanov Bulat Date: Sun, 21 Nov 2021 21:38:36 +0300 Subject: [PATCH] Fix --- src/app/services/base.py | 2 +- src/app/services/fl_downloader.py | 4 ++-- src/app/views.py | 6 +++--- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/app/services/base.py b/src/app/services/base.py index 8947bac..11b177e 100644 --- a/src/app/services/base.py +++ b/src/app/services/base.py @@ -3,5 +3,5 @@ from typing import Protocol class BaseDownloader(Protocol): @classmethod - async def download(cls, book_id: int, file_type: str, source_id: int) -> tuple[bytes, str]: + async def download(cls, remote_id: int, file_type: str, source_id: int) -> tuple[bytes, str]: ... diff --git a/src/app/services/fl_downloader.py b/src/app/services/fl_downloader.py index 1b64c38..f70aef7 100644 --- a/src/app/services/fl_downloader.py +++ b/src/app/services/fl_downloader.py @@ -192,6 +192,6 @@ class FLDownloader(BaseDownloader): return tasks[0].result() @classmethod - async def download(cls, book_id: int, file_type: str, source_id: int) -> tuple[bytes, str]: - downloader = cls(book_id, file_type, source_id) + async def download(cls, remote_id: int, file_type: str, source_id: int) -> tuple[bytes, str]: + downloader = cls(remote_id, file_type, source_id) return await downloader._download() diff --git a/src/app/views.py b/src/app/views.py index 0f84a1e..e3773fd 100644 --- a/src/app/views.py +++ b/src/app/views.py @@ -12,11 +12,11 @@ router = APIRouter( ) -@router.get("/download/{source_id}/{book_id}/{file_type}") -async def download(source_id: int, book_id: int, file_type: str): +@router.get("/download/{source_id}/{remote_id}/{file_type}") +async def download(source_id: int, remote_id: int, file_type: str): downloader = await DownloadersManager.get_downloader(source_id) - content, filename = await downloader.download(book_id, file_type, source_id) + content, filename = await downloader.download(remote_id, file_type, source_id) return Response( content,