From a48f621bae3d25e9a1ea3cbc0df833446d43ea64 Mon Sep 17 00:00:00 2001 From: Max Shytikov Date: Wed, 30 Apr 2014 03:02:24 +0200 Subject: [PATCH] Updated _remove_headers to use latest Headers structure Probably we need API in Request object like 'remove_header' --- vcr/filters.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/vcr/filters.py b/vcr/filters.py index 06838d6..0e7d106 100644 --- a/vcr/filters.py +++ b/vcr/filters.py @@ -3,11 +3,10 @@ import copy def _remove_headers(request, headers_to_remove): - out = [] - for k, v in request.headers: - if k.lower() not in [h.lower() for h in headers_to_remove]: - out.append((k, v)) - request.headers = frozenset(out) + headers_to_remove = [h.lower() for h in headers_to_remove] + keys = [k for k in request.headers if k.lower() in headers_to_remove] + for k in keys: + request.headers.pop(k) return request