blob: 30f07635827810007fd75f50b6b18c38c2508694 (
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
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
"""
Rhizi REST web API:
- make use of rz_kernel for core logic execution
- make use of rz_api_common for common API logic
"""
from flask import Flask
from flask import escape
from flask import jsonify
from flask import make_response
from flask import redirect
from flask import render_template
from flask import request
from flask import send_from_directory
from flask import session
from flask import url_for
import flask
import logging
import traceback
from model.graph import Attr_Diff
from model.graph import Topo_Diff
from rz_api_common import sanitize_input__attr_diff
from rz_api_common import sanitize_input__topo_diff
from rz_api_common import validate_obj__attr_diff
log = logging.getLogger('rhizi')
db_ctl = None # injected: DB controller
def __response_wrap(data=None, error=None):
"""
wrap response data/errors as dict - this should always be used when returning
data to allow easy return of list objects, assist in error case distinction, etc.
"""
return dict(data=data, error=error)
def __common_resp_handle(data=None, error=None):
"""
provide common response handling
@data must be json serializable
@error will be serialized with str()
"""
if error is None:
error_str = ""
else:
error_str = str(error) # convert any Exception objects to serializable form
ret_data = __response_wrap(data, error_str)
resp = jsonify(ret_data) # this will create a Flask Response object
resp.headers['Access-Control-Allow-Origin'] = '*'
# more response processing
return resp
def diff_commit__topo():
"""
REST API wrapper around diff_commit__topo():
- extract topo_diff from request
- handle success/error outcomes
"""
def sanitize_input(req):
topo_diff_dict = request.get_json()['topo_diff']
topo_diff = Topo_Diff.from_json_dict(topo_diff_dict)
sanitize_input__topo_diff(topo_diff)
return topo_diff;
try:
topo_diff = sanitize_input(request)
except Exception as e:
return __common_resp_handle(error='malformed input')
try:
kernel = flask.current_app.kernel
topo_diff = kernel.diff_commit__topo(topo_diff)
topo_diff_json = topo_diff.to_json_dict()
return __common_resp_handle(data=topo_diff_json)
except Exception as e:
log.error(e.message)
log.error(traceback.print_exc())
return __common_resp_handle(error=e)
def diff_commit__attr():
"""
commit a graph attribute diff
"""
def sanitize_input(req):
attr_diff_dict = request.get_json()['attr_diff']
attr_diff = Attr_Diff.from_json_dict(attr_diff_dict)
sanitize_input__attr_diff(attr_diff)
return attr_diff;
def on_error(e):
# handle DB ERRORS, eg. name attr change error
return __common_resp_handle(error='error occurred')
try:
attr_diff = sanitize_input(request)
validate_obj__attr_diff(attr_diff)
except Exception as e:
return __common_resp_handle(error='malformed input')
try:
kernel = flask.current_app.kernel
attr_diff = kernel.diff_commit__attr(attr_diff)
return __common_resp_handle(data=attr_diff)
except Exception as e:
log.error(e.message)
log.error(traceback.print_exc())
return __common_resp_handle(error=e)
def diff_commit__vis():
pass
|
