63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
from trytond.transaction import Transaction
|
|
|
|
__all__ = ['RPC']
|
|
|
|
|
|
class RPC(object):
|
|
'''Define RPC behavior
|
|
|
|
readonly: The transaction mode
|
|
instantiate: The position or the slice of the arguments to be instanciated
|
|
result: The function to transform the result
|
|
check_access: If access right must be checked
|
|
'''
|
|
|
|
__slots__ = ('readonly', 'instantiate', 'result', 'check_access')
|
|
|
|
def __init__(self, readonly=True, instantiate=None, result=None,
|
|
check_access=True):
|
|
self.readonly = readonly
|
|
self.instantiate = instantiate
|
|
if result is None:
|
|
result = lambda r: r
|
|
self.result = result
|
|
self.check_access = check_access
|
|
|
|
def convert(self, obj, *args, **kwargs):
|
|
args = list(args)
|
|
kwargs = kwargs.copy()
|
|
if 'context' in kwargs:
|
|
context = kwargs.pop('context')
|
|
else:
|
|
context = args.pop()
|
|
timestamp = None
|
|
for key in context.keys():
|
|
if key == '_timestamp':
|
|
timestamp = context[key]
|
|
# Remove all private keyword but _datetime for history
|
|
if key.startswith('_') and key != '_datetime':
|
|
del context[key]
|
|
if self.instantiate is not None:
|
|
|
|
def instance(data):
|
|
with Transaction().set_context(context):
|
|
if isinstance(data, (int, long)):
|
|
return obj(data)
|
|
elif isinstance(data, dict):
|
|
return obj(**data)
|
|
else:
|
|
return obj.browse(data)
|
|
if isinstance(self.instantiate, slice):
|
|
for i, data in enumerate(args[self.instantiate]):
|
|
start, _, step = self.instantiate.indices(len(args))
|
|
i = i * step + start
|
|
args[i] = instance(data)
|
|
else:
|
|
data = args[self.instantiate]
|
|
args[self.instantiate] = instance(data)
|
|
if self.check_access:
|
|
context['_check_access'] = True
|
|
return args, kwargs, context, timestamp
|