Skip to content

Commit

Permalink
Revert to main branch add_file to test again
Browse files Browse the repository at this point in the history
  • Loading branch information
MHHukiewitz committed Feb 5, 2024
1 parent 0d21dc0 commit 0563272
Showing 1 changed file with 7 additions and 13 deletions.
20 changes: 7 additions & 13 deletions src/aleph/web/controllers/storage.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,18 +134,9 @@ class MultipartUploadedFile:
_content: Optional[bytes]
size: int

def __init__(self, file_field: FileField, size: Optional[int] = None):
def __init__(self, file_field: FileField, size: int):
self.file_field = file_field
if size is not None:
self.size = size
else:
try:
content_length_str = file_field.headers["Content-Length"]
self.size = int(content_length_str)
except (KeyError, ValueError):
raise web.HTTPUnprocessableEntity(
reason="Invalid/missing Content-Length header."
)
self.size = size
self._content = None

@property
Expand Down Expand Up @@ -253,8 +244,11 @@ async def storage_add_file(request: web.Request):
raise web.HTTPUnprocessableEntity(reason="Missing 'file' in multipart form.")

if isinstance(file_field, FileField):
content_length = int(headers["Content-Length"]) if headers.get("Content-Length") else None
uploaded_file: UploadedFile = MultipartUploadedFile(file_field, content_length)
try:
content_length = int(headers.get("Content-Length", file_field.headers["Content-Length"]))
uploaded_file: UploadedFile = MultipartUploadedFile(file_field, content_length)
except (KeyError, ValueError):
raise web.HTTPUnprocessableEntity(reason="Invalid/missing Content-Length header.")
else:
uploaded_file = RawUploadedFile(file_field)

Expand Down

0 comments on commit 0563272

Please sign in to comment.