1 from manifold.core.query import Query
3 from django.views.generic.base import TemplateView
4 from django.shortcuts import render_to_response
6 from unfold.loginrequired import LoginRequiredView
7 from django.http import HttpResponse
9 from manifold.core.query import Query, AnalyzedQuery
10 from manifoldapi.manifoldapi import execute_query
12 from string import join
17 # handles serialization of datetime in json
18 DateEncoder = lambda obj: obj.strftime("%B %d, %Y %H:%M:%S") if isinstance(obj, datetime.datetime) else None
20 # support converting decimal in json
21 json.encoder.FLOAT_REPR = lambda o: format(o, '.2f')
23 # handles decimal numbers serialization in json
24 class DecimalEncoder(json.JSONEncoder):
25 def _iterencode(self, o, markers=None):
26 if isinstance(o, decimal.Decimal):
27 return (str(o) for o in [o])
28 return super(DecimalEncoder, self)._iterencode(o, markers)
30 def dispatch(request, object_type, object_name):
32 object_properties = None
35 # XXX TODO: What about the local: objects?
36 # Example: local:user (Manifold) is different from user (SFA GW)
38 'platform' : platform,
40 'resource' : resource,
42 'authority' : authority,
45 if request.method == 'POST':
46 req_items = request.POST.items()
47 elif request.method == 'GET':
48 req_items = request.GET.items()
51 if el[0].startswith('filters'):
52 object_filters[el[0][8:-1]] = el[1]
53 elif el[0].startswith('columns'):
54 object_properties = request.POST.getlist('columns[]')
57 if ((object_type == 'platform') or (object_type == 'testbed')) :
58 object_type = 'local:platform'
59 if object_properties == None :
60 object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
61 return switch.get('platform', error)(request, object_name, object_properties, object_filters)
63 if object_properties == None :
64 query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
65 results = execute_query(request, query)
67 object_properties = []
68 for r in results[0]['column'] :
69 object_properties.append(r['name'])
71 return error(request, object_name, {})
72 return switch.get(object_type, error)(request, object_name, object_properties, object_filters)
74 def platform(request, object_name, object_properties, object_filters = None):
75 query = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type', '==', 'sfa').filter_by('platform', '!=', 'myslice')
77 for k, f in object_filters.iteritems() :
78 query.filter_by(k, '==', f)
79 query.select(object_properties)
80 return send(request, execute_query(request, query), object_properties)
82 # Add different filters possibilities [['user.user_hrn','==','$user_hrn'],['parent_authority','==','ple.upmc']]
83 def slice(request, object_name, object_properties, object_filters = None):
84 query = Query().get('slice')#.filter_by('user.user_hrn', '==', '$user_hrn')
86 for k, f in object_filters.iteritems() :
87 query.filter_by(k, '==', f)
88 query.select(object_properties)
89 return send(request, execute_query(request, query), object_properties)
91 def resource(request, object_name, object_properties, object_filters = None):
92 query = Query().get('resource')
94 for k, f in object_filters.iteritems() :
95 query.filter_by(k, '==', f)
96 query.select(object_properties)
97 return send(request, execute_query(request, query), object_properties)
99 def user(request, object_name, object_properties, object_filters = None):
100 query = Query().get('user')#.filter_by('user_hrn', '==', '$user_hrn')
102 for k, f in object_filters.iteritems() :
103 query.filter_by(k, '==', f)
104 query.select(object_properties)
105 return send(request, execute_query(request, query), object_properties)
107 def authority(request, object_name, object_properties, object_filters = None):
108 query = Query().get('authority')#.filter_by('user_hrn', '==', '$user_hrn')
110 for k, f in object_filters.iteritems() :
111 query.filter_by(k, '==', f)
112 query.select(object_properties)
113 return send(request, execute_query(request, query), object_properties)
115 def send(request, response, object_properties):
116 if request.path.split('/')[1] == 'rest' :
117 response_data = response
118 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
119 elif request.path.split('/')[1] == 'table' :
120 return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
121 elif request.path.split('/')[1] == 'datatable' :
123 response_data['columns'] = object_properties
124 response_data['labels'] = object_properties
125 response_data['data'] = []
126 response_data['total'] = len(response)
129 for p in object_properties :
133 response_data['data'].append(d)
135 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
137 def error(request, object_name, object_properties):
138 return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")