Merge branch 'onelab' of ssh://git.onelab.eu/git/myslice into onelab

[unfold.git] / rest / __init__.py

diff --git a/rest/__init__.py b/rest/__init__.py
index 71fe44d..5d1f58a 100644 (file)
--- a/rest/__init__.py
+++ b/rest/__init__.py
@@ -1,6 +1,7 @@
 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
@@ -12,18 +13,30 @@ from string import join
 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
+
+# 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)
+
 def dispatch(request, object_type, object_name):
 
     switch = {
          'platform' : platform,
          'slice' : slice,
+         'resource' : resource,
          'user' : user
     }
 
     # platform is local
-    if (object_type == 'platform') :
+    if ((object_type == 'platform') or (object_type == 'testbed')) :
         object_type = 'local:platform'
         object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
+        return switch.get('platform', error)(request, object_name, object_properties)
     else :
         query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
         results = execute_query(request, query)
@@ -32,31 +45,38 @@ def dispatch(request, object_type, object_name):
             for r in results[0]['column'] :
                 object_properties.append(r['name'])
         else :
-            return error()
-
-    return switch.get(object_type, error)(request, object_type, object_name, object_properties)
+            return error(request, object_name, {})
+        return switch.get(object_type, error)(request, object_name, object_properties)
 
 #     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 platform(request, object_name, object_properties):
+    query  = Query().get('local:platform').filter_by('disabled', '==', '0').select(object_properties)
+    return send(request, execute_query(request, query), object_properties)
 
-def slice():
-    return HttpResponse(json.dumps({'user' : 'error message'}), content_type="application/json")
+# Add different filters possibilities [['user.user_hrn','==','$user_hrn'],['parent_authority','==','ple.upmc']]
+def slice(request, object_name, object_properties):
+    #query = Query().get('slice').filter_by('user.user_hrn', '==', '$user_hrn').select(object_properties)
+    query = Query().get('slice').filter_by('parent_authority', '==', 'ple.upmc').select(object_properties)
+    return send(request, execute_query(request, query), object_properties)
 
-def user(request, object_type, object_name, object_properties):
+def resource(request, object_name, object_properties):
+    pass
+
+def user(request, object_name, object_properties):
     query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn').select(object_properties)
-    #.select('slice.slice_hrn')
-        
     return send(request, execute_query(request, query), object_properties)
 
 def send(request, response, object_properties):
     if request.path.split('/')[1] == 'rest' :
         response_data = response
-    else :
+        return HttpResponse(json.dumps(response_data), content_type="application/json")
+    elif request.path.split('/')[1] == 'table' :
+        return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
+    elif request.path.split('/')[1] == 'datatable' :
         response_data = {}
         response_data['columns'] = object_properties
         response_data['labels'] = object_properties
@@ -70,41 +90,8 @@ def send(request, response, object_properties):
             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")
+def error(request, object_name, object_properties):
+    return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")