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
36 'platform' : platform,
38 'resource' : resource,
42 if request.method == 'POST':
43 req_items = request.POST.items()
44 elif request.method == 'GET':
45 req_items = request.GET.items()
48 if el[0].startswith('filters'):
49 object_filters[el[0][8:-1]] = el[1]
50 elif el[0].startswith('columns'):
51 object_properties = request.POST.getlist('columns[]')
54 if ((object_type == 'platform') or (object_type == 'testbed')) :
55 object_type = 'local:platform'
56 if object_properties == None :
57 object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
58 return switch.get('platform', error)(request, object_name, object_properties, object_filters)
60 if object_properties == None :
61 query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
62 results = execute_query(request, query)
64 object_properties = []
65 for r in results[0]['column'] :
66 object_properties.append(r['name'])
68 return error(request, object_name, {})
69 return switch.get(object_type, error)(request, object_name, object_properties, object_filters)
71 def platform(request, object_name, object_properties, object_filters = None):
72 query = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type', '==', 'sfa').filter_by('platform', '!=', 'myslice')
74 for k, f in object_filters.iteritems() :
75 query.filter_by(k, '==', f)
76 query.select(object_properties)
77 return send(request, execute_query(request, query), object_properties)
79 # Add different filters possibilities [['user.user_hrn','==','$user_hrn'],['parent_authority','==','ple.upmc']]
80 def slice(request, object_name, object_properties, object_filters = None):
81 query = Query().get('slice')#.filter_by('user.user_hrn', '==', '$user_hrn')
83 for k, f in object_filters.iteritems() :
84 query.filter_by(k, '==', f)
85 query.select(object_properties)
86 return send(request, execute_query(request, query), object_properties)
88 def resource(request, object_name, object_properties, object_filters = None):
89 query = Query().get('resource')
91 for k, f in object_filters.iteritems() :
92 query.filter_by(k, '==', f)
93 query.select(object_properties)
94 return send(request, execute_query(request, query), object_properties)
96 def user(request, object_name, object_properties, object_filters = None):
97 query = Query().get('user')#.filter_by('user_hrn', '==', '$user_hrn')
99 for k, f in object_filters.iteritems() :
100 query.filter_by(k, '==', f)
101 query.select(object_properties)
102 return send(request, execute_query(request, query), object_properties)
104 def send(request, response, object_properties):
105 if request.path.split('/')[1] == 'rest' :
106 response_data = response
107 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
108 elif request.path.split('/')[1] == 'table' :
109 return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
110 elif request.path.split('/')[1] == 'datatable' :
112 response_data['columns'] = object_properties
113 response_data['labels'] = object_properties
114 response_data['data'] = []
115 response_data['total'] = len(response)
118 for p in object_properties :
122 response_data['data'].append(d)
124 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
126 def error(request, object_name, object_properties):
127 return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")