mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2025-01-01 10:02:01 +00:00
104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
# coding=utf-8
|
||
import hashlib
|
||
import time
|
||
import random
|
||
|
||
from django.shortcuts import render
|
||
from django.core.paginator import Paginator
|
||
|
||
from rest_framework.response import Response
|
||
|
||
|
||
def error_page(request, error_reason):
|
||
return render(request, "utils/error.html", {"error": error_reason})
|
||
|
||
|
||
def error_response(error_reason):
|
||
return Response(data={"code": 1, "data": error_reason})
|
||
|
||
|
||
def serializer_invalid_response(serializer):
|
||
for k, v in serializer.errors.iteritems():
|
||
return error_response(k + " : " + v[0])
|
||
|
||
|
||
def success_response(data):
|
||
return Response(data={"code": 0, "data": data})
|
||
|
||
|
||
def paginate(request, query_set, object_serializer):
|
||
"""
|
||
用于分页的函数
|
||
如果 url 里面不含有paging=true，那么将返回全部数据。类似
|
||
[
|
||
{
|
||
"username": "1111111",
|
||
"password": "123456"
|
||
}
|
||
]
|
||
如果 url 中有 paging=true 的参数，
|
||
然后还需要读取其余的两个参数，page=[int]，需要的页码，p
|
||
age_size=[int]，一页的数据条数
|
||
参数错误的时候，返回{"code": 1, "data": u"参数错误"}
|
||
返回的数据格式
|
||
{
|
||
"code": 0,
|
||
"data": {
|
||
"previous_page": null,
|
||
"results": [
|
||
{
|
||
"username": "1111111",
|
||
"password": "123456"
|
||
}
|
||
],
|
||
"next_page": 2
|
||
}
|
||
}
|
||
:param query_set 数据库查询结果
|
||
:param object_serializer: 序列化单个object的serializer
|
||
:return response
|
||
"""
|
||
need_paginate = request.GET.get("paging", None)
|
||
# 如果请求的参数里面没有paging=true的话 就返回全部数据
|
||
if need_paginate != "true":
|
||
return success_response(data=object_serializer(query_set, many=True).data)
|
||
|
||
page_size = request.GET.get("page_size", None)
|
||
if not page_size:
|
||
return error_response(u"参数错误")
|
||
|
||
try:
|
||
page_size = int(page_size)
|
||
except Exception:
|
||
return error_response(u"参数错误")
|
||
|
||
paginator = Paginator(query_set, page_size)
|
||
page = request.GET.get("page", None)
|
||
|
||
try:
|
||
current_page = paginator.page(page)
|
||
except Exception:
|
||
return error_response(u"参数错误")
|
||
|
||
data = {"results": object_serializer(current_page, many=True).data,
|
||
"previous_page": None,
|
||
"next_page": None,
|
||
"count": paginator.count,
|
||
"total_page": paginator.num_pages}
|
||
|
||
try:
|
||
data["previous_page"] = current_page.previous_page_number()
|
||
except Exception:
|
||
pass
|
||
|
||
try:
|
||
data["next_page"] = current_page.next_page_number()
|
||
except Exception:
|
||
pass
|
||
|
||
return success_response(data)
|
||
|
||
|
||
def rand_str(length=32):
|
||
string = hashlib.md5(str(time.time()) + str(random.randrange(1, 9999999900))).hexdigest()
|
||
return string[0:length] |