mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
62d19e5cc1
Because of the filter implementation here we nedd to work only with clone of the headers and request. subj to refactor
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from six.moves.urllib.parse import urlparse, urlencode, urlunparse
|
|
import copy
|
|
|
|
|
|
def _remove_headers(request, headers_to_remove):
|
|
headers = copy.copy(request.headers)
|
|
headers_to_remove = [h.lower() for h in headers_to_remove]
|
|
keys = [k for k in headers if k.lower() in headers_to_remove]
|
|
if keys:
|
|
for k in keys:
|
|
headers.pop(k)
|
|
request.headers = headers
|
|
return request
|
|
|
|
|
|
def _remove_query_parameters(request, query_parameters_to_remove):
|
|
query = request.query
|
|
new_query = [(k, v) for (k, v) in query
|
|
if k not in query_parameters_to_remove]
|
|
if len(new_query) != len(query):
|
|
uri_parts = list(urlparse(request.uri))
|
|
uri_parts[4] = urlencode(new_query)
|
|
request.uri = urlunparse(uri_parts)
|
|
print urlunparse(uri_parts)
|
|
return request
|
|
|
|
|
|
def filter_request(
|
|
request,
|
|
filter_headers,
|
|
filter_query_parameters,
|
|
before_record
|
|
):
|
|
request = copy.copy(request) # don't mutate request object
|
|
if hasattr(request, 'headers') and filter_headers:
|
|
request = _remove_headers(request, filter_headers)
|
|
if filter_query_parameters:
|
|
request = _remove_query_parameters(request, filter_query_parameters)
|
|
if before_record:
|
|
request = before_record(request)
|
|
return request
|