X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=rest%2F__init__.py;h=f87196634c5125be96a7a7882650a55da840b3be;hb=4e8925bca017d05c0325edd7f9146b83835267db;hp=892e4ca5a9488968badd7ddfb7903caadef7bbed;hpb=574ab14b3b3daf0f407dcc2a73a54ad5f87af372;p=myslice.git diff --git a/rest/__init__.py b/rest/__init__.py index 892e4ca5..f8719663 100644 --- a/rest/__init__.py +++ b/rest/__init__.py @@ -1,66 +1,125 @@ from manifold.core.query import Query from django.views.generic.base import TemplateView +from django.shortcuts import render_to_response from unfold.loginrequired import LoginRequiredView from django.http import HttpResponse from manifold.core.query import Query, AnalyzedQuery -from manifold.manifoldapi import execute_query +from manifoldapi.manifoldapi import execute_query from string import join +import decimal +import datetime import json +# handles serialization of datetime in json +DateEncoder = lambda obj: obj.strftime("%B %d, %Y %H:%M:%S") if isinstance(obj, datetime.datetime) else None -def dispatch(request, object_type, object_name): +# support converting decimal in json +json.encoder.FLOAT_REPR = lambda o: format(o, '.2f') + +# handles decimal numbers serialization in json +class DecimalEncoder(json.JSONEncoder): + def _iterencode(self, o, markers=None): + if isinstance(o, decimal.Decimal): + return (str(o) for o in [o]) + return super(DecimalEncoder, self)._iterencode(o, markers) +def dispatch(request, object_type, object_name): + + object_properties = None + object_filters = {} + switch = { 'platform' : platform, 'slice' : slice, - 'user' : user + 'resource' : resource, + 'user' : user, + 'authority' : authority, } + + if request.method == 'POST': + req_items = request.POST.items() + elif request.method == 'GET': + req_items = request.GET.items() + + for el in req_items: + if el[0].startswith('filters'): + object_filters[el[0][8:-1]] = el[1] + elif el[0].startswith('columns'): + object_properties = request.POST.getlist('columns[]') # platform is local - if (object_type == 'platform') : + if ((object_type == 'platform') or (object_type == 'testbed')) : object_type = 'local:platform' - object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type']; + if object_properties == None : + object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type']; + return switch.get('platform', error)(request, object_name, object_properties, object_filters) else : - query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name') - results = execute_query(request, query) - if results : - object_properties = [] - for r in results[0]['column'] : - object_properties.append(r['name']) - else : - return error() - - return switch.get(object_type, error)(request, object_type, object_name, object_properties) - -# if request.method == 'GET': -# return switch.get(request, object_type, object_name, object_properties) -# elif request.method == 'POST': -# return post(request, object_type, object_name) - -def platform(): - return HttpResponse(json.dumps({'user' : 'error message'}), content_type="application/json") - -def slice(): - return HttpResponse(json.dumps({'user' : 'error message'}), content_type="application/json") - -def user(request, object_type, object_name, object_properties): - query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn').select(object_properties) - #.select('slice.slice_hrn') - + if object_properties == None : + query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name') + results = execute_query(request, query) + if results : + object_properties = [] + for r in results[0]['column'] : + object_properties.append(r['name']) + else : + return error(request, object_name, {}) + return switch.get(object_type, error)(request, object_name, object_properties, object_filters) + +def platform(request, object_name, object_properties, object_filters = None): + query = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type', '==', 'sfa').filter_by('platform', '!=', 'myslice') + if object_filters : + for k, f in object_filters.iteritems() : + query.filter_by(k, '==', f) + query.select(object_properties) + return send(request, execute_query(request, query), object_properties) + +# Add different filters possibilities [['user.user_hrn','==','$user_hrn'],['parent_authority','==','ple.upmc']] +def slice(request, object_name, object_properties, object_filters = None): + query = Query().get('slice')#.filter_by('user.user_hrn', '==', '$user_hrn') + if object_filters : + for k, f in object_filters.iteritems() : + query.filter_by(k, '==', f) + query.select(object_properties) + return send(request, execute_query(request, query), object_properties) + +def resource(request, object_name, object_properties, object_filters = None): + query = Query().get('resource') + if object_filters : + for k, f in object_filters.iteritems() : + query.filter_by(k, '==', f) + query.select(object_properties) + return send(request, execute_query(request, query), object_properties) + +def user(request, object_name, object_properties, object_filters = None): + query = Query().get('user')#.filter_by('user_hrn', '==', '$user_hrn') + if object_filters : + for k, f in object_filters.iteritems() : + query.filter_by(k, '==', f) + query.select(object_properties) + return send(request, execute_query(request, query), object_properties) + +def authority(request, object_name, object_properties, object_filters = None): + query = Query().get('authority')#.filter_by('user_hrn', '==', '$user_hrn') + if object_filters : + for k, f in object_filters.iteritems() : + query.filter_by(k, '==', f) + query.select(object_properties) return send(request, execute_query(request, query), object_properties) def send(request, response, object_properties): if request.path.split('/')[1] == 'rest' : response_data = response - else : + return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json") + elif request.path.split('/')[1] == 'table' : + return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties}) + elif request.path.split('/')[1] == 'datatable' : response_data = {} response_data['columns'] = object_properties response_data['labels'] = object_properties - #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ] response_data['data'] = [] response_data['total'] = len(response) for r in response : @@ -70,41 +129,8 @@ def send(request, response, object_properties): print d response_data['data'].append(d) - #response_data['data'].append([ r['platform'], r['platform_longname'], r['platform_url'], r['platform_description'], r['gateway_type'] ]) - - return HttpResponse(json.dumps(response_data), content_type="application/json") - - - - - -def get(request, object_type, object_name, object_properties): - - query = Query().get(object_type).select(object_properties).filter_by('user_hrn', '=', '$user_hrn') - if (object_name) : - query = query.filter_by(object_type + '_hrn', '=', object_name) - - response = execute_query(request, query) - - response_data = {} - response_data['columns'] = object_properties - response_data['labels'] = object_properties - #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ] - response_data['data'] = [] - response_data['total'] = len(response) - for r in response : - d = [] - for p in object_properties : - d.append(r[p]) - print d - - response_data['data'].append(d) - #response_data['data'].append([ r['platform'], r['platform_longname'], r['platform_url'], r['platform_description'], r['gateway_type'] ]) - return HttpResponse(json.dumps(response_data), content_type="application/json") - -def post(request, object_type, object_name): - pass + return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json") -def error(): - return HttpResponse(json.dumps({'error' : 'error message'}), content_type="application/json") \ No newline at end of file +def error(request, object_name, object_properties): + return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")