Added replace for request and headers - Fixed missing replace option

1 files changed, 3 insertions, 2 deletions

diff --git a/netlib/http/headers.py b/netlib/http/headers.py
index 36e5060c..60a36515 100644
--- a/netlib/http/headers.py
+++ b/netlib/http/headers.py
@@ -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)