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 json
  14
  15
  16 # handles serialization of datetime in json
  17 DateEncoder = lambda obj: obj.strftime("%B %d, %Y %H:%M:%S") if isinstance(obj, datetime.datetime) else None
  18
  19 # handles decimal numbers serialization in json
  20 class DecimalEncoder(json.JSONEncoder):
  21     def _iterencode(self, o, markers=None):
  22         if isinstance(o, decimal.Decimal):
  23             return (str(o) for o in [o])
  24         return super(DecimalEncoder, self)._iterencode(o, markers)
  25
  26 def dispatch(request, object_type, object_name):
  27
  28     switch = {
  29          'platform' : platform,
  30          'slice' : slice,
  31          'resource' : resource,
  32          'user' : user
  33     }
  34
  35     # platform is local
  36     if ((object_type == 'platform') or (object_type == 'testbed')) :
  37         object_type = 'local:platform'
  38         object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
  39         return switch.get('platform', error)(request, object_name, object_properties)
  40     else :
  41         query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
  42         results = execute_query(request, query)
  43         if results :
  44             object_properties = []
  45             for r in results[0]['column'] :
  46                 object_properties.append(r['name'])
  47         else :
  48             return error(request, object_name, {})
  49         return switch.get(object_type, error)(request, object_name, object_properties)
  50
  51 #     if request.method == 'GET':
  52 #         return switch.get(request, object_type, object_name, object_properties)
  53 #     elif request.method == 'POST':
  54 #         return post(request, object_type, object_name)
  55
  56 def platform(request, object_name, object_properties):
  57     query  = Query().get('local:platform').filter_by('disabled', '==', '0').select(object_properties)
  58     return send(request, execute_query(request, query), object_properties)
  59
  60 def slice(request, object_name, object_properties):
  61     query = Query().get('slice').filter_by('user.user_hrn', '==', '$user_hrn').select(object_properties)
  62     return send(request, execute_query(request, query), object_properties)
  63
  64 def resource(request, object_name, object_properties):
  65     pass
  66
  67 def user(request, object_name, object_properties):
  68     query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn').select(object_properties)
  69     return send(request, execute_query(request, query), object_properties)
  70
  71 def send(request, response, object_properties):
  72     if request.path.split('/')[1] == 'rest' :
  73         response_data = response
  74         return HttpResponse(json.dumps(response_data), content_type="application/json")
  75     elif request.path.split('/')[1] == 'table' :
  76         return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
  77     elif request.path.split('/')[1] == 'datatable' :
  78         response_data = {}
  79         response_data['columns'] = object_properties
  80         response_data['labels'] = object_properties
  81         #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ]
  82         response_data['data'] = []
  83         response_data['total'] = len(response)
  84         for r in response :
  85             d = []
  86             for p in object_properties :
  87                 d.append(r[p])
  88             print d
  89
  90             response_data['data'].append(d)
  91
  92         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
  93
  94 def error(request, object_name, object_properties):
  95     return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")