Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 09:13:23 +00:00
Code Releases Activity

Updated _remove_query_parameters to use latest Request Api

Browse Source
This commit is contained in:
Max Shytikov
2014-04-30 03:29:09 +02:00
parent faa83b9aba
commit 998dde61ec
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
vcr/filters.py
View File

@@ -1,4 +1,4 @@
from six.moves.urllib.parse import urlparse, parse_qsl, urlencode from six.moves.urllib.parse import urlparse, urlencode, urlunparse
import copy import copy
@@ -11,15 +11,14 @@ def _remove_headers(request, headers_to_remove):
def _remove_query_parameters(request, query_parameters_to_remove): def _remove_query_parameters(request, query_parameters_to_remove):
if not hasattr(request, 'path' or not query_parameters_to_remove): query = request.query
return request new_query = [(k, v) for (k, v) in query
url = urlparse(request.url) if k not in query_parameters_to_remove]
q = parse_qsl(url.query) if len(new_query) != len(query):
q = [(k, v) for k, v in q if k not in query_parameters_to_remove] uri_parts = list(urlparse(request.uri))
if q: uri_parts[4] = urlencode(new_query)
request.path = url.path + '?' + urlencode(q) request.uri = urlunparse(uri_parts)
else: print urlunparse(uri_parts)
request.path = url.path
return request return request