File: field_function.py

package info (click to toggle)
tryton-server 7.0.40-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 7,748 kB
  • sloc: python: 53,502; xml: 5,194; sh: 803; sql: 217; makefile: 28
file content (91 lines) | stat: -rw-r--r-- 2,676 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# 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, ModelStorage, fields
from trytond.pool import Pool
from trytond.transaction import Transaction


class FunctionDefinition(ModelStorage):
    "Function Definition"
    __name__ = 'test.function.definition'
    function = fields.Function(
        fields.Integer("Integer"),
        'on_change_with_function', searcher='search_function')

    def on_change_with_function(self, name=None):
        return self.id

    @classmethod
    def search_function(cls, name, clause):
        return [('id',) + tuple(clause[1:])]


class FunctionAccessor(ModelSQL):
    "Function Accessor"
    __name__ = 'test.function.accessor'

    target = fields.Many2One('test.function.accessor.target', "Target")
    function = fields.Function(
        fields.Many2One('test.function.accessor.target', "Function"),
        'on_change_with_function')

    @fields.depends('target')
    def on_change_with_function(self, name=None):
        if self.target:
            return self.target.id


class FunctionAccessorTarget(ModelSQL):
    "Function Accessor Target"
    __name__ = 'test.function.accessor.target'


class FunctionGetterContext(ModelSQL):
    "Function Getter Context"
    __name__ = 'test.function.getter_context'

    function_with_context = fields.Function(
        fields.Char("Function"),
        'getter', getter_with_context=True)
    function_without_context = fields.Function(
        fields.Char("Function"),
        'getter', getter_with_context=False)

    def getter(self, name):
        context = Transaction().context
        return '%s - %s' % (
            context.get('language', 'empty'), context.get('test', 'empty'))


class FunctionGetterLocalCache(ModelSQL):
    "Function Getter with local cache"
    __name__ = 'test.function.getter_local_cache'

    function1 = fields.Function(
        fields.Char("Char 1"), 'get_function1')
    function2 = fields.Function(
        fields.Char("Char 2"), 'get_function2')

    def get_function1(self, name):
        return "test"

    def get_function2(self, name):
        return self.function1.upper()

    @classmethod
    def index_get_field(cls, name):
        index = super().index_get_field(name)
        if name == 'function2':
            index = cls.index_get_field('function1') + 1
        return index


def register(module):
    Pool.register(
        FunctionDefinition,
        FunctionAccessor,
        FunctionAccessorTarget,
        FunctionGetterContext,
        FunctionGetterLocalCache,
        module=module, type_='model')