Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +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
@@ -11,15 +11,14 @@ def _remove_headers(request, headers_to_remove):
def _remove_query_parameters(request, query_parameters_to_remove):
if not hasattr(request, 'path' or not query_parameters_to_remove):
return request
url = urlparse(request.url)
q = parse_qsl(url.query)
q = [(k, v) for k, v in q if k not in query_parameters_to_remove]
if q:
request.path = url.path + '?' + urlencode(q)
else:
request.path = url.path
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