Fixed Content-Length key error during image upload.

closes #921
pulp · Jul 29, 2022 · e7b75cf · e7b75cf
1 parent b4f2004
commit e7b75cf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CHANGES/921.bugfix b/CHANGES/921.bugfix
@@ -0,0 +1 @@
+Fixed the Content-Length key error raised when uploading images.
diff --git a/pulp_container/app/registry_api.py b/pulp_container/app/registry_api.py
@@ -634,7 +634,7 @@ def partial_update(self, request, path, pk=None):
             length = end - start + 1
 
         else:
-            length = int(request.headers["Content-Length"])
+            length = int(request.headers.get("Content-Length", 0))
             start = 0
 
         with transaction.atomic():
@@ -662,6 +662,8 @@ def partial_update(self, request, path, pk=None):
                     artifact = Artifact.objects.get(sha256=artifact.sha256)
                     artifact.touch()
                 upload.artifact = artifact
+                if not length:
+                    length = artifact.size
 
             upload.size += length
             upload.save()
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Fixed the Content-Length key error raised when uploading images.