rest/__init__.py

   1 from manifold.core.query            import Query
   2
   3 from django.views.generic.base      import TemplateView
   4 from django.shortcuts               import render_to_response
   5
   6 from unfold.loginrequired           import LoginRequiredView
   7 from django.http                    import HttpResponse
   8
   9 from manifold.core.query            import Query, AnalyzedQuery
  10 from manifoldapi.manifoldapi        import execute_query
  11
  12 from string import join
  13 import decimal
  14 import datetime
  15 import json
  16
  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
  19
  20 # support converting decimal in json
  21 json.encoder.FLOAT_REPR = lambda o: format(o, '.2f')
  22
  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)
  29
  30 def dispatch(request, object_type, object_name):
  31
  32     object_properties = None
  33     object_filters = {}
  34
  35     # XXX TODO: What about the local: objects?
  36     # Example: local:user (Manifold) is different from user (SFA GW)
  37     switch = {
  38          'platform' : platform,
  39          'slice' : slice,
  40          'resource' : resource,
  41          'user' : user,
  42          'authority' : authority,
  43     }
  44
  45     if request.method == 'POST':
  46         req_items = request.POST.items()
  47     elif request.method == 'GET':
  48         req_items = request.GET.items()
  49
  50     for el in req_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[]')
  55
  56     # platform is local
  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)
  62     else :
  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)
  66             if results :
  67                 object_properties = []
  68                 for r in results[0]['column'] :
  69                     object_properties.append(r['name'])
  70             else :
  71                 return error(request, object_name, {})
  72         return switch.get(object_type, error)(request, object_name, object_properties, object_filters)
  73
  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')
  76     if object_filters :
  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)
  81
  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')
  85     if object_filters :
  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)
  90
  91 def resource(request, object_name, object_properties, object_filters = None):
  92     query = Query().get('resource')
  93     if object_filters :
  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)
  98
  99 def user(request, object_name, object_properties, object_filters = None):
 100     query = Query().get('user')#.filter_by('user_hrn', '==', '$user_hrn')
 101     if object_filters :
 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)
 106
 107 def authority(request, object_name, object_properties, object_filters = None):
 108     query = Query().get('authority')#.filter_by('user_hrn', '==', '$user_hrn')
 109     if object_filters :
 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)
 114
 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' :
 122         response_data = {}
 123         response_data['columns'] = object_properties
 124         response_data['labels'] = object_properties
 125         response_data['data'] = []
 126         response_data['total'] = len(response)
 127         for r in response :
 128             d = []
 129             for p in object_properties :
 130                 d.append(r[p])
 131             print d
 132
 133             response_data['data'].append(d)
 134
 135         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
 136
 137 def error(request, object_name, object_properties):
 138     return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")