encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-10 07:14:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Py3 compat

Browse Source
This commit is contained in:
Tom Christie 2016-09-09 17:05:32 +01:00
parent b56e7d3a64
commit b41ec30ff7
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_api_client.py
View File

@ -61,13 +61,19 @@ def get_schema():
) )
def _iterlists(querydict):
if hasattr(querydict, 'iterlists'):
return querydict.iterlists()
return querydict.lists()
def _get_query_params(request): def _get_query_params(request):
# Return query params in a plain dict, using a list value if more # Return query params in a plain dict, using a list value if more
# than one item is present for a given key. # than one item is present for a given key.
return { return {
key: (value[0] if len(value) == 1 else value) key: (value[0] if len(value) == 1 else value)
for key, value in for key, value in
request.query_params.iterlists() _iterlists(request.query_params)
} }
@ -76,11 +82,11 @@ def _get_data(request):
return request.data return request.data
# Coerce multidict into regular dict, and remove files to # Coerce multidict into regular dict, and remove files to
# make assertions simpler. # make assertions simpler.
if hasattr(request.data, 'iterlists'): if hasattr(request.data, 'iterlists') or hasattr(request.data, 'lists'):
# Use a list value if a QueryDict contains multiple items for a key. # Use a list value if a QueryDict contains multiple items for a key.
return { return {
key: value[0] if len(value) == 1 else value key: value[0] if len(value) == 1 else value
for key, value in request.data.iterlists() for key, value in _iterlists(request.data)
if key not in request.FILES if key not in request.FILES
} }
return { return {