✨ Quick save

classquiz/storage/__init__.py

@@ -50,8 +50,8 @@ class Storage:
         """
         yield self.instance.download(file_name)
 
-    async def upload(self, file_name: str, file_data: BinaryIO) -> None:
-        return await self.instance.upload(file=file_data, file_name=file_name)
+    async def upload(self, file_name: str, file_data: BinaryIO, mime_type: str | None = None) -> None:
+        return await self.instance.upload(file=file_data, file_name=file_name, mime_type=mime_type)
 
     async def delete(self, file_names: [str]) -> None:
         return await self.instance.delete(file_names=file_names)

classquiz/storage/local_storage.py

@@ -27,7 +27,7 @@ class LocalStorage:
         except FileNotFoundError:
             yield None
 
-    async def upload(self, file_name: str, data: BinaryIO) -> None:
+    async def upload(self, file_name: str, data: BinaryIO, mime_type: str | None = None) -> None:
         async with aiofiles.open(file=os.path.join(self.base_path, file_name), mode="wb") as f:
             await aioshutil_copyfileobj(data, f)
 

classquiz/storage/s3_storage.py

@@ -112,9 +112,7 @@ class S3Storage:
 
         return headers, request_url
 
-    async def upload(
-        self, file: BinaryIO, file_name: str, content_type: str | None = "application/octet-stream"
-    ) -> None:
+    async def upload(self, file: BinaryIO, file_name: str, mime_type: str | None = "application/octet-stream") -> None:
         headers, url = self._generate_aws_signature_v4(method="PUT", path=f"/{file_name}")
         async with ClientSession() as session, session.put(url, headers=headers, data=file) as resp:
             if resp.status == 200: