Use six.ensure_binary

webrecorder · May 14, 2024 · ced80a5 · ced80a5
1 parent 872e63e
commit ced80a5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/warcio/capture_http.py b/warcio/capture_http.py
@@ -2,6 +2,7 @@
 
 from io import BytesIO
 
+from six import ensure_binary
 from six.moves import http_client as httplib
 
 from contextlib import contextmanager
@@ -165,11 +166,11 @@ def set_remote_ip(self, remote_ip):
 
     def write_request(self, buff):
         if self.started_req:
-            self.request_out.write(buff.encode())
+            self.request_out.write(ensure_binary(buff))
 
     def write_response(self, buff):
         if self.started_req:
-            self.response_out.write(buff.encode())
+            self.response_out.write(ensure_binary(buff))
 
     def _create_record(self, out, record_type):
         length = out.tell()