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