blob: d6fd9d7bdcefba8a9f9061c1456afed3ded8a942 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
"""Functions and classes that deal with responses and conversion to JSON."""
import json
from flask import make_response
class OutputEncoder(json.JSONEncoder):
"""
Class to encode output into JSON, for objects which the default
json.JSONEncoder class does not have default encoding for.
"""
def default(self, o):
if isinstance(o, bytes):
return str(o, encoding="utf-8")
return json.JSONEncoder.default(self, o)
def build_response(data):
"""Build the responses for the API"""
status_codes = {
"error": 400, "not-found": 404, "success": 200, "exception": 500}
response = make_response(
json.dumps(data, cls=OutputEncoder),
status_codes[data["status"]])
response.headers["Content-Type"] = "application/json"
return response
|
