🐛 Fix s3 upload failing

This commit is contained in:
Mawoka
2024-06-15 21:16:27 +02:00
parent dbb8bdddb0
commit fe12cf099f
+2
View File
@@ -5,6 +5,7 @@
import hashlib import hashlib
import hmac import hmac
import sys
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Tuple, BinaryIO, Generator from typing import Tuple, BinaryIO, Generator
@@ -110,6 +111,7 @@ class S3Storage:
# skipcq: PYL-W0613 # skipcq: PYL-W0613
async def upload(self, file: BinaryIO, file_name: str, mime_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}") headers, url = self._generate_aws_signature_v4(method="PUT", path=f"/{file_name}")
headers["Content-Length"] = sys.getsizeof(file)
async with ClientSession() as session, session.put(url, headers=headers, data=file) as resp: async with ClientSession() as session, session.put(url, headers=headers, data=file) as resp:
if resp.status == 200: if resp.status == 200:
return None return None