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