This commit is contained in:
2023-08-06 16:09:26 +02:00
parent 8dc25ef953
commit 2e8f4ac3a9

View File

@@ -1,4 +1,4 @@
from typing import Any, BinaryIO, Optional from typing import Optional
from fastapi import UploadFile from fastapi import UploadFile
@@ -6,21 +6,6 @@ from app.serializers import Data, UploadBackend, UploadedFile
from app.services.storages import BotStorage, StoragesContainer, UserStorage from app.services.storages import BotStorage, StoragesContainer, UserStorage
class Wrapper(BinaryIO):
def __init__(self, wrapped: Any, filename: str) -> None:
self.wrapped = wrapped
self.filename = filename
def seekable(self):
return True
def __getattr__(self, __name: str) -> Any:
if __name == "name":
return self.filename
return getattr(self.wrapped, __name)
class FileUploader: class FileUploader:
_bot_storage_index = 0 _bot_storage_index = 0
_user_storage_index = 0 _user_storage_index = 0
@@ -62,10 +47,8 @@ class FileUploader:
assert self.file.filename assert self.file.filename
wrapped = Wrapper(self.file.file, self.file.filename)
data = await storage.upload( data = await storage.upload(
wrapped, self.file, # type: ignore
file_size=self.file_size, file_size=self.file_size,
filename=self.file.filename, filename=self.file.filename,
caption=self.caption, caption=self.caption,