20 lines
744 B
Python
20 lines
744 B
Python
from rest_framework.response import Response
|
|
|
|
|
|
class ApiResponse(Response):
|
|
|
|
def __init__(self, code=None, message="ok", data=None, status=200, total=None, res_type='success',
|
|
template_name=None, headers=None, exception=False, content_type='application/json'):
|
|
super(Response, self).__init__(None, status=status)
|
|
code = code or 0
|
|
self.data = {"code": code, "message": message, "result": data, 'type': res_type}
|
|
if total:
|
|
self.data.update({'total': total})
|
|
self.template_name = template_name
|
|
self.exception = exception
|
|
self.content_type = content_type
|
|
|
|
if headers:
|
|
for name, value in headers.items():
|
|
self[name] = value
|