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
  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
  20
  21 # support converting decimal in json
  22 encoder.FLOAT_REPR = lambda o: format(o, '.2f')
  23
  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)
  30
  31 def dispatch(request, object_type, object_name):
  32
  33     switch = {
  34          'platform' : platform,
  35          'slice' : slice,
  36          'resource' : resource,
  37          'user' : user
  38     }
  39
  40     # platform is local
  41     if ((object_type == 'platform') or (object_type == 'testbed')) :
  42         object_type = 'local:platform'
  43         object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
  44         return switch.get('platform', error)(request, object_name, object_properties)
  45     else :
  46         query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
  47         results = execute_query(request, query)
  48         if results :
  49             object_properties = []
  50             for r in results[0]['column'] :
  51                 object_properties.append(r['name'])
  52         else :
  53             return error(request, object_name, {})
  54         return switch.get(object_type, error)(request, object_name, object_properties)
  55
  56 #     if request.method == 'GET':
  57 #         return switch.get(request, object_type, object_name, object_properties)
  58 #     elif request.method == 'POST':
  59 #         return post(request, object_type, object_name)
  60
  61 def platform(request, object_name, object_properties):
  62     query  = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type', '==', 'sfa').filter_by('platform', '!=', 'myslice').select(object_properties)
  63     return send(request, execute_query(request, query), object_properties)
  64
  65 def slice(request, object_name, object_properties):
  66     query = Query().get('slice').filter_by('user.user_hrn', '==', '$user_hrn').select(object_properties)
  67     return send(request, execute_query(request, query), object_properties)
  68
  69 def resource(request, object_name, object_properties):
  70     pass
  71
  72 def user(request, object_name, object_properties):
  73     query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn').select(object_properties)
  74     return send(request, execute_query(request, query), object_properties)
  75
  76 def send(request, response, object_properties):
  77     if request.path.split('/')[1] == 'rest' :
  78         response_data = response
  79         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
  80     elif request.path.split('/')[1] == 'table' :
  81         return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
  82     elif request.path.split('/')[1] == 'datatable' :
  83         response_data = {}
  84         response_data['columns'] = object_properties
  85         response_data['labels'] = object_properties
  86         #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ]
  87         response_data['data'] = []
  88         response_data['total'] = len(response)
  89         for r in response :
  90             d = []
  91             for p in object_properties :
  92                 d.append(r[p])
  93             print d
  94
  95             response_data['data'].append(d)
  96
  97         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
  98
  99 def error(request, object_name, object_properties):
 100     return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")