REST get interface reworked, added filter operators
[unfold.git] / rest / __init__.py
index 892e4ca..1ce245b 100644 (file)
 from manifold.core.query            import Query
 
 from django.views.generic.base      import TemplateView
+from django.shortcuts               import render_to_response
 
 from unfold.loginrequired           import LoginRequiredView
 from django.http                    import HttpResponse
 
 from manifold.core.query            import Query, AnalyzedQuery
-from manifold.manifoldapi           import execute_query
+from manifoldapi.manifoldapi        import execute_query
 
 from string import join
+import decimal
+import datetime
 import json
 
+# handles serialization of datetime in json
+DateEncoder = lambda obj: obj.strftime("%B %d, %Y %H:%M:%S") if isinstance(obj, datetime.datetime) else None
 
-def dispatch(request, object_type, object_name):
-
-    switch = {
-         'platform' : platform,
-         'slice' : slice,
-         'user' : user
-    }
-
-    # platform is local
-    if (object_type == 'platform') :
-        object_type = 'local:platform'
-        object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
-    else :
-        query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
-        results = execute_query(request, query)
-        if results :
-            object_properties = []
-            for r in results[0]['column'] :
-                object_properties.append(r['name'])
-        else :
-            return error()
+# support converting decimal in json
+json.encoder.FLOAT_REPR = lambda o: format(o, '.2f')
 
-    return switch.get(object_type, error)(request, object_type, object_name, object_properties)
+# handles decimal numbers serialization in json
+class DecimalEncoder(json.JSONEncoder):
+    def _iterencode(self, o, markers=None):
+        if isinstance(o, decimal.Decimal):
+            return (str(o) for o in [o])
+        return super(DecimalEncoder, self)._iterencode(o, markers)
 
-#     if request.method == 'GET':
-#         return switch.get(request, object_type, object_name, object_properties)
-#     elif request.method == 'POST':
-#         return post(request, object_type, object_name)
-
-def platform():
-    return HttpResponse(json.dumps({'user' : 'error message'}), content_type="application/json")
-
-def slice():
-    return HttpResponse(json.dumps({'user' : 'error message'}), content_type="application/json")
-
-def user(request, object_type, object_name, object_properties):
-    query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn').select(object_properties)
-    #.select('slice.slice_hrn')
+class objectRequest(object):
+    
+    def __init__(self, request, object_type, object_name):
+        self.type = object_type
+        self.name = object_name
+        self.properties = []
+        self.filters = {}
+        self.options = None
         
-    return send(request, execute_query(request, query), object_properties)
+        self.request = request
+
+        if ((self.type == 'platform') or (self.type == 'testbed')) :
+            self.type = 'local:platform'
+            self.id = 'platform'
+            self.properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
+            self.filters['disabled'] = '0'
+            self.filters['gateway_type'] = 'sfa'
+            self.filters['platform'] = '!myslice'
+        else :
+            self.id = 'hrn'
+            query = Query.get('local:object').filter_by('table', '==', self.type).select('column.name')
+            results = execute_query(self.request, query)
+            if results :
+                for r in results[0]['column'] :
+                    self.properties.append(r['name'])
+            else :
+                return error('db error')
+        return None
+    
+    def addFilters(self, properties):
+        selected_properties = []
+        for p in properties :
+            if p in self.properties :
+                selected_properties.append(p)
+        self.properties = selected_properties
+        self.setId()
+    
+    def setId(self):
+        if self.id in self.properties :
+            self.properties.remove(self.id)
+            [self.id].extend(self.properties)
+    
+    def execute(self):
+        query = Query.get(self.type).select(self.properties)
+        if self.filters :
+            for k, f in self.filters.iteritems() :
+                if (f[:1] == "!") :
+                    query.filter_by(k, '!=', f[1:])
+                elif (f[:2] == ">=") :
+                    query.filter_by(k, '>=', f[2:])
+                elif (f[:1] == ">") :
+                    query.filter_by(k, '>', f[1:])
+                elif (f[:2] == "<=") :
+                    query.filter_by(k, '<=', f[2:])
+                elif (f[:1] == "<") :
+                    query.filter_by(k, '<', f[1:])
+                else :
+                    query.filter_by(k, '==', f)
+        return execute_query(self.request, query)
 
-def send(request, response, object_properties):
+def dispatch(request, object_type, object_name):
+    
+    o = objectRequest(request, object_type, object_name)
+    
+    if request.method == 'POST':
+        req_items = request.POST
+    elif request.method == 'GET':
+        req_items = request.GET
+    
+    print req_items
+    
+    for el in req_items.items():
+        if el[0].startswith('filters'):
+            o.filters[el[0][8:-1]] = el[1]
+        elif el[0].startswith('columns'):
+            o.addFilters(req_items.getlist('columns[]'))
+        elif el[0].startswith('options'):
+            o.options = req_items.getlist('options[]')
+
+    response = o.execute()
+    
     if request.path.split('/')[1] == 'rest' :
         response_data = response
-    else :
+        print response
+        return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
+    elif request.path.split('/')[1] == 'table' :
+        print o.properties
+        print o.options
+        print response
+        return render_to_response('table-default.html', {'data' : response, 'properties' : o.properties, 'id' : o.id, 'options' : o.options})
+    elif request.path.split('/')[1] == 'datatable' :
         response_data = {}
-        response_data['columns'] = object_properties
-        response_data['labels'] = object_properties
-        #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ]
+        response_data['columns'] = o.properties
+        response_data['labels'] = o.properties
         response_data['data'] = []
         response_data['total'] = len(response)
         for r in response :
             d = []
-            for p in object_properties :
+            for p in o.properties :
                 d.append(r[p])
-            print d
+            #print d
             
             response_data['data'].append(d)
-            #response_data['data'].append([ r['platform'], r['platform_longname'], r['platform_url'], r['platform_description'], r['gateway_type'] ])
-         
-    return HttpResponse(json.dumps(response_data), content_type="application/json")
-
-
-
-
-
-def get(request, object_type, object_name, object_properties):
-        
-    query = Query().get(object_type).select(object_properties).filter_by('user_hrn', '=', '$user_hrn')
-    if (object_name) :
-        query = query.filter_by(object_type + '_hrn', '=', object_name)
-    
-    response = execute_query(request, query)
-    
-    response_data = {}
-    response_data['columns'] = object_properties
-    response_data['labels'] = object_properties
-    #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ]
-    response_data['data'] = []
-    response_data['total'] = len(response)
-    for r in response :
-        d = []
-        for p in object_properties :
-            d.append(r[p])
-        print d
-        
-        response_data['data'].append(d)
-        #response_data['data'].append([ r['platform'], r['platform_longname'], r['platform_url'], r['platform_description'], r['gateway_type'] ])
          
-    return HttpResponse(json.dumps(response_data), content_type="application/json")
-
-def post(request, object_type, object_name):
-    pass
+        return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
 
-def error():
-    return HttpResponse(json.dumps({'error' : 'error message'}), content_type="application/json")
\ No newline at end of file
+def error(msg):
+    return HttpResponse(json.dumps({'error' : msg}), content_type="application/json")