mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-11 20:05:04 -05:00
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
|
import sys
|
||
|
|
import cherrypy
|
||
|
|
|
||
|
|
if sys.version_info >= (2, 6):
|
||
|
|
# Python 2.6: simplejson is part of the standard library
|
||
|
|
import json
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
import simplejson as json
|
||
|
|
except ImportError:
|
||
|
|
json = None
|
||
|
|
|
||
|
|
if json is None:
|
||
|
|
def json_decode(s):
|
||
|
|
raise ValueError('No JSON library is available')
|
||
|
|
def json_encode(s):
|
||
|
|
raise ValueError('No JSON library is available')
|
||
|
|
else:
|
||
|
|
json_decode = json.JSONDecoder().decode
|
||
|
|
json_encode = json.JSONEncoder().iterencode
|
||
|
|
|
||
|
|
def json_in(force=True, debug=False):
|
||
|
|
request = cherrypy.serving.request
|
||
|
|
def json_processor(entity):
|
||
|
|
"""Read application/json data into request.json."""
|
||
|
|
if not entity.headers.get(u"Content-Length", u""):
|
||
|
|
raise cherrypy.HTTPError(411)
|
||
|
|
|
||
|
|
body = entity.fp.read()
|
||
|
|
try:
|
||
|
|
request.json = json_decode(body)
|
||
|
|
except ValueError:
|
||
|
|
raise cherrypy.HTTPError(400, 'Invalid JSON document')
|
||
|
|
if force:
|
||
|
|
request.body.processors.clear()
|
||
|
|
request.body.default_proc = cherrypy.HTTPError(
|
||
|
|
415, 'Expected an application/json content type')
|
||
|
|
request.body.processors[u'application/json'] = json_processor
|
||
|
|
|
||
|
|
def json_out(debug=False):
|
||
|
|
request = cherrypy.serving.request
|
||
|
|
response = cherrypy.serving.response
|
||
|
|
|
||
|
|
real_handler = request.handler
|
||
|
|
def json_handler(*args, **kwargs):
|
||
|
|
response.headers['Content-Type'] = 'application/json'
|
||
|
|
value = real_handler(*args, **kwargs)
|
||
|
|
return json_encode(value)
|
||
|
|
request.handler = json_handler
|
||
|
|
|