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