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