ddd2ba251e
--HG-- rename : wsme/protocols/__init__.py => wsme/protocol.py rename : wsme/protocols/commons.py => wsme/rest/args.py rename : wsme/protocols/restjson.py => wsme/rest/json.py rename : wsme/protocols/rest.py => wsme/rest/protocol.py rename : wsme/protocols/restxml.py => wsme/rest/xml.py
77 lines
1.9 KiB
Python
77 lines
1.9 KiB
Python
import inspect
|
|
import sys
|
|
|
|
import json
|
|
|
|
import xml.etree.ElementTree as et
|
|
|
|
import wsme
|
|
import wsme.rest.args
|
|
import wsme.rest.json
|
|
import wsme.rest.xml
|
|
|
|
pecan = sys.modules['pecan']
|
|
|
|
|
|
class JSonRenderer(object):
|
|
def __init__(self, path, extra_vars):
|
|
pass
|
|
|
|
def render(self, template_path, namespace):
|
|
data = wsme.rest.json.tojson(
|
|
namespace['datatype'],
|
|
namespace['result']
|
|
)
|
|
return json.dumps(data)
|
|
|
|
|
|
class XMLRenderer(object):
|
|
def __init__(self, path, extra_vars):
|
|
pass
|
|
|
|
def render(self, template_path, namespace):
|
|
data = wsme.rest.xml.toxml(
|
|
namespace['datatype'],
|
|
'result',
|
|
namespace['result']
|
|
)
|
|
return et.tostring(data)
|
|
|
|
pecan.templating._builtin_renderers['wsmejson'] = JSonRenderer
|
|
pecan.templating._builtin_renderers['wsmexml'] = XMLRenderer
|
|
|
|
|
|
def wsexpose(*args, **kwargs):
|
|
pecan_json_decorate = pecan.expose(
|
|
template='wsmejson:',
|
|
content_type='application/json',
|
|
generic=False)
|
|
pecan_xml_decorate = pecan.expose(
|
|
template='wsmexml:',
|
|
content_type='application/xml',
|
|
generic=False
|
|
)
|
|
sig = wsme.signature(*args, **kwargs)
|
|
|
|
def decorate(f):
|
|
sig(f)
|
|
funcdef = wsme.api.FunctionDefinition.get(f)
|
|
|
|
def callfunction(self, *args, **kwargs):
|
|
args, kwargs = wsme.rest.args.get_args(
|
|
funcdef, args, kwargs,
|
|
pecan.request.body, pecan.request.content_type
|
|
)
|
|
result = f(self, *args, **kwargs)
|
|
return dict(
|
|
datatype=funcdef.return_type,
|
|
result=result
|
|
)
|
|
|
|
pecan_xml_decorate(callfunction)
|
|
pecan_json_decorate(callfunction)
|
|
pecan.util._cfg(callfunction)['argspec'] = inspect.getargspec(f)
|
|
return callfunction
|
|
|
|
return decorate
|