Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 10:16:27 +00:00
Code Issues 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
Show Stats 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)

6
netlib/http/request.py
View File

@ -80,7 +80,7 @@ class Request(message.Message):
self.method, hostport, path
)
def replace(self, pattern, repl, flags=0):
def replace(self, pattern, repl, flags=0, count=0):
"""
Replaces a regular expression pattern with repl in the headers, the
request path and the body of the request. Encoded content will be
@ -94,9 +94,9 @@ class Request(message.Message):
if isinstance(repl, six.text_type):
repl = strutils.escaped_str_to_bytes(repl)
c = super(Request, self).replace(pattern, repl, flags)
c = super(Request, self).replace(pattern, repl, flags, count)
self.path, pc = re.subn(
pattern, repl, self.data.path, flags=flags
pattern, repl, self.data.path, flags=flags, count=count
)
c += pc
return c