Node with related resource information.
[odoo/odoo.git] / addons / process / process.py
1 # -*- encoding: utf-8 -*-
2 ##############################################################################
3 #
4 # Copyright (c) 2005-TODAY TINY SPRL. (http://tiny.be) All Rights Reserved.
5 #
6 # WARNING: This program as such is intended to be used by professional
7 # programmers who take the whole responsability of assessing all potential
8 # consequences resulting from its eventual inadequacies and bugs
9 # End users who are looking for a ready-to-use solution with commercial
10 # garantees and support are strongly adviced to contract a Free Software
11 # Service Company
12 #
13 # This program is Free Software; you can redistribute it and/or
14 # modify it under the terms of the GNU General Public License
15 # as published by the Free Software Foundation; either version 2
16 # of the License, or (at your option) any later version.
17 #
18 # This program is distributed in the hope that it will be useful,
19 # but WITHOUT ANY WARRANTY; without even the implied warranty of
20 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21 # GNU General Public License for more details.
22 #
23 # You should have received a copy of the GNU General Public License
24 # along with this program; if not, write to the Free Software
25 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
26 #
27 ##############################################################################
28
29 import netsvc
30 import pooler, tools
31
32 from osv import fields, osv
33
34 class Env(dict):
35     
36     def __init__(self, obj, user):
37         self.__obj = obj
38         self.__usr = user
39         
40     def __getitem__(self, name):
41         
42         if name in ('__obj', '__user'):
43             return super(ExprContext, self).__getitem__(name)
44         
45         if name == 'user':
46             return self.__user
47         
48         if name == 'object':
49             return self.__obj
50         
51         return self.__obj[name]
52
53 class process_process(osv.osv):
54     _name = "process.process"
55     _description = "Process"
56     _columns = {
57         'name': fields.char('Name', size=30,required=True, translate=True),
58         'active': fields.boolean('Active'),
59         'note': fields.text('Notes', translate=True),
60         'node_ids': fields.one2many('process.node', 'process_id', 'Nodes')
61     }
62     _defaults = {
63         'active' : lambda *a: True,
64     }
65
66     def graph_get(self, cr, uid, id, res_model, res_id, scale, context):
67         
68         pool = pooler.get_pool(cr.dbname)
69         
70         process = pool.get('process.process').browse(cr, uid, [id])[0]
71         current_object = pool.get(res_model).browse(cr, uid, [res_id], context)[0]
72         current_user = pool.get('res.users').browse(cr, uid, [uid], context)[0]
73         
74         expr_context = Env(current_object, current_user)
75
76         def get_resource_info(node):
77             ret = False
78
79             src_model = res_model
80             src_id = res_id
81             
82             if node.transition_in:
83                 tr = node.transition_in[0]
84                 src = nodes.get(tr.source_node_id.id)
85                 if src['res']:
86                     src_model = src['res']['model']
87                     src_id = src['res']['id']
88                 else:
89                     return False
90
91             fields = pool.get(src_model).fields_get(cr, uid, context=context)
92
93             for name, field in fields.items():
94                 if node.model_id and field.get('relation', False) == node.model_id.model:
95                     src_obj = pool.get(src_model).browse(cr, uid, [src_id], context)[0]
96                     rel = src_obj[name]
97                     if rel:
98                         if isinstance(rel, (list, tuple)):
99                             rel = rel[0]
100                         ret = {}
101                         ret['name'] = rel.name_get(context)[0][1]
102                         ret['model'] = field['relation']
103                         ret['id'] = rel.id
104
105             return ret
106         
107         nodes = {}
108         start = []
109         transitions = {}
110
111         for node in process.node_ids:
112             data = {}
113             data['name'] = node.name
114             data['model'] = (node.model_id or None) and node.model_id.model
115             data['kind'] = node.kind
116             data['subflow'] = (node.subflow_id or None) and node.subflow_id.id
117             data['notes'] = node.note
118             data['active'] = False
119             data['gray'] = False
120             data['res'] = get_resource_info(node)
121
122             if node.menu_id:
123                 data['menu'] = {'name': node.menu_id.complete_name, 'id': node.menu_id.id}
124             
125             if node.model_id and node.model_id.model == res_model:
126
127                 data['res'] = resource = {}
128                 resource['name'] = current_object.name_get(context)[0][1]
129                 resource['model'] = res_model
130                 resource['id'] = res_id
131
132                 try:
133                     data['active'] = eval(node.model_states, expr_context)
134                 except Exception, e:
135                     # waring: invalid state expression
136                     pass
137
138             if not data['active']:
139                 try:
140                     gray = True
141                     for cond in node.condition_ids:
142                         if cond.model_id and cond.model_id.model == res_model:
143                             gray = gray and eval(cond.model_states, expr_context)
144                     data['gray'] = not gray
145                 except:
146                     pass
147
148             nodes[node.id] = data
149             if node.flow_start:
150                 start.append(node.id)
151
152             for tr in node.transition_out:
153                 data = {}
154                 data['name'] = tr.name
155                 data['source'] = tr.source_node_id.id
156                 data['target'] = tr.target_node_id.id
157                 data['notes'] = tr.note
158                 data['buttons'] = buttons = []
159                 for b in tr.action_ids:
160                     button = {}
161                     button['name'] = b.name
162                     button['state'] = b.state
163                     button['action'] = b.action
164                     buttons.append(button)
165                 data['roles'] = roles = []
166                 for r in tr.transition_ids:
167                     if r.role_id:
168                         role = {}
169                         role['name'] = r.role_id.name
170                         roles.append(role)
171                 transitions[tr.id] = data
172
173         g = tools.graph(nodes.keys(), map(lambda x: (x['source'], x['target']), transitions.values()))
174         g.process(start)
175         #g.scale(100, 100, 180, 120)
176         g.scale(*scale)
177         graph = g.result_get()
178         miny = -1
179
180         for k,v in nodes.items():
181             x = graph[k]['y']
182             y = graph[k]['x']
183             if miny == -1:
184                 miny = y
185             miny = min(y, miny)
186             v['x'] = x
187             v['y'] = y
188
189         for k, v in nodes.items():
190             y = v['y']
191             v['y'] = min(y - miny + 10, y)
192         return dict(nodes=nodes, transitions=transitions)
193
194 process_process()
195
196 class process_node(osv.osv):
197     _name = 'process.node'
198     _description ='Process Nodes'
199     _columns = {
200         'name': fields.char('Name', size=30,required=True, translate=True),
201         'process_id': fields.many2one('process.process', 'Process', required=True, ondelete='cascade'),
202         'kind': fields.selection([('state','State'), ('subflow','Subflow')], 'Kind of Node', required=True),
203         'menu_id': fields.many2one('ir.ui.menu', 'Related Menu'),
204         'note': fields.text('Notes', translate=True),
205         'model_id': fields.many2one('ir.model', 'Object', ondelete='set null'),
206         'model_states': fields.char('States Expression', size=128),
207         'subflow_id': fields.many2one('process.process', 'Subflow', ondelete='set null'),
208         'flow_start': fields.boolean('Starting Flow'),
209         'transition_in': fields.one2many('process.transition', 'target_node_id', 'Starting Transitions'),
210         'transition_out': fields.one2many('process.transition', 'source_node_id', 'Ending Transitions'),
211         'condition_ids': fields.one2many('process.condition', 'node_id', 'Conditions')
212     }
213     _defaults = {
214         'kind': lambda *args: 'state',
215         'model_states': lambda *args: False,
216         'flow_start': lambda *args: False,
217     }
218 process_node()
219
220 class process_node_condition(osv.osv):
221     _name = 'process.condition'
222     _description = 'Condition'
223     _columns = {
224         'name': fields.char('Name', size=30, required=True),
225         'node_id': fields.many2one('process.node', 'Node', required=True, ondelete='cascade'),
226         'model_id': fields.many2one('ir.model', 'Object', ondelete='set null'),
227         'model_states': fields.char('Expression', required=True, size=128)
228     }
229 process_node_condition()
230
231 class process_transition(osv.osv):
232     _name = 'process.transition'
233     _description ='Process Transitions'
234     _columns = {
235         'name': fields.char('Name', size=32, required=True, translate=True),
236         'source_node_id': fields.many2one('process.node', 'Source Node', required=True, ondelete='cascade'),
237         'target_node_id': fields.many2one('process.node', 'Target Node', required=True, ondelete='cascade'),
238         'action_ids': fields.one2many('process.transition.action', 'transition_id', 'Buttons'),
239         'transition_ids': fields.many2many('workflow.transition', 'process_transition_ids', 'ptr_id', 'wtr_id', 'Workflow Transitions'),
240         'note': fields.text('Description', translate=True),
241     }
242 process_transition()
243
244 class process_transition_action(osv.osv):
245     _name = 'process.transition.action'
246     _description ='Process Transitions Actions'
247     _columns = {
248         'name': fields.char('Name', size=32, required=True, translate=True),
249         'state': fields.selection([('dummy','Dummy'),
250                                    ('object','Object Method'),
251                                    ('workflow','Workflow Trigger'),
252                                    ('action','Action')], 'Type', required=True),
253         'action': fields.char('Action ID', size=64, states={
254             'dummy':[('readonly',1)],
255             'object':[('required',1)],
256             'workflow':[('required',1)],
257             'action':[('required',1)],
258         },),
259         'transition_id': fields.many2one('process.transition', 'Transition', required=True, ondelete='cascade')
260     }
261     _defaults = {
262         'state': lambda *args: 'dummy',
263     }
264 process_transition_action()
265