50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
from __future__ import print_function
|
|
import datetime
|
|
from django.utils import datetime_safe
|
|
from django.http import HttpResponse, Http404
|
|
import json
|
|
from django.conf import settings
|
|
|
|
class HttpErrorJson(Http404):
|
|
def __init__(self, response):
|
|
self.response = response
|
|
|
|
def json_response(data=None, status=200, text='ok'):
|
|
if not data:
|
|
data = {}
|
|
return {'status': {'code': status, 'text': text}, 'data': data}
|
|
|
|
def _to_json(python_object):
|
|
if isinstance(python_object, datetime.datetime):
|
|
if python_object.year < 1900:
|
|
tt = python_object.timetuple()
|
|
return '%d-%02d-%02dT%02d:%02d%02dZ' % tuple(list(tt)[:6])
|
|
return python_object.strftime('%Y-%m-%dT%H:%M:%SZ')
|
|
if isinstance(python_object, datetime_safe.datetime):
|
|
return python_object.strftime('%Y-%m-%dT%H:%M:%SZ')
|
|
raise TypeError(u'%s %s is not JSON serializable' % (repr(python_object), type(python_object)))
|
|
|
|
def render_to_json_response(dictionary, content_type="text/json", status=200):
|
|
indent=None
|
|
if settings.DEBUG:
|
|
content_type = "text/javascript"
|
|
indent = 2
|
|
if getattr(settings, 'JSON_DEBUG', False):
|
|
print(json.dumps(dictionary, indent=2, default=_to_json, ensure_ascii=False).encode('utf-8'))
|
|
|
|
return HttpResponse(json.dumps(dictionary, indent=indent, default=_to_json,
|
|
ensure_ascii=False).encode('utf-8'), content_type=content_type, status=status)
|
|
|
|
def get_object_or_404_json(klass, *args, **kwargs):
|
|
from django.shortcuts import _get_queryset
|
|
queryset = _get_queryset(klass)
|
|
try:
|
|
return queryset.get(*args, **kwargs)
|
|
except queryset.model.DoesNotExist:
|
|
response = {'status': {'code': 404,
|
|
'text': '%s not found' % queryset.model._meta.object_name}}
|
|
raise HttpErrorJson(response)
|
|
|