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
18 # handles serialization of datetime in json
19 DateEncoder = lambda obj: obj.strftime("%B %d, %Y %H:%M:%S") if isinstance(obj, datetime.datetime) else None
21 # support converting decimal in json
22 json.encoder.FLOAT_REPR = lambda o: format(o, '.2f')
24 # handles decimal numbers serialization in json
25 class DecimalEncoder(json.JSONEncoder):
26 def _iterencode(self, o, markers=None):
27 if isinstance(o, decimal.Decimal):
28 return (str(o) for o in [o])
29 return super(DecimalEncoder, self)._iterencode(o, markers)
31 def dispatch(request, object_type, object_name):
33 object_properties = []
37 'platform' : platform,
39 'resource' : resource,
43 if request.method == 'POST':
44 req_items = request.POST.items()
45 elif request.method == 'GET':
46 req_items = request.GET.items()
49 if el[0].startswith('filters'):
50 object_filters[el[0][8:-1]] = el[1]
51 elif el[0].startswith('columns'):
52 object_properties = request.POST.getlist('columns[]')
56 if ((object_type == 'platform') or (object_type == 'testbed')) :
57 object_type = 'local:platform'
58 if object_properties == None :
59 object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
60 return switch.get('platform', error)(request, object_name, object_properties, object_filters)
62 if object_properties == None :
63 query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
64 results = execute_query(request, query)
66 object_properties = []
67 for r in results[0]['column'] :
68 object_properties.append(r['name'])
70 return error(request, object_name, {})
71 return switch.get(object_type, error)(request, object_name, object_properties, object_filters)
73 def platform(request, object_name, object_properties, object_filters = None):
74 query = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type', '==', 'sfa').filter_by('platform', '!=', 'myslice')
76 for k, f in object_filters.iteritems() :
77 query.filter_by(k, '==', f)
78 query.select(object_properties)
79 return send(request, execute_query(request, query), object_properties)
81 # Add different filters possibilities [['user.user_hrn','==','$user_hrn'],['parent_authority','==','ple.upmc']]
82 def slice(request, object_name, object_properties, object_filters = None):
83 query = Query().get('slice').filter_by('user.user_hrn', '==', '$user_hrn')
85 for k, f in object_filters.iteritems() :
86 query.filter_by(k, '==', f)
87 query.select(object_properties)
88 return send(request, execute_query(request, query), object_properties)
90 def resource(request, object_name, object_properties, object_filters = None):
91 query = Query().get('resource')
93 for k, f in object_filters.iteritems() :
94 query.filter_by(k, '==', f)
95 query.select(object_properties)
96 return send(request, execute_query(request, query), object_properties)
98 def user(request, object_name, object_properties, object_filters = None):
99 # query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn').select(object_properties)
100 query = Query().get('user').filter_by('parent_authority', '==', 'ple.upmc').select(object_properties)
101 #query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn')
103 for k, f in object_filters.iteritems() :
104 query.filter_by(k, '==', f)
105 query.select(object_properties)
106 return send(request, execute_query(request, query), object_properties)
108 def send(request, response, object_properties):
109 if request.path.split('/')[1] == 'rest' :
110 response_data = response
111 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
112 elif request.path.split('/')[1] == 'table' :
113 return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
114 elif request.path.split('/')[1] == 'datatable' :
116 response_data['columns'] = object_properties
117 response_data['labels'] = object_properties
118 response_data['data'] = []
119 response_data['total'] = len(response)
122 for p in object_properties :
126 response_data['data'].append(d)
128 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
130 def error(request, object_name, object_properties):
131 return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")