From 8dc25ef9533890fcdddaf7d5e8c40cf23c531dc1 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Sun, 6 Aug 2023 15:45:02 +0200 Subject: [PATCH] Fix --- fastapi_file_server/app/services/file_uploader.py | 5 ++++- fastapi_file_server/app/services/storages.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/fastapi_file_server/app/services/file_uploader.py b/fastapi_file_server/app/services/file_uploader.py index 8a225e5..a76b69f 100644 --- a/fastapi_file_server/app/services/file_uploader.py +++ b/fastapi_file_server/app/services/file_uploader.py @@ -65,7 +65,10 @@ class FileUploader: wrapped = Wrapper(self.file.file, self.file.filename) data = await storage.upload( - wrapped, file_size=self.file_size, caption=self.caption + wrapped, + file_size=self.file_size, + filename=self.file.filename, + caption=self.caption, ) if not data: diff --git a/fastapi_file_server/app/services/storages.py b/fastapi_file_server/app/services/storages.py index b0f36e9..f55b905 100644 --- a/fastapi_file_server/app/services/storages.py +++ b/fastapi_file_server/app/services/storages.py @@ -22,11 +22,14 @@ class BaseStorage: async def upload( self, file: telethon.hints.FileLike, + filename: str, file_size: int, caption: Optional[str] = None, ) -> Optional[tuple[int, int]]: try: - uploaded_file = await self.client.upload_file(file, file_size=file_size) + uploaded_file = await self.client.upload_file( + file, file_size=file_size, file_name=filename + ) if caption: message = await self.client.send_file(