feat: handle content length for requests

msqd · Jul 29, 2024 · 82d5ad9 · 82d5ad9
1 parent e72a045
commit 82d5ad9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/harp/http/requests.py b/harp/http/requests.py
@@ -108,6 +108,7 @@ async def aread(self):
         nothing. This method does nothing if the body has already been read."""
         if not hasattr(self, "_body"):
             self._body = b"".join([part async for part in self._stream])
+            self.headers["content-length"] = str(len(self.body))
         if not isinstance(self._stream, ByteStream):
             self._stream = ByteStream(self._body)
         return self.body

diff --git a/harp/http/responses.py b/harp/http/responses.py
@@ -46,6 +46,10 @@ def status(self) -> int:
     def headers(self) -> CIMultiDict:
         return self._headers
 
+    @headers.setter
+    def headers(self, headers: CIMultiDict):
+        self._headers = CIMultiDict(headers)
+
     @property
     def content_type(self) -> str:
         return self._headers.get("content-type", "text/plain")