← Back to branch summary

~ubuntu-branches/ubuntu/quantal/openerp6.1/quantal

« back to all changes in this revision

Viewing changes to openerp/tools/yaml_tag.py

  • Committer: Package Import Robot
  • Author(s): Yolanda Robla
  • Date: 2012-09-20 15:29:00 UTC
  • Revision ID: package-import@ubuntu.com-20120920152900-woyy3yww8z6acmsk
Tags: upstream-6.1-1+dfsg
Import upstream version 6.1-1+dfsg

Show diffs side-by-side

added added

removed removed

Lines of Context:
openerp/tools/yaml_tag.py
 
1
import yaml
 
2
import logging
 
3
 
 
4
class YamlTag(object):
 
5
    """
 
6
    Superclass for constructors of custom tags defined in yaml file.
 
7
    __str__ is overriden in subclass and used for serialization in module recorder.
 
8
    """
 
9
    def __init__(self, **kwargs):
 
10
        self.__dict__.update(kwargs)
 
11
    def __getitem__(self, key):
 
12
        return getattr(self, key)
 
13
    def __getattr__(self, attr):
 
14
        return None
 
15
    def __repr__(self):
 
16
        return "<%s %s>" % (self.__class__.__name__, sorted(self.__dict__.items()))
 
17
 
 
18
class Assert(YamlTag):
 
19
    def __init__(self, model, id=None, severity=logging.WARNING, string="NONAME", **kwargs):
 
20
        self.model = model
 
21
        self.id = id
 
22
        self.severity = severity
 
23
        self.string = string
 
24
        super(Assert, self).__init__(**kwargs)
 
25
    
 
26
class Record(YamlTag):
 
27
    def __init__(self, model, id, use='id', view=True, **kwargs):
 
28
        self.model = model
 
29
        self.id = id
 
30
        self.view = view
 
31
        super(Record, self).__init__(**kwargs)
 
32
    def __str__(self):
 
33
        return '!record {model: %s, id: %s}:' % (str(self.model,), str(self.id,))
 
34
 
 
35
class Python(YamlTag):
 
36
    def __init__(self, model, severity=logging.ERROR, name="", **kwargs):
 
37
        self.model= model
 
38
        self.severity = severity
 
39
        self.name = name
 
40
        super(Python, self).__init__(**kwargs)
 
41
    def __str__(self):
 
42
        return '!python {model: %s}: |' % (str(self.model), )
 
43
 
 
44
class Menuitem(YamlTag):
 
45
    def __init__(self, id, name, **kwargs):
 
46
        self.id = id
 
47
        self.name = name
 
48
        super(Menuitem, self).__init__(**kwargs)
 
49
 
 
50
class Workflow(YamlTag):
 
51
    def __init__(self, model, action, ref=None, **kwargs):
 
52
        self.model = model
 
53
        self.action = action
 
54
        self.ref = ref
 
55
        super(Workflow, self).__init__(**kwargs)
 
56
    def __str__(self):
 
57
        return '!workflow {model: %s, action: %s, ref: %s}' % (str(self.model,), str(self.action,), str(self.ref,))
 
58
 
 
59
class ActWindow(YamlTag):
 
60
    def __init__(self, **kwargs):
 
61
        super(ActWindow, self).__init__(**kwargs)
 
62
 
 
63
class Function(YamlTag):
 
64
    def __init__(self, model, name, **kwargs):
 
65
        self.model = model
 
66
        self.name = name
 
67
        super(Function, self).__init__(**kwargs)
 
68
 
 
69
class Report(YamlTag):
 
70
    def __init__(self, model, name, string, **kwargs):
 
71
        self.model = model
 
72
        self.name = name
 
73
        self.string = string
 
74
        super(Report, self).__init__(**kwargs)
 
75
 
 
76
class Delete(YamlTag):
 
77
    def __init__(self, **kwargs):
 
78
        super(Delete, self).__init__(**kwargs)
 
79
 
 
80
class Context(YamlTag):
 
81
    def __init__(self, **kwargs):
 
82
        super(Context, self).__init__(**kwargs)
 
83
 
 
84
class Url(YamlTag):
 
85
    def __init__(self, **kwargs):
 
86
        super(Url, self).__init__(**kwargs)
 
87
 
 
88
class Eval(YamlTag):
 
89
    def __init__(self, expression):
 
90
        self.expression = expression
 
91
        super(Eval, self).__init__()
 
92
    def __str__(self):
 
93
        return '!eval %s' % str(self.expression)
 
94
    
 
95
class Ref(YamlTag):
 
96
    def __init__(self, expr="False", *args, **kwargs):
 
97
        self.expr = expr
 
98
        super(Ref, self).__init__(*args, **kwargs)
 
99
    def __str__(self):
 
100
        return 'ref(%s)' % repr(self.expr)
 
101
    
 
102
class IrSet(YamlTag):
 
103
    def __init__(self):
 
104
        super(IrSet, self).__init__()
 
105
 
 
106
def assert_constructor(loader, node):
 
107
    kwargs = loader.construct_mapping(node)
 
108
    return Assert(**kwargs)
 
109
 
 
110
def record_constructor(loader, node):
 
111
    kwargs = loader.construct_mapping(node)
 
112
    return Record(**kwargs)
 
113
 
 
114
def python_constructor(loader, node):
 
115
    kwargs = loader.construct_mapping(node)
 
116
    return Python(**kwargs)
 
117
 
 
118
def menuitem_constructor(loader, node):
 
119
    kwargs = loader.construct_mapping(node)
 
120
    return Menuitem(**kwargs)
 
121
 
 
122
def workflow_constructor(loader, node):
 
123
    kwargs = loader.construct_mapping(node)
 
124
    return Workflow(**kwargs)
 
125
 
 
126
def act_window_constructor(loader, node):
 
127
    kwargs = loader.construct_mapping(node)
 
128
    return ActWindow(**kwargs)
 
129
 
 
130
def function_constructor(loader, node):
 
131
    kwargs = loader.construct_mapping(node)
 
132
    return Function(**kwargs)
 
133
 
 
134
def report_constructor(loader, node):
 
135
    kwargs = loader.construct_mapping(node)
 
136
    return Report(**kwargs)
 
137
 
 
138
def delete_constructor(loader, node):
 
139
    kwargs = loader.construct_mapping(node)
 
140
    return Delete(**kwargs)
 
141
 
 
142
def context_constructor(loader, node):
 
143
    kwargs = loader.construct_mapping(node)
 
144
    return Context(**kwargs)
 
145
 
 
146
def url_constructor(loader, node):
 
147
    kwargs = loader.construct_mapping(node)
 
148
    return Url(**kwargs)
 
149
 
 
150
def eval_constructor(loader, node):
 
151
    expression = loader.construct_scalar(node)
 
152
    return Eval(expression)
 
153
    
 
154
def ref_constructor(loader, tag_suffix, node):
 
155
    if tag_suffix == "id":
 
156
        kwargs = {"id": loader.construct_scalar(node)}
 
157
    else:
 
158
        kwargs = loader.construct_mapping(node)
 
159
    return Ref(**kwargs)
 
160
    
 
161
def ir_set_constructor(loader, node):
 
162
    kwargs = loader.construct_mapping(node)
 
163
    return IrSet(**kwargs)
 
164
    
 
165
# Registers constructors for custom tags.
 
166
# Constructors are actually defined globally: do not redefined them in another
 
167
# class/file/package.  This means that module recorder need import this file.
 
168
def add_constructors():
 
169
    yaml.add_constructor(u"!assert", assert_constructor)
 
170
    yaml.add_constructor(u"!record", record_constructor)
 
171
    yaml.add_constructor(u"!python", python_constructor)
 
172
    yaml.add_constructor(u"!menuitem", menuitem_constructor)
 
173
    yaml.add_constructor(u"!workflow", workflow_constructor)
 
174
    yaml.add_constructor(u"!act_window", act_window_constructor)
 
175
    yaml.add_constructor(u"!function", function_constructor)
 
176
    yaml.add_constructor(u"!report", report_constructor)
 
177
    yaml.add_constructor(u"!context", context_constructor)
 
178
    yaml.add_constructor(u"!delete", delete_constructor)
 
179
    yaml.add_constructor(u"!url", url_constructor)
 
180
    yaml.add_constructor(u"!eval", eval_constructor)
 
181
    yaml.add_multi_constructor(u"!ref", ref_constructor)
 
182
    yaml.add_constructor(u"!ir_set", ir_set_constructor)
 
183
add_constructors()
 
184
 
 
185
 
 
186
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: