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
  10 from manifoldapi.manifoldapi        import execute_query
  11
  12 from string                         import join
  13
  14 import decimal
  15 import datetime
  16 import json
  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 json.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 class objectRequest(object):
  32
  33     def __init__(self, request, object_type, object_name):
  34         self.type = object_type
  35         self.name = object_name
  36         self.properties = []
  37         self.filters = {}
  38         self.options = None
  39
  40         self.request = request
  41
  42     # XXX TODO: What about the local: objects?
  43     # Example: local:user (Manifold) is different from user (SFA GW)
  44
  45         if ((self.type == 'platform') or (self.type == 'testbed')) :
  46             self.type = 'local:platform'
  47             self.id = 'platform'
  48             self.properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
  49             self.filters['disabled'] = '0'
  50             self.filters['gateway_type'] = 'sfa'
  51             self.filters['platform'] = '!myslice'
  52         else :
  53             self.id = 'hrn'
  54             query = Query.get('local:object').filter_by('table', '==', self.type).select('column.name')
  55             results = execute_query(self.request, query)
  56             if results :
  57                 for r in results[0]['column'] :
  58                     self.properties.append(r['name'])
  59             else :
  60                 return error('db error')
  61         return None
  62
  63     def addFilters(self, properties):
  64         selected_properties = []
  65         for p in properties :
  66             if p in self.properties :
  67                 selected_properties.append(p)
  68         self.properties = selected_properties
  69         self.setId()
  70
  71     def setId(self):
  72         if self.id in self.properties :
  73             self.properties.remove(self.id)
  74             [self.id].extend(self.properties)
  75
  76     def execute(self):
  77         query = Query.get(self.type).select(self.properties)
  78         if self.filters :
  79             for k, f in self.filters.iteritems() :
  80                 if (f[:1] == "!") :
  81                     query.filter_by(k, '!=', f[1:])
  82                 elif (f[:2] == ">=") :
  83                     query.filter_by(k, '>=', f[2:])
  84                 elif (f[:1] == ">") :
  85                     query.filter_by(k, '>', f[1:])
  86                 elif (f[:2] == "<=") :
  87                     query.filter_by(k, '<=', f[2:])
  88                 elif (f[:1] == "<") :
  89                     query.filter_by(k, '<', f[1:])
  90                 else :
  91                     query.filter_by(k, '==', f)
  92         return execute_query(self.request, query)
  93
  94 def dispatch(request, object_type, object_name):
  95
  96     o = objectRequest(request, object_type, object_name)
  97
  98     if request.method == 'POST':
  99         req_items = request.POST
 100     elif request.method == 'GET':
 101         req_items = request.GET
 102
 103     for el in req_items.items():
 104         if el[0].startswith('filters'):
 105             o.filters[el[0][8:-1]] = el[1]
 106         elif el[0].startswith('columns'):
 107             o.addFilters(req_items.getlist('columns[]'))
 108         elif el[0].startswith('options'):
 109             o.options = req_items.getlist('options[]')
 110
 111     response = o.execute()
 112
 113     if request.path.split('/')[1] == 'rest' :
 114         response_data = response
 115         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
 116     elif request.path.split('/')[1] == 'table' :
 117         return render_to_response('table-default.html', {'data' : response, 'properties' : o.properties, 'id' : o.id, 'options' : o.options})
 118     elif request.path.split('/')[1] == 'datatable' :
 119         response_data = {}
 120         response_data['columns'] = o.properties
 121         response_data['labels'] = o.properties
 122         response_data['data'] = []
 123         response_data['total'] = len(response)
 124         for r in response :
 125             d = []
 126             for p in o.properties :
 127                 d.append(r[p])
 128             response_data['data'].append(d)
 129
 130         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
 131
 132 def error(msg):
 133     return HttpResponse(json.dumps({'error' : msg}), content_type="application/json")