Cédric Krier
3e9784c8d7
Field may have a context defined so when it instantiate a relational record, the field should set this context like when the record is read from the database in ModelStorage.__getattr__. And when such field is read as related field, the context should also be set. issue8225 review287171002
59 lines
1.6 KiB
Python
59 lines
1.6 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.model import ModelSQL, fields
|
|
from trytond.pool import Pool
|
|
from trytond.transaction import Transaction
|
|
|
|
|
|
class Reference(ModelSQL):
|
|
'Reference'
|
|
__name__ = 'test.reference'
|
|
reference = fields.Reference('Reference', selection=[
|
|
(None, ''),
|
|
('test.reference.target', 'Target'),
|
|
])
|
|
|
|
|
|
class ReferenceTarget(ModelSQL):
|
|
'Reference Target'
|
|
__name__ = 'test.reference.target'
|
|
name = fields.Char('Name', required=True)
|
|
|
|
|
|
class ReferenceRequired(ModelSQL):
|
|
'Reference Required'
|
|
__name__ = 'test.reference_required'
|
|
reference = fields.Reference('Reference', selection=[
|
|
(None, ''),
|
|
('test.reference.target', 'Target'),
|
|
], required=True)
|
|
|
|
|
|
class ReferenceContext(ModelSQL):
|
|
"Reference Context"
|
|
__name__ = 'test.reference_context'
|
|
target = fields.Reference("Reference", selection=[
|
|
('test.reference_context.target', "Target"),
|
|
], context={'test': 'foo'})
|
|
|
|
|
|
class ReferenceContextTarget(ModelSQL):
|
|
"Reference Context Target"
|
|
__name__ = 'test.reference_context.target'
|
|
context = fields.Function(fields.Char("context"), 'get_context')
|
|
|
|
def get_context(self, name):
|
|
context = Transaction().context
|
|
return context.get('test')
|
|
|
|
|
|
def register(module):
|
|
Pool.register(
|
|
Reference,
|
|
ReferenceTarget,
|
|
ReferenceRequired,
|
|
ReferenceContext,
|
|
ReferenceContextTarget,
|
|
module=module, type_='model')
|