TcloudServer/library/api/render.py at master · guyuecode/TcloudServer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import json

from flask import make_response, jsonify

from public_config import MSG_MAP


def _render(resp):
    response = make_response(jsonify(resp))
    #    response.headers["Access-Control-Allow-Origin"] = "*"
    return response


def json_render(response):
    if 'code' not in response.keys():
        response['code'] = 0
    if 'message' not in response.keys():
        response['message'] = MSG_MAP.get(response['code'], '')
    return _render(response)


def json_list_render(code, data, limit, offset, message=None):
    if message is None:
        message = MSG_MAP.get(code)
    resp = dict(
        code=code, limit=limit, offset=offset, message=message, data=data)
    return _render(resp)


def json_list_render2(code, data, page_size, page_index, total, message=None):
    if message is None:
        message = MSG_MAP.get(code)
    resp = dict(
        code=code, page_size=page_size, page_index=page_index, total=total, message=message, data=data)
    return _render(resp)


def json_detail_render(code, data=None, message=None, total=None):
    if data is None:
        data = []
    if message is None:
        message = MSG_MAP.get(code)
    resp = dict(code=code, message=message, data=data)
    if total:
        resp['total'] = total
    return _render(resp)


def json_token_render(code, token, message=None):
    if message is None:
        message = MSG_MAP.get(code)
    resp = dict(code=code, token=token, message=message)
    return _render(resp)


def json_detail_render_sse(code, data=None, message=None):
    if data is None:
        data = []
    if message is None:
        message = MSG_MAP.get(code)
    resp = dict(code=code, message=message, data=data)
    return json.dumps(resp)


def row2list(rows, keys):
    data = []
    for i in rows:
        data.append({key: getattr(i, key) for key in keys})
    return data