Merge pull request #8 from csev/issue-4

issue-4 - Add decode and fix content-length
edcrewe · Oct 26, 2019 · b87b1ab · b87b1ab
2 parents fa4c7da + 61359cf
commit b87b1ab
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cookieless/middleware.py b/cookieless/middleware.py
@@ -199,7 +199,7 @@ def new_url(match):
                 try:
                     response.content = self._re_links.sub(
                         new_url, response.content.decode()
-                    )
+                    ).encode()
                 except:
                     pass
 
@@ -217,9 +217,11 @@ def new_url(match):
                 try:
                     response.content = self._re_forms.sub(
                         repl_form, response.content.decode()
-                    )
+                    ).encode()
                 except:
                     pass
+            response['Content-Length'] = len(response.content);
             return response
         else:
+            response['Content-Length'] = len(response.content);
             return response