2017-01-23 08:48:04 +00:00
|
|
|
import functools
|
2016-11-19 04:32:23 +00:00
|
|
|
import json
|
|
|
|
import logging
|
|
|
|
|
|
|
|
from django.http import HttpResponse, QueryDict
|
|
|
|
from django.utils.decorators import method_decorator
|
|
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
from django.views.generic import View
|
|
|
|
|
2017-01-25 08:29:00 +00:00
|
|
|
logger = logging.getLogger("")
|
2016-11-19 04:32:23 +00:00
|
|
|
|
|
|
|
|
|
|
|
class ContentType(object):
|
|
|
|
json_request = "application/json"
|
|
|
|
json_response = "application/json;charset=UTF-8"
|
|
|
|
url_encoded_request = "application/x-www-form-urlencoded"
|
|
|
|
binary_response = "application/octet-stream"
|
|
|
|
|
|
|
|
|
|
|
|
class JSONParser(object):
|
|
|
|
content_type = ContentType.json_request
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def parse(body):
|
|
|
|
return json.loads(body.decode("utf-8"))
|
|
|
|
|
|
|
|
|
|
|
|
class URLEncodedParser(object):
|
|
|
|
content_type = ContentType.url_encoded_request
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def parse(body):
|
2016-12-03 12:22:14 +00:00
|
|
|
return QueryDict(body)
|
2016-11-19 04:32:23 +00:00
|
|
|
|
|
|
|
|
|
|
|
class JSONResponse(object):
|
|
|
|
content_type = ContentType.json_response
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def response(cls, data):
|
|
|
|
resp = HttpResponse(json.dumps(data, indent=4), content_type=cls.content_type)
|
|
|
|
resp.data = data
|
|
|
|
return resp
|
|
|
|
|
|
|
|
|
|
|
|
class APIView(View):
|
|
|
|
"""
|
|
|
|
Django view的父类, 和django-rest-framework的用法基本一致
|
|
|
|
- request.data获取解析之后的json或者urlencoded数据, dict类型
|
|
|
|
- self.success, self.error和self.invalid_serializer可以根据业需求修改,
|
|
|
|
写到父类中是为了不同的人开发写法统一,不再使用自己的success/error格式
|
|
|
|
- self.response 返回一个django HttpResponse, 具体在self.response_class中实现
|
|
|
|
- parse请求的类需要定义在request_parser中, 目前只支持json和urlencoded的类型, 用来解析请求的数据
|
|
|
|
"""
|
|
|
|
request_parsers = (JSONParser, URLEncodedParser)
|
|
|
|
response_class = JSONResponse
|
|
|
|
|
|
|
|
def _get_request_data(self, request):
|
2017-01-24 08:48:39 +00:00
|
|
|
if request.method not in ["GET", "DELETE"]:
|
2016-11-19 04:32:23 +00:00
|
|
|
body = request.body
|
|
|
|
content_type = request.META.get("CONTENT_TYPE")
|
|
|
|
if not content_type:
|
|
|
|
raise ValueError("content_type is required")
|
|
|
|
for parser in self.request_parsers:
|
|
|
|
if content_type.startswith(parser.content_type):
|
|
|
|
break
|
|
|
|
else:
|
|
|
|
raise ValueError("unknown content_type '%s'" % content_type)
|
|
|
|
if body:
|
|
|
|
return parser.parse(body)
|
|
|
|
return {}
|
|
|
|
return request.GET
|
|
|
|
|
|
|
|
def response(self, data):
|
|
|
|
return self.response_class.response(data)
|
|
|
|
|
|
|
|
def success(self, data=None):
|
|
|
|
return self.response({"error": None, "data": data})
|
|
|
|
|
|
|
|
def error(self, msg, err="error"):
|
|
|
|
return self.response({"error": err, "data": msg})
|
|
|
|
|
|
|
|
def invalid_serializer(self, serializer):
|
|
|
|
for k, v in serializer.errors.items():
|
|
|
|
if k != "non_field_errors":
|
|
|
|
return self.error(err="invalid-" + k, msg=k + ": " + v[0])
|
|
|
|
else:
|
|
|
|
return self.error(err="invalid-field", msg=k[0])
|
|
|
|
|
|
|
|
def server_error(self):
|
|
|
|
return self.error(err="server-error", msg="server error")
|
|
|
|
|
|
|
|
def paginate_data(self, request, query_set, object_serializer=None):
|
|
|
|
"""
|
|
|
|
:param request: django的request
|
|
|
|
:param query_set: django model的query set或者其他list like objects
|
|
|
|
:param object_serializer: 用来序列化query set, 如果为None, 则直接对query set切片
|
|
|
|
:return:
|
|
|
|
"""
|
|
|
|
need_paginate = request.GET.get("limit", None)
|
|
|
|
if need_paginate is None:
|
|
|
|
if object_serializer:
|
|
|
|
return object_serializer(query_set, many=True).data
|
|
|
|
else:
|
|
|
|
return query_set
|
|
|
|
try:
|
|
|
|
limit = int(request.GET.get("limit", "100"))
|
|
|
|
except ValueError:
|
|
|
|
limit = 100
|
|
|
|
if limit < 0:
|
|
|
|
limit = 100
|
|
|
|
try:
|
|
|
|
offset = int(request.GET.get("offset", "0"))
|
|
|
|
except ValueError:
|
|
|
|
offset = 0
|
|
|
|
if offset < 0:
|
|
|
|
offset = 0
|
|
|
|
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,
|
|
|
|
"total": count}
|
|
|
|
return data
|
|
|
|
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
2016-12-03 12:22:14 +00:00
|
|
|
if self.request_parsers:
|
|
|
|
try:
|
|
|
|
request.data = self._get_request_data(self.request)
|
|
|
|
except ValueError as e:
|
|
|
|
return self.error(err="invalid-request", msg=str(e))
|
2016-11-19 04:32:23 +00:00
|
|
|
try:
|
|
|
|
return super(APIView, self).dispatch(request, *args, **kwargs)
|
|
|
|
except Exception as e:
|
|
|
|
logger.exception(e)
|
|
|
|
return self.server_error()
|
|
|
|
|
|
|
|
|
|
|
|
class CSRFExemptAPIView(APIView):
|
|
|
|
@method_decorator(csrf_exempt)
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
|
|
return super(CSRFExemptAPIView, self).dispatch(request, *args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
def validate_serializer(serializer):
|
|
|
|
"""
|
|
|
|
@validate_serializer(TestSerializer)
|
|
|
|
def post(self, request):
|
|
|
|
return self.success(request.data)
|
|
|
|
"""
|
|
|
|
def validate(view_method):
|
2016-12-03 12:22:14 +00:00
|
|
|
@functools.wraps(view_method)
|
2016-11-19 04:32:23 +00:00
|
|
|
def handle(*args, **kwargs):
|
|
|
|
self = args[0]
|
|
|
|
request = args[1]
|
|
|
|
s = serializer(data=request.data)
|
|
|
|
if s.is_valid():
|
|
|
|
request.data = s.data
|
|
|
|
request.serializer = s
|
|
|
|
return view_method(*args, **kwargs)
|
|
|
|
else:
|
|
|
|
return self.invalid_serializer(s)
|
|
|
|
|
|
|
|
return handle
|
|
|
|
|
2016-12-03 12:22:14 +00:00
|
|
|
return validate
|