302 lines
9.8 KiB
Python
302 lines
9.8 KiB
Python
#!/usr/bin/python
|
|
import functools
|
|
import optparse
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import traceback
|
|
import uuid
|
|
import xmlrpclib
|
|
|
|
import cherrypy
|
|
import cherrypy.lib.static
|
|
import simplejson
|
|
|
|
import xmlrpctimeout
|
|
|
|
#----------------------------------------------------------
|
|
# OpenERP Client Library
|
|
#----------------------------------------------------------
|
|
class OpenERPUnboundException(Exception):
|
|
pass
|
|
|
|
|
|
class OpenERPConnector(object):
|
|
pass
|
|
|
|
|
|
class OpenERPAuth(object):
|
|
pass
|
|
|
|
|
|
class OpenERPModel(object):
|
|
def __init__(self, session, model):
|
|
self._session = session
|
|
self._model = model
|
|
|
|
def __getattr__(self, name):
|
|
return lambda *l:self._session.execute(self._model, name, *l)
|
|
|
|
|
|
class OpenERPSession(object):
|
|
def __init__(self, server='127.0.0.1', port=8069,
|
|
model_factory=OpenERPModel):
|
|
self._server = server
|
|
self._port = port
|
|
self._db = False
|
|
self._uid = False
|
|
self._login = False
|
|
self._password = False
|
|
self.model_factory = model_factory
|
|
|
|
def proxy(self, service):
|
|
s = xmlrpctimeout.TimeoutServerProxy('http://%s:%s/xmlrpc/%s' % (self._server, self._port, service), timeout=5)
|
|
return s
|
|
|
|
def bind(self, db, uid, password):
|
|
self._db = db
|
|
self._uid = uid
|
|
self._password = password
|
|
|
|
def login(self, db, login, password):
|
|
uid = self.proxy('common').login(db, login, password)
|
|
self.bind(db, uid, password)
|
|
self._login = login
|
|
return uid
|
|
|
|
def execute(self, model, func, *l, **d):
|
|
if not (self._db and self._uid and self._password):
|
|
raise OpenERPUnboundException()
|
|
r = self.proxy('object').execute(self._db, self._uid, self._password, model, func, *l, **d)
|
|
return r
|
|
|
|
def model(self, model):
|
|
return self.model_factory(self, model)
|
|
|
|
#----------------------------------------------------------
|
|
# OpenERP Web RequestHandler
|
|
#----------------------------------------------------------
|
|
|
|
class JsonRequest(object):
|
|
""" JSON-RPC2 over HTTP POST using non standard POST encoding.
|
|
Difference with the standard:
|
|
- the json string is passed as a form parameter named "request"
|
|
- method is currently ignored
|
|
|
|
Sucessful request:
|
|
--> {"jsonrpc": "2.0", "method": "call", "params": {"session_id": "SID", "context": {}, "arg1": "val1" }, "id": null}
|
|
<-- {"jsonrpc": "2.0", "result": { "res1": "val1" }, "id": null}
|
|
|
|
Request producing a error:
|
|
--> {"jsonrpc": "2.0", "method": "call", "params": {"session_id": "SID", "context": {}, "arg1": "val1" }, "id": null}
|
|
<-- {"jsonrpc": "2.0", "error": {"code": 1, "message": "End user error message.", "data": {"code": "codestring", "debug": "traceback" } }, "id": null}
|
|
|
|
"""
|
|
|
|
def parse(self, request):
|
|
self.params = request.get("params", {})
|
|
self.session_id = self.params.pop("session_id", None) or uuid.uuid4().hex
|
|
self.session = cherrypy.session.setdefault(self.session_id, OpenERPSession())
|
|
self.context = self.params.pop('context', None)
|
|
return self.params
|
|
|
|
def dispatch(self, controller, method, requestf=None, request=None):
|
|
''' Calls the method asked for by the JSON-RPC2 request
|
|
|
|
:param controller: the instance of the controller which received the request
|
|
:type controller: type
|
|
:param method: the method which received the request
|
|
:type method: callable
|
|
:param requestf: a file-like object containing an encoded JSON-RPC2 request
|
|
:type requestf: <read() -> bytes>
|
|
:param request: an encoded JSON-RPC2 request
|
|
:type request: bytes
|
|
|
|
:returns: a string-encoded JSON-RPC2 reply
|
|
:rtype: bytes
|
|
'''
|
|
if requestf:
|
|
request = simplejson.load(requestf)
|
|
else:
|
|
request = simplejson.loads(request)
|
|
try:
|
|
print "--> %s.%s %s" % (controller.__class__.__name__, method.__name__, request)
|
|
error = None
|
|
result = method(controller, self, **self.parse(request))
|
|
except OpenERPUnboundException:
|
|
error = {
|
|
'code': 100,
|
|
'message': "OpenERP Session Invalid",
|
|
'data': {
|
|
'type': 'session_invalid',
|
|
'debug': traceback.format_exc()
|
|
}
|
|
}
|
|
except xmlrpclib.Fault, e:
|
|
error = {
|
|
'code': 200,
|
|
'message': "OpenERP Server Error",
|
|
'data': {
|
|
'type': 'server_exception',
|
|
'fault_code': e.faultCode,
|
|
'debug': "Client %s\nServer %s" % (
|
|
"".join(traceback.format_exception("", None, sys.exc_traceback)), e.faultString)
|
|
}
|
|
}
|
|
except Exception:
|
|
error = {
|
|
'code': 300,
|
|
'message': "OpenERP WebClient Error",
|
|
'data': {
|
|
'type': 'client_exception',
|
|
'debug': "Client %s" % traceback.format_exc()
|
|
}
|
|
}
|
|
response = {"jsonrpc": "2.0", "id": request.get('id')}
|
|
if error:
|
|
response["error"] = error
|
|
else:
|
|
response["result"] = result
|
|
|
|
print "<--", response
|
|
print
|
|
|
|
content = simplejson.dumps(response)
|
|
cherrypy.response.headers['Content-Type'] = 'application/json'
|
|
cherrypy.response.headers['Content-Length'] = len(content)
|
|
return content
|
|
|
|
|
|
def jsonrequest(f):
|
|
@cherrypy.expose
|
|
@functools.wraps(f)
|
|
def json_handler(self):
|
|
return JsonRequest().dispatch(self, f, requestf=cherrypy.request.body)
|
|
|
|
return json_handler
|
|
|
|
|
|
class HttpRequest(object):
|
|
""" Regular GET/POST request
|
|
"""
|
|
|
|
def __init__(self):
|
|
# result may be filled, it's content will be updated by the return
|
|
# value of the dispatched function if it's a dict
|
|
self.result = ""
|
|
|
|
def dispatch(self, controller, f, request):
|
|
print "GET/POST --> %s.%s %s" % (controller.__class__.__name__, f.__name__, request)
|
|
r = f(controller, self, request)
|
|
return r
|
|
|
|
|
|
def httprequest(f):
|
|
# check cleaner wrapping:
|
|
# functools.wraps(f)(lambda x: JsonRequest().dispatch(x, f))
|
|
l = lambda self, request: HttpRequest().dispatch(self, f, request)
|
|
l.exposed = 1
|
|
return l
|
|
|
|
#-----------------------------------------------------------
|
|
# Cherrypy stuff
|
|
#-----------------------------------------------------------
|
|
|
|
path_root = os.path.dirname(os.path.dirname(os.path.normpath(__file__)))
|
|
path_addons = os.path.join(path_root, 'addons')
|
|
cherrypy_root = None
|
|
|
|
# globals might move into a pool if needed
|
|
addons_module = {}
|
|
addons_manifest = {}
|
|
controllers_class = {}
|
|
controllers_object = {}
|
|
controllers_path = {}
|
|
|
|
class ControllerType(type):
|
|
def __init__(cls, name, bases, attrs):
|
|
super(ControllerType, cls).__init__(name, bases, attrs)
|
|
controllers_class["%s.%s" % (cls.__module__, cls.__name__)] = cls
|
|
|
|
|
|
class Controller(object):
|
|
__metaclass__ = ControllerType
|
|
|
|
|
|
class Root(object):
|
|
def __init__(self):
|
|
self.addons = {}
|
|
self._load_addons()
|
|
|
|
def _load_addons(self):
|
|
if path_addons not in sys.path:
|
|
sys.path.insert(0, path_addons)
|
|
for i in os.listdir(path_addons):
|
|
if i not in sys.modules:
|
|
manifest_path = os.path.join(path_addons, i, '__openerp__.py')
|
|
if os.path.isfile(manifest_path):
|
|
manifest = eval(open(manifest_path).read())
|
|
print "Loading", i
|
|
m = __import__(i)
|
|
addons_module[i] = m
|
|
addons_manifest[i] = manifest
|
|
for k, v in controllers_class.items():
|
|
if k not in controllers_object:
|
|
o = v()
|
|
controllers_object[k] = o
|
|
if hasattr(o, '_cp_path'):
|
|
controllers_path[o._cp_path] = o
|
|
|
|
def default(self, *l, **kw):
|
|
#print "default",l,kw
|
|
# handle static files
|
|
if len(l) > 2 and l[1] == 'static':
|
|
# sanitize path
|
|
p = os.path.normpath(os.path.join(*l))
|
|
return cherrypy.lib.static.serve_file(os.path.join(path_addons, p))
|
|
elif len(l) > 1:
|
|
for i in range(1, len(l) + 1):
|
|
ps = "/" + "/".join(l[0:i])
|
|
if ps in controllers_path:
|
|
c = controllers_path[ps]
|
|
rest = l[i:] or ['index']
|
|
meth = rest[0]
|
|
m = getattr(c, meth)
|
|
if getattr(m, 'exposed', 0):
|
|
print "Calling", ps, c, meth, m
|
|
return m(**kw)
|
|
|
|
else:
|
|
raise cherrypy.HTTPRedirect('/base/static/openerp/base.html', 301)
|
|
default.exposed = True
|
|
|
|
|
|
def main(argv):
|
|
# Parse config
|
|
op = optparse.OptionParser()
|
|
op.add_option("-p", "--port", dest="socket_port", help="listening port", metavar="NUMBER", default=8002)
|
|
op.add_option("-s", "--session-path", dest="storage_path",
|
|
help="directory used for session storage", metavar="DIR",
|
|
default=os.path.join(tempfile.gettempdir(), "cpsessions"))
|
|
(o, args) = op.parse_args(argv[1:])
|
|
|
|
# Prepare cherrypy config from options
|
|
if not os.path.exists(o.storage_path):
|
|
os.mkdir(o.storage_path, 0700)
|
|
config = {
|
|
'server.socket_port': int(o.socket_port),
|
|
'server.socket_host': '0.0.0.0',
|
|
#'server.thread_pool' = 10,
|
|
'tools.sessions.on': True,
|
|
'tools.sessions.storage_type': 'file',
|
|
'tools.sessions.storage_path': o.storage_path,
|
|
'tools.sessions.timeout': 60
|
|
}
|
|
|
|
# Setup and run cherrypy
|
|
cherrypy.tree.mount(Root())
|
|
cherrypy.config.update(config)
|
|
cherrypy.server.subscribe()
|
|
cherrypy.engine.start()
|
|
cherrypy.engine.block()
|