flibusta-apps/telegram_files_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/telegram_files_server.git synced 2025-12-06 12:35:39 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-08-06 15:04:22 +02:00
parent c472d5038d
commit d1ccc71e4b
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
fastapi_file_server/app/services/file_uploader.py
View File

@@ -35,8 +35,11 @@ class FileUploader:
def user_storages(cls) -> list[UserStorage]:
return StoragesContainer.USER_STORAGES
def __init__(self, file: UploadFile, caption: Optional[str] = None) -> None:
def __init__(
self, file: UploadFile, file_size: int, caption: Optional[str] = None
) -> None:
self.file = file
self.file_size = file_size
self.caption = caption
self.upload_data: Optional[Data] = None
@@ -61,7 +64,9 @@ class FileUploader:
wrapped = Wrapper(self.file.file, self.file.filename)
data = await storage.upload(wrapped, caption=self.caption)
data = await storage.upload(
wrapped, file_size=self.file_size, caption=self.caption
)
if not data:
return False
@@ -101,9 +106,9 @@ class FileUploader:
@classmethod
async def upload(
cls, file: UploadFile, caption: Optional[str] = None
cls, file: UploadFile, file_size: int, caption: Optional[str] = None
) -> Optional[UploadedFile]:
uploader = cls(file, caption)
uploader = cls(file, file_size, caption)
upload_result = await uploader._upload()
if not upload_result: