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 import urlparse
  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 def dispatch(request, object_type, object_name):
  32
  33     object_properties = []
  34     object_filters = {}
  35
  36     switch = {
  37          'platform' : platform,
  38          'slice' : slice,
  39          'resource' : resource,
  40          'user' : user
  41     }
  42
  43     if request.method == 'POST':
  44         for el in request.POST.items():
  45             if el[0].startswith('filters'):
  46                 object_filters[el[0][8:-1]] = el[1]
  47             elif el[0].startswith('columns'):
  48                 object_properties = request.POST.getlist('columns[]')
  49
  50     # platform is local
  51     if ((object_type == 'platform') or (object_type == 'testbed')) :
  52         object_type = 'local:platform'
  53         if object_properties == None :
  54             object_properties = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
  55         return switch.get('platform', error)(request, object_name, object_properties, object_filters)
  56     else :
  57         if object_properties == None :
  58             query = Query.get('local:object').filter_by('table', '==', object_type).select('column.name')
  59             results = execute_query(request, query)
  60             if results :
  61                 object_properties = []
  62                 for r in results[0]['column'] :
  63                     object_properties.append(r['name'])
  64             else :
  65                 return error(request, object_name, {})
  66         return switch.get(object_type, error)(request, object_name, object_properties, object_filters)
  67
  68 def platform(request, object_name, object_properties, object_filters = None):
  69     query  = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type', '==', 'sfa').filter_by('platform', '!=', 'myslice')
  70     if object_filters :
  71         for k, f in object_filters.iteritems() :
  72             query.filter_by(k, '==', f)
  73     query.select(object_properties)
  74     return send(request, execute_query(request, query), object_properties)
  75
  76 def slice(request, object_name, object_properties, object_filters = None):
  77     query = Query().get('slice').filter_by('user.user_hrn', '==', '$user_hrn')
  78     if object_filters :
  79         for k, f in object_filters.iteritems() :
  80             query.filter_by(k, '==', f)
  81     query.select(object_properties)
  82     return send(request, execute_query(request, query), object_properties)
  83
  84 def resource(request, object_name, object_properties, object_filters = None):
  85     query = Query().get('resource')
  86     if object_filters :
  87         for k, f in object_filters.iteritems() :
  88             query.filter_by(k, '==', f)
  89     query.select(object_properties)
  90     return send(request, execute_query(request, query), object_properties)
  91
  92 def user(request, object_name, object_properties, object_filters = None):
  93     query = Query().get('user').filter_by('user_hrn', '==', '$user_hrn')
  94     if object_filters :
  95         for k, f in object_filters.iteritems() :
  96             query.filter_by(k, '==', f)
  97     query.select(object_properties)
  98     return send(request, execute_query(request, query), object_properties)
  99
 100 def send(request, response, object_properties):
 101     if request.path.split('/')[1] == 'rest' :
 102         response_data = response
 103         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
 104     elif request.path.split('/')[1] == 'table' :
 105         return render_to_response('table-default.html', {'data' : response, 'properties' : object_properties})
 106     elif request.path.split('/')[1] == 'datatable' :
 107         response_data = {}
 108         response_data['columns'] = object_properties
 109         response_data['labels'] = object_properties
 110         #response_data['labels'] = [ 'Platform', 'Name', 'Url', 'Description','Gateway Type' ]
 111         response_data['data'] = []
 112         response_data['total'] = len(response)
 113         for r in response :
 114             d = []
 115             for p in object_properties :
 116                 d.append(r[p])
 117             print d
 118
 119             response_data['data'].append(d)
 120
 121         return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
 122
 123 def error(request, object_name, object_properties):
 124     return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")
 125
 126 def getDictArray(post, name):
 127     dic = {}
 128     for k in post.keys():
 129         if k.startswith(name):
 130             rest = k[len(name):]
 131
 132             # split the string into different components
 133             parts = [p[:-1] for p in rest.split('[')][1:]
 134             print parts
 135             id = int(parts[0])
 136
 137             # add a new dictionary if it doesn't exist yet
 138             if id not in dic:
 139                 dic[id] = {}
 140
 141             # add the information to the dictionary
 142             dic[id][parts[1]] = post.get(k)
 143     return dic