kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-05-30 00:03:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added replace for request and headers - Fixed missing replace option

Browse Source
This commit is contained in:
arjun23496
2016-08-29 16:50:14 +05:30
parent 70f94c7171
commit 99744cc780
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
netlib/http/headers.py
View File

@@ -158,7 +158,7 @@ class Headers(multidict.MultiDict):
else:
return super(Headers, self).items()
def replace(self, pattern, repl, flags=0):
def replace(self, pattern, repl, flags=0, count=0):
"""
Replaces a regular expression pattern with repl in each "name: value"
header line.
@@ -175,7 +175,7 @@ class Headers(multidict.MultiDict):
fields = []
for name, value in self.fields:
line, n = pattern.subn(repl, name + b": " + value)
line, n = pattern.subn(repl, name + b": " + value, count=count)
try:
name, value = line.split(b": ", 1)
except ValueError:
@@ -183,6 +183,7 @@ 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
fields.append((name, value))
self.fields = tuple(fields)