mirror of
https://github.com/flibusta-apps/telegram_files_server.git
synced 2025-12-06 20:45:37 +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
|
||||
|
||||
@@ -6,21 +6,6 @@ from app.serializers import Data, UploadBackend, UploadedFile
|
||||
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:
|
||||
_bot_storage_index = 0
|
||||
_user_storage_index = 0
|
||||
@@ -62,10 +47,8 @@ class FileUploader:
|
||||
|
||||
assert self.file.filename
|
||||
|
||||
wrapped = Wrapper(self.file.file, self.file.filename)
|
||||
|
||||
data = await storage.upload(
|
||||
wrapped,
|
||||
self.file, # type: ignore
|
||||
file_size=self.file_size,
|
||||
filename=self.file.filename,
|
||||
caption=self.caption,
|
||||
|
||||
Reference in New Issue
Block a user