diff options
| author | arjun23496 <arjun.k018@gmail.com> | 2016-08-31 16:11:59 +0530 |
|---|---|---|
| committer | arjun23496 <arjun.k018@gmail.com> | 2016-08-31 16:11:59 +0530 |
| commit | dd2e0d1df3eeb934c291dd6197f722e419dd3e43 (patch) | |
| tree | ea77f34522f034ee7a6c3d0c149c896458031e3d /netlib/http/headers.py | |
| parent | d2cfa5ca41c0d8b55000cc4c9fca924c28dfc14a (diff) | |
| download | mitmproxy-dd2e0d1df3eeb934c291dd6197f722e419dd3e43.tar.gz mitmproxy-dd2e0d1df3eeb934c291dd6197f722e419dd3e43.tar.bz2 mitmproxy-dd2e0d1df3eeb934c291dd6197f722e419dd3e43.zip | |
minor bug fix header replace
Diffstat (limited to 'netlib/http/headers.py')
| -rw-r--r-- | netlib/http/headers.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/netlib/http/headers.py b/netlib/http/headers.py index f4cf8b7b..5f724f6d 100644 --- a/netlib/http/headers.py +++ b/netlib/http/headers.py @@ -172,6 +172,7 @@ class Headers(multidict.MultiDict): repl = strutils.escaped_str_to_bytes(repl) pattern = re.compile(pattern, flags) replacements = 0 + flag_count = count > 0 fields = [] for name, value in self.fields: @@ -183,10 +184,13 @@ class Headers(multidict.MultiDict): # There's not much we can do about this, so we just keep the header as-is. pass else: - count -= n replacements += n - if count == 0: - break; + + if flag_count: + count -= n + if count == 0: + break; + fields.append((name, value)) self.fields = tuple(fields) return replacements |