Merge branch 'onelab' of ssh://git.onelab.eu/git/myslice into onelab
[myslice.git] / rest / delete.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         # No params in delete    
36         self.properties = []
37         self.filters = {}
38         self.options = None
39
40         self.request = request
41
42         # What about key formed of multiple fields???
43         query = Query.get('local:object').filter_by('table', '==', self.type).select('key')
44         results = execute_query(self.request, query)
45         print "key of object = %s" % results
46         if results :
47             for r in results[0]['key'] :
48                 self.id = r
49         else :
50             return error('Manifold db error')
51
52         query = Query.get('local:object').filter_by('table', '==', self.type).select('column.name')
53         results = execute_query(self.request, query)
54         if results :
55             for r in results[0]['column'] :
56                 self.properties.append(r['name'])
57         else :
58             return error('Manifold db error')
59
60     def addFilters(self, properties):
61         selected_properties = []
62         for p in properties :
63             if p in self.properties :
64                 selected_properties.append(p)
65         self.properties = selected_properties
66         self.setId()
67
68     def setId(self):
69         if self.id in self.properties :
70             self.properties.remove(self.id)
71             [self.id].extend(self.properties)
72
73     def execute(self):
74         query = Query.delete(self.type)
75         if self.filters :
76             for k, f in self.filters.iteritems() :
77                 if (f[:1] == "!") :
78                     query.filter_by(k, '!=', f[1:])
79                 elif (f[:2] == ">=") :
80                     query.filter_by(k, '>=', f[2:])
81                 elif (f[:1] == ">") :
82                     query.filter_by(k, '>', f[1:])
83                 elif (f[:2] == "<=") :
84                     query.filter_by(k, '<=', f[2:])
85                 elif (f[:1] == "<") :
86                     query.filter_by(k, '<', f[1:])
87                 else :
88                     query.filter_by(k, '==', f)
89         else:
90             raise Exception, "Filters are required for delete"
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     object_filters = {}
98     object_params = {}
99     result = {}
100     
101     if request.method == 'POST':
102         req_items = request.POST
103     elif request.method == 'GET':
104         #return HttpResponse(json.dumps({'error' : 'only post request is supported'}), content_type="application/json")
105         req_items = request.GET
106
107     for el in req_items.items():
108         if el[0].startswith('filters'):
109             o.filters[el[0][8:-1]] = el[1]
110         elif el[0].startswith('columns'):
111             o.addFilters(req_items.getlist('columns[]'))
112         elif el[0].startswith('options'):
113             o.options = req_items.getlist('options[]')
114
115     try:
116         response = o.execute()
117
118         if response :
119             return HttpResponse(json.dumps({'success' : 'record deleted'}), content_type="application/json")
120         else :
121             return HttpResponse(json.dumps({'error' : 'an error has occurred'}), content_type="application/json")
122  
123     except Exception, e:
124         return HttpResponse(json.dumps({'error' : str(e)}), content_type="application/json")
125