QingdaoU/OnlineJudge
1
0
Fork 0
mirror of https://github.com/QingdaoU/OnlineJudge.git synced 2024-09-21 00:13:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add more validation in paginate_data function

Browse Source
This commit is contained in:
virusdefender 2016-10-31 00:18:17 +08:00
parent d3fa05242a
commit de4d50dd99
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
utils/shortcuts.py
View File

@ -76,16 +76,24 @@ def paginate_data(request, query_set, object_serializer):
try:
limit = int(request.GET.get("limit", "100"))
except Exception:
except ValueError:
limit = 100
if limit < 0:
limit = 100
try:
offset = int(request.GET.get("offset", "100"))
except Exception:
offset = 100
offset = int(request.GET.get("offset", "0"))
except ValueError:
offset = 0
if offset < 0:
offset = 0
count = query_set.count()
results = object_serializer(query_set[offset:offset + limit], many=True).data
results = query_set[offset:offset + limit]
if object_serializer:
count = query_set.count()
results = object_serializer(results, many=True).data
else:
count = len(query_set)
data = {"results": results,
"count": count}