1 from manifold.core.query import Query
2 from manifoldapi.manifoldapi import execute_query
4 from django.http import HttpResponse
11 # handles serialization of datetime in json
12 DateEncoder = lambda obj: obj.strftime("%B %d, %Y %H:%M:%S") if isinstance(obj, datetime.datetime) else None
14 # support converting decimal in json
15 json.encoder.FLOAT_REPR = lambda o: format(o, '.2f')
17 # handles decimal numbers serialization in json
18 class DecimalEncoder(json.JSONEncoder):
19 def _iterencode(self, o, markers=None):
20 if isinstance(o, decimal.Decimal):
21 return (str(o) for o in [o])
22 return super(DecimalEncoder, self)._iterencode(o, markers)
24 class ObjectRequest(object):
26 def __init__(self, request, object_type, object_name):
27 self.type = object_type
28 self.name = object_name
34 self.request = request
36 if ((self.type == 'platform') or (self.type == 'testbed')) :
37 self.type = 'local:platform'
39 self.fields = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
40 self.filters['disabled'] = '0'
41 self.filters['gateway_type'] = 'sfa'
42 self.filters['platform'] = '!myslice'
43 elif(self.type.startswith('local:')):
44 # XXX TODO: find a generic Query to get the fields like
45 # select column.name from local:object where table == local:user
46 table = self.type.split(':')
49 self.id = table + '_id'
50 self.fields = ['user_id', 'email', 'password', 'config','status'];
51 elif table == "account":
52 # XXX TODO: Multiple key for account = (platform_id, user_id)
53 self.id = "platform_id, user_id"
54 self.fields = ['platform_id', 'user_id', 'auth_type', 'config'];
55 elif table == "platform":
57 self.fields = ['platform', 'platform_longname', 'platform_url', 'platform_description','gateway_type'];
63 # What about key formed of multiple fields???
64 query = Query.get('local:object').filter_by('table', '==', self.type).select('key')
65 results = execute_query(self.request, query)
66 print "key of object = %s" % results
68 for r in results[0]['key'] :
71 raise Exception, 'Manifold db error'
73 def setLocalFields(self):
74 query = Query.get('local:object').filter_by('table', '==', self.type).select('column.name')
75 results = execute_query(self.request, query)
77 for r in results[0]['column'] :
78 self.fields.append(r['name'])
80 raise Exception, 'Manifold db error'
82 def setFields(self, fields):
83 # selected_fields = []
85 # if p in self.fields :
86 # selected_fields.append(p)
90 def applyFilters(self, query, force_filters = False):
91 if (force_filters and not self.filters) :
92 raise Exception, "Filters required"
94 for k, f in self.filters.iteritems() :
96 query.filter_by(k, '!=', f[1:])
97 elif (f[:2] == ">=") :
98 query.filter_by(k, '>=', f[2:])
100 query.filter_by(k, '>', f[1:])
101 elif (f[:2] == "<=") :
102 query.filter_by(k, '<=', f[2:])
103 elif (f[:1] == "<") :
104 query.filter_by(k, '<', f[1:])
106 query.filter_by(k, '==', f)
110 query = Query.get(self.type)
111 if (self.id is not None) and (self.id not in self.fields) :
112 query.select(self.fields + [self.id])
114 query.select(self.fields)
116 query = self.applyFilters(query)
117 return execute_query(self.request, query)
120 query = Query.create(self.type)
121 # No filters for create
123 for p in self.params :
124 for k,v in p.iteritems() :
125 print "param: %s : %s" % (k,v)
127 print "query = ",query
129 raise Exception, "Params are required for create"
130 return execute_query(self.request, query)
133 query = Query.update(self.type)
134 query = self.applyFilters(query, True)
137 for p in self.params :
138 for k,v in p.iteritems() :
139 print "param: %s : %s" % (k,v)
141 print "query = ",query
143 raise Exception, "Params are required for update"
145 if self.id is not None:
146 query.select(self.id)
148 return execute_query(self.request, query)
151 query = Query.delete(self.type)
152 query = self.applyFilters(query, True)
154 query.set(self.filters)
156 raise Exception, "Filters are required for delete"
157 return execute_query(self.request, query)
160 return HttpResponse(json.dumps(self.get(), cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
163 response = self.get()
165 response_data['fields'] = self.fields
166 response_data['labels'] = self.fields
167 response_data['data'] = []
168 response_data['total'] = len(response)
171 for p in self.fields :
173 response_data['data'].append(d)
175 return HttpResponse(json.dumps(response_data, cls=DecimalEncoder, default=DateEncoder), content_type="application/json")
178 return HttpResponse(json.dumps({'error' : msg}), content_type="application/json")
181 return HttpResponse(json.dumps({'success' : msg}), content_type="application/json")