diff options
| author | Maximilian Hils <git@maximilianhils.com> | 2016-07-01 14:10:48 -0700 |
|---|---|---|
| committer | Maximilian Hils <git@maximilianhils.com> | 2016-07-01 14:10:48 -0700 |
| commit | 536c7acd13426d42dc863ae8b50e6c3a4cb2e858 (patch) | |
| tree | e868530aef8353273a3107b57645fd5b02e717b0 /netlib/http/message.py | |
| parent | e0ed7699ca1258414a99812720f168e14a6ca219 (diff) | |
| download | mitmproxy-536c7acd13426d42dc863ae8b50e6c3a4cb2e858.tar.gz mitmproxy-536c7acd13426d42dc863ae8b50e6c3a4cb2e858.tar.bz2 mitmproxy-536c7acd13426d42dc863ae8b50e6c3a4cb2e858.zip | |
py3++
Diffstat (limited to 'netlib/http/message.py')
| -rw-r--r-- | netlib/http/message.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/netlib/http/message.py b/netlib/http/message.py index b633b671..0583c246 100644 --- a/netlib/http/message.py +++ b/netlib/http/message.py @@ -1,5 +1,6 @@ from __future__ import absolute_import, print_function, division +import re import warnings import six @@ -196,11 +197,14 @@ class Message(basetypes.Serializable): Returns: The number of replacements made. """ - # TODO: Proper distinction between text and bytes. + if isinstance(pattern, six.text_type): + pattern = strutils.escaped_str_to_bytes(pattern) + if isinstance(repl, six.text_type): + repl = strutils.escaped_str_to_bytes(repl) replacements = 0 if self.content: with decoded(self): - self.content, replacements = strutils.safe_subn( + self.content, replacements = re.subn( pattern, repl, self.content, flags=flags ) replacements += self.headers.replace(pattern, repl, flags) |