mirror of
https://github.com/flibusta-apps/telegram_files_server.git
synced 2025-12-06 12:35:39 +01:00
Fix
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user