presik_pos/app/proxy.py

104 lines
2.7 KiB
Python

import requests
from datetime import date
import simplejson as json
def encoder(obj):
# FIXME: add datetime, buffer, bytes
if isinstance(obj, date):
return {
'__class__': 'date',
'year': obj.year,
'month': obj.month,
'day': obj.day,
}
raise TypeError(repr(obj) + " is not JSON serializable")
class FastModel(object):
def __init__(self, model, ctx):
# params = parent.params
# self.model = model['name']
self.model = model
# self.fields = model['fields']
self.ctx = ctx
self.api = '/'.join(
['http:/', ctx['params']['api_url'], ctx['params']['database']]
)
def __getattr__(self, name, *args):
'Return attribute value'
self.method = name
return self
def find(self, domain, order=None, limit=100, ctx=None):
if ctx:
self.ctx.update(ctx)
route = self.get_route('search')
args_ = {
'model': self.model,
'domain': domain,
'order': order,
'limit': limit,
'context': self.ctx,
}
data = json.dumps(args_, default=encoder)
res = requests.get(route, data=data)
print('res ', res)
return res.json()
def write(self, ids, values):
route = self.get_route('save')
args_ = {
'model': self.model,
'id': ids[0],
'record_data': values,
'context': self.ctx,
}
data = json.dumps(args_, default=encoder)
res = requests.put(route, data=data)
return res.json()
def delete(self, ids):
route = self.get_route('delete')
args_ = {
'model': self.model,
'ids': ids,
'context': self.ctx,
}
data = json.dumps(args_, default=encoder)
res = requests.delete(route, data=data)
return res.json()
def get_route(self, target):
route = self.api + '/' + target
return route
def __call__(self, args=None):
args_ = {
'model': self.model,
'method': self.method,
'args': args,
'context': self.ctx,
}
route = self.get_route('model_method')
data = json.dumps(args_, default=encoder)
res = requests.post(route, data=data)
data = res.json()
return data
if __name__ == "__main__":
params = {'api_url': 'localhost:5070', 'database': 'DEMORD'}
model = {'model': 'sale.sale'}
test_model = FastModel(params, model)
id = 180
data = {
'reference': 'OC-02874'
}
# res = test_model.find(dom)
res = test_model.write([id], data)
print(res)