activate the python code in the server action
[odoo/odoo.git] / bin / addons / base / ir / ir_actions.py
1 # -*- encoding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution    
5 #    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
6 #    $Id$
7 #
8 #    This program is free software: you can redistribute it and/or modify
9 #    it under the terms of the GNU General Public License as published by
10 #    the Free Software Foundation, either version 3 of the License, or
11 #    (at your option) any later version.
12 #
13 #    This program is distributed in the hope that it will be useful,
14 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
15 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 #    GNU General Public License for more details.
17 #
18 #    You should have received a copy of the GNU General Public License
19 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20 #
21 ##############################################################################
22
23 from osv import fields,osv
24 import tools
25 import time
26 from tools.config import config
27 import netsvc
28 import re
29
30 class actions(osv.osv):
31     _name = 'ir.actions.actions'
32     _table = 'ir_actions'
33     _columns = {
34         'name': fields.char('Action Name', required=True, size=64),
35         'type': fields.char('Action Type', required=True, size=32),
36         'usage': fields.char('Action Usage', size=32),
37     }
38     _defaults = {
39         'usage': lambda *a: False,
40     }
41 actions()
42
43 class report_custom(osv.osv):
44     _name = 'ir.actions.report.custom'
45     _table = 'ir_act_report_custom'
46     _sequence = 'ir_actions_id_seq'
47     _columns = {
48         'name': fields.char('Report Name', size=64, required=True, translate=True),
49         'type': fields.char('Report Type', size=32, required=True),
50         'model':fields.char('Object', size=64, required=True),
51         'report_id': fields.integer('Report Ref.', required=True),
52         'usage': fields.char('Action Usage', size=32),
53         'multi': fields.boolean('On multiple doc.', help="If set to true, the action will not be displayed on the right toolbar of a form views.")
54     }
55     _defaults = {
56         'multi': lambda *a: False,
57         'type': lambda *a: 'ir.actions.report.custom',
58     }
59 report_custom()
60
61 class report_xml(osv.osv):
62
63     def _report_content(self, cursor, user, ids, name, arg, context=None):
64         res = {}
65         for report in self.browse(cursor, user, ids, context=context):
66             data = report[name + '_data']
67             if not data and report[name[:-8]]:
68                 try:
69                     fp = tools.file_open(report[name[:-8]], mode='rb')
70                     data = fp.read()
71                 except:
72                     data = False
73             res[report.id] = data
74         return res
75
76     def _report_content_inv(self, cursor, user, id, name, value, arg, context=None):
77         self.write(cursor, user, id, {name+'_data': value}, context=context)
78
79     def _report_sxw(self, cursor, user, ids, name, arg, context=None):
80         res = {}
81         for report in self.browse(cursor, user, ids, context=context):
82             if report.report_rml:
83                 res[report.id] = report.report_rml.replace('.rml', '.sxw')
84             else:
85                 res[report.id] = False
86         return res
87
88     _name = 'ir.actions.report.xml'
89     _table = 'ir_act_report_xml'
90     _sequence = 'ir_actions_id_seq'
91     _columns = {
92         'name': fields.char('Name', size=64, required=True, translate=True),
93         'type': fields.char('Report Type', size=32, required=True),
94         'model': fields.char('Object', size=64, required=True),
95         'report_name': fields.char('Internal Name', size=64, required=True),
96         'report_xsl': fields.char('XSL path', size=256),
97         'report_xml': fields.char('XML path', size=256),
98         'report_rml': fields.char('RML path', size=256,
99             help="The .rml path of the file or NULL if the content is in report_rml_content"),
100         'report_sxw': fields.function(_report_sxw, method=True, type='char',
101             string='SXW path'),
102         'report_sxw_content_data': fields.binary('SXW content'),
103         'report_rml_content_data': fields.binary('RML content'),
104         'report_sxw_content': fields.function(_report_content,
105             fnct_inv=_report_content_inv, method=True,
106             type='binary', string='SXW content',),
107         'report_rml_content': fields.function(_report_content,
108             fnct_inv=_report_content_inv, method=True,
109             type='binary', string='RML content'),
110         'auto': fields.boolean('Automatic XSL:RML', required=True),
111         'usage': fields.char('Action Usage', size=32),
112         'header': fields.boolean('Add RML header',
113             help="Add or not the coporate RML header"),
114         'multi': fields.boolean('On multiple doc.',
115             help="If set to true, the action will not be displayed on the right toolbar of a form views."),
116         'report_type': fields.selection([
117             ('pdf', 'pdf'),
118             ('html', 'html'),
119             ('raw', 'raw'),
120             ('sxw', 'sxw'),
121             ], string='Type', required=True),
122         'groups_id': fields.many2many('res.groups', 'res_groups_report_rel', 'uid', 'gid', 'Groups'),
123         'attachment': fields.char('Save As Attachment Prefix', size=32, help='This is the prefix of the file name the print will be saved as attachement. Keep empty to not save the printed reports')
124     }
125     _defaults = {
126         'type': lambda *a: 'ir.actions.report.xml',
127         'multi': lambda *a: False,
128         'auto': lambda *a: True,
129         'header': lambda *a: True,
130         'report_sxw_content': lambda *a: False,
131         'report_type': lambda *a: 'pdf',
132         'attachment': lambda *a: False,
133     }
134
135 report_xml()
136
137 class act_window(osv.osv):
138     _name = 'ir.actions.act_window'
139     _table = 'ir_act_window'
140     _sequence = 'ir_actions_id_seq'
141
142 #    def search(self, cr, uid, args, offset=0, limit=2000, order=None,
143 #            context=None, count=False):
144 #        if context is None:
145 #            context = {}
146 #        ids = osv.orm.orm.search(self, cr, uid, args, offset, limit, order,
147 #                context=context)
148 #        if uid==1:
149 #            return ids
150 #        user_groups = self.pool.get('res.users').read(cr, uid, [uid])[0]['groups_id']
151 #        result = []
152 #        for act in self.browse(cr, uid, ids):
153 #            if not len(act.groups_id):
154 #                result.append(act.id)
155 #                continue
156 #            for g in act.groups_id:
157 #                if g.id in user_groups:
158 #                    result.append(act.id)
159 #                    break
160 #        return result
161
162     def _views_get_fnc(self, cr, uid, ids, name, arg, context={}):
163         res={}
164         for act in self.browse(cr, uid, ids):
165             res[act.id]=[(view.view_id.id, view.view_mode) for view in act.view_ids]
166             modes = act.view_mode.split(',')
167             if len(modes)>len(act.view_ids):
168                 find = False
169                 if act.view_id:
170                     res[act.id].append((act.view_id.id, act.view_id.type))
171                 for t in modes[len(act.view_ids):]:
172                     if act.view_id and (t == act.view_id.type) and not find:
173                         find = True
174                         continue
175                     res[act.id].append((False, t))
176         return res
177
178     _columns = {
179         'name': fields.char('Action Name', size=64, translate=True),
180         'type': fields.char('Action Type', size=32, required=True),
181         'view_id': fields.many2one('ir.ui.view', 'View Ref.', ondelete='cascade'),
182         'domain': fields.char('Domain Value', size=250),
183         'context': fields.char('Context Value', size=250),
184         'res_model': fields.char('Object', size=64),
185         'src_model': fields.char('Source Object', size=64),
186         'target': fields.selection([('current','Current Window'),('new','New Window')], 'Target Window'),
187         'view_type': fields.selection((('tree','Tree'),('form','Form')),string='Type of view'),
188         'view_mode': fields.char('Mode of view', size=250),
189         'usage': fields.char('Action Usage', size=32),
190         'view_ids': fields.one2many('ir.actions.act_window.view', 'act_window_id', 'Views'),
191         'views': fields.function(_views_get_fnc, method=True, type='binary', string='Views'),
192         'limit': fields.integer('Limit', help='Default limit for the list view'),
193         'auto_refresh': fields.integer('Auto-Refresh',
194             help='Add an auto-refresh on the view'),
195         'groups_id': fields.many2many('res.groups', 'ir_act_window_group_rel',
196             'act_id', 'gid', 'Groups'),
197     }
198     _defaults = {
199         'type': lambda *a: 'ir.actions.act_window',
200         'view_type': lambda *a: 'form',
201         'view_mode': lambda *a: 'tree,form',
202         'context': lambda *a: '{}',
203         'limit': lambda *a: 80,
204         'target': lambda *a: 'current',
205         'auto_refresh': lambda *a: 0,
206     }
207 act_window()
208
209 class act_window_view(osv.osv):
210     _name = 'ir.actions.act_window.view'
211     _table = 'ir_act_window_view'
212     _rec_name = 'view_id'
213     _columns = {
214         'sequence': fields.integer('Sequence'),
215         'view_id': fields.many2one('ir.ui.view', 'View'),
216         'view_mode': fields.selection((
217             ('tree', 'Tree'),
218             ('form', 'Form'),
219             ('graph', 'Graph'),
220             ('calendar', 'Calendar'),
221             ('gantt', 'Gantt')), string='Type of view', required=True),
222         'act_window_id': fields.many2one('ir.actions.act_window', 'Action', ondelete='cascade'),
223         'multi': fields.boolean('On multiple doc.',
224             help="If set to true, the action will not be displayed on the right toolbar of a form views."),
225     }
226     _defaults = {
227         'multi': lambda *a: False,
228     }
229     _order = 'sequence'
230 act_window_view()
231
232 class act_wizard(osv.osv):
233     _name = 'ir.actions.wizard'
234     _table = 'ir_act_wizard'
235     _sequence = 'ir_actions_id_seq'
236     _columns = {
237         'name': fields.char('Wizard info', size=64, required=True, translate=True),
238         'type': fields.char('Action type', size=32, required=True),
239         'wiz_name': fields.char('Wizard name', size=64, required=True),
240         'multi': fields.boolean('Action on multiple doc.', help="If set to true, the wizard will not be displayed on the right toolbar of a form views."),
241         'groups_id': fields.many2many('res.groups', 'res_groups_wizard_rel', 'uid', 'gid', 'Groups'),
242         'model': fields.char('Object', size=64),
243     }
244     _defaults = {
245         'type': lambda *a: 'ir.actions.wizard',
246         'multi': lambda *a: False,
247     }
248 act_wizard()
249
250 class act_url(osv.osv):
251     _name = 'ir.actions.url'
252     _table = 'ir_act_url'
253     _sequence = 'ir_actions_id_seq'
254     _columns = {
255         'name': fields.char('Action Name', size=64, translate=True),
256         'type': fields.char('Action Type', size=32, required=True),
257         'url': fields.text('Action Url',required=True),
258         'target': fields.selection((
259             ('new', 'New Window'),
260             ('self', 'This Window')),
261             'Action Target', required=True
262         )
263     }
264     _defaults = {
265         'type': lambda *a: 'ir.actions.act_url',
266         'target': lambda *a: 'new'
267     }
268 act_url()
269
270 def model_get(self, cr, uid, context={}):
271     wkf_pool = self.pool.get('workflow')
272     ids = wkf_pool.search(cr, uid, [])
273     osvs = wkf_pool.read(cr, uid, ids, ['osv'])
274
275     res = []
276     mpool = self.pool.get('ir.model')
277     for osv in osvs:
278         model = osv.get('osv')
279         id = mpool.search(cr, uid, [('model','=',model)])
280         name = mpool.read(cr, uid, id)[0]['name']
281         res.append((model, name))
282
283     return res
284
285 class ir_model_fields(osv.osv):
286     _inherit = 'ir.model.fields'
287     _rec_name = 'field_description'
288     _columns = {
289         'complete_name': fields.char('Complete Name', size=64, select=1),
290     }
291
292     def name_search(self, cr, uid, name, args=None, operator='ilike', context=None, limit=800):
293         def get_fields(cr, uid, field, rel):
294             result = []
295             mobj = self.pool.get('ir.model')
296             id = mobj.search(cr, uid, [('model','=',rel)])
297
298             obj = self.pool.get('ir.model.fields')
299             ids = obj.search(cr, uid, [('model_id','in',id)])
300             records = obj.read(cr, uid, ids)
301             for record in records:
302                 id = record['id']
303                 fld = field + '/' + record['name']
304
305                 result.append((id, fld))
306             return result
307
308         if not args:
309             args=[]
310         if not context:
311             context={}
312             return super(ir_model_fields, self).name_search(cr, uid, name, args, operator, context, limit)
313
314         if context.get('key') != 'server_action':
315             return super(ir_model_fields, self).name_search(cr, uid, name, args, operator, context, limit)
316
317         result = []
318         obj = self.pool.get('ir.model.fields')
319         ids = obj.search(cr, uid, args)
320         records = obj.read(cr, uid, ids)
321         for record in records:
322             id = record['id']
323             field = record['name']
324
325             if record['ttype'] == 'many2one':
326                 rel = record['relation']
327                 res = get_fields(cr, uid, field, record['relation'])
328                 for rs in res:
329                     result.append(rs)
330
331             result.append((id, field))
332
333         for rs in result:
334             obj.write(cr, uid, [rs[0]], {'complete_name':rs[1]})
335
336         iids = []
337         for rs in result:
338             iids.append(rs[0])
339
340         result = super(ir_model_fields, self).name_search(cr, uid, name, [('complete_name','ilike',name), ('id','in',iids)], operator, context, limit)
341
342         return result
343
344 ir_model_fields()
345
346 class server_object_lines(osv.osv):
347     _name = 'ir.server.object.lines'
348     _sequence = 'ir_actions_id_seq'
349     _columns = {
350         'server_id': fields.many2one('ir.actions.server', 'Object Mapping'),
351         'col1': fields.many2one('ir.model.fields', 'Destination', required=True),
352         'value': fields.text('Value', required=True),
353         'type': fields.selection([
354             ('value','Value'),
355             ('equation','Formula')
356         ], 'Type', required=True, size=32, change_default=True),
357     }
358     _defaults = {
359         'type': lambda *a: 'equation',
360     }
361 server_object_lines()
362
363 ##
364 # Actions that are run on the server side
365 #
366 class actions_server(osv.osv):
367     
368     def fields_view_get(self, cr, user, view_id=None, view_type='form', context=None, toolbar=False):
369         res = super(actions_server, self).fields_view_get(cr, user, view_id, view_type, context, toolbar)
370         #print 'RES : ',res
371         return res
372         
373     def _select_signals(self, cr, uid, context={}):
374         cr.execute("select distinct t.signal as key, t.signal || ' - [ ' || w.osv || ' ] ' as val from wkf w, wkf_activity a, wkf_transition t "\
375                         " where w.id = a.wkf_id " \
376                         " and t.act_from = a.wkf_id " \
377                         " or t.act_to = a.wkf_id and t.signal not in (null, NULL)")
378         result = cr.fetchall() or []
379         res = []
380         for rs in result:
381             if not rs[0] == None and not rs[1] == None:
382                 res.append(rs)
383         return res
384     
385     _name = 'ir.actions.server'
386     _table = 'ir_act_server'
387     _sequence = 'ir_actions_id_seq'
388     _columns = {
389         'name': fields.char('Action Name', required=True, size=64),
390         'state': fields.selection([
391             ('client_action','Client Action'),
392             ('dummy','Dummy'),
393             ('code','Python Code'),
394             ('trigger','Trigger'),
395             ('email','Email'),
396             ('sms','SMS'),
397             ('object_create','Create Object'),
398             ('object_write','Write Object'),
399             ('other','Multi Actions'),
400         ], 'Action State', required=True, size=32),
401         'code':fields.text('Python Code'),
402         'sequence': fields.integer('Sequence'),
403         'model_id': fields.many2one('ir.model', 'Object', required=True),
404         'action_id': fields.many2one('ir.actions.actions', 'Client Action'),
405         'trigger_name': fields.selection(_select_signals, string='Trigger Name', size=128),
406         'wkf_model_id': fields.many2one('ir.model', 'Workflow on'),
407         'trigger_obj_id': fields.many2one('ir.model.fields','Trigger On'),
408         'email': fields.many2one('ir.model.fields', 'Contact'),
409         'message': fields.text('Message', translate=True),
410         'mobile': fields.many2one('ir.model.fields', 'Contact'),
411         'sms': fields.char('SMS', size=160, translate=True),
412         'child_ids': fields.many2many('ir.actions.server', 'rel_server_actions', 'server_id', 'action_id', 'Others Actions'),
413         'usage': fields.char('Action Usage', size=32),
414         'type': fields.char('Action Type', size=32, required=True),
415         'srcmodel_id': fields.many2one('ir.model', 'Model', help="In which object you want to create / write the object if its empty refer to the Object field"),
416         'fields_lines': fields.one2many('ir.server.object.lines', 'server_id', 'Fields Mapping'),
417         'record_id':fields.many2one('ir.model.fields', 'Record Id', help="privide the field name from where the record id refers, if its empty it will refer to the active id of the object")
418     }
419     _defaults = {
420         'state': lambda *a: 'dummy',
421         'type': lambda *a: 'ir.actions.server',
422         'sequence': lambda *a: 5,
423         'code': lambda *a: """# You can use the following variables
424 #    - object
425 #    - object2
426 #    - time
427 #    - cr
428 #    - uid
429 #    - ids
430 # If you plan to return an action, assign: action = {...}
431 """,
432     }
433
434     
435     def get_email(self, cr, uid, action, context):
436         logger = netsvc.Logger()
437         obj_pool = self.pool.get(action.model_id.model)
438         id = context.get('active_id')
439         obj = obj_pool.browse(cr, uid, id)
440
441         fields = None
442
443         if '/' in action.email.complete_name:
444             fields = action.email.complete_name.split('/')
445         elif '.' in action.email.complete_name:
446             fields = action.email.complete_name.split('.')
447
448         for field in fields:
449             try:
450                 obj = getattr(obj, field)
451             except Exception,e :
452                 logger.notifyChannel('Workflow', netsvc.LOG_ERROR, 'Failed to parse : %s' % (field))
453
454         return obj
455
456
457     def get_mobile(self, cr, uid, action, context):
458         logger = netsvc.Logger()
459         obj_pool = self.pool.get(action.model_id.model)
460         id = context.get('active_id')
461         obj = obj_pool.browse(cr, uid, id)
462
463         fields = None
464
465         if '/' in action.mobile.complete_name:
466             fields = action.mobile.complete_name.split('/')
467         elif '.' in action.mobile.complete_name:
468             fields = action.mobile.complete_name.split('.')
469
470         for field in fields:
471             try:
472                 obj = getattr(obj, field)
473             except Exception,e :
474                 logger.notifyChannel('Workflow', netsvc.LOG_ERROR, 'Failed to parse : %s' % (field))
475
476         return obj
477
478     def merge_message(self, cr, uid, keystr, action, context):
479         logger = netsvc.Logger()
480         def merge(match):
481             obj_pool = self.pool.get(action.model_id.model)
482             id = context.get('active_id')
483             obj = obj_pool.browse(cr, uid, id)
484             exp = str(match.group()[2:-2]).strip()
485             result = eval(exp, {'object':obj, 'context': context,'time':time})
486             if result in (None, False):
487                 return str("--------")
488             return str(result)
489         
490         com = re.compile('(\[\[.+?\]\])')
491         message = com.sub(merge, keystr)
492         
493         return message
494
495     # Context should contains:
496     #   ids : original ids
497     #   id  : current id of the object
498     # OUT:
499     #   False : Finnished correctly
500     #   ACTION_ID : Action to launch
501     def run(self, cr, uid, ids, context={}):
502         logger = netsvc.Logger()
503         for action in self.browse(cr, uid, ids, context):
504             
505             if action.state=='client_action':
506                 if not action.action_id:
507                     raise osv.except_osv(_('Error'), _("Please specify an action to launch !"))
508                 return self.pool.get(action.action_id.type).read(cr, uid, action.action_id.id, context=context)
509             
510             if action.state=='python':
511                 localdict = {
512                     'self': self.pool.get(action.model_id.model),
513                     'context': context,
514                     'time': time,
515                     'ids': ids,
516                     'cr': cr,
517                     'uid': uid
518                 }
519                 exec action.code in localdict
520                 if 'action' in localdict:
521                     return localdict['action']
522
523             if action.state == 'email':
524                 user = config['email_from']
525                 subject = action.name
526                 address = self.get_email(cr, uid, action, context)
527                 if not address:
528                     raise osv.except_osv(_('Error'), _("Please specify the Partner Email address !"))
529                 if not user:
530                     raise osv.except_osv(_('Error'), _("Please specify server option --smtp-from !"))
531                 
532                 body = self.merge_message(cr, uid, str(action.message), action, context)
533                 if tools.email_send(user, [address], subject, body, debug=False, subtype='html') == True:
534                     logger.notifyChannel('email', netsvc.LOG_INFO, 'Email successfully send to : %s' % (address))
535                 else:
536                     logger.notifyChannel('email', netsvc.LOG_ERROR, 'Failed to send email to : %s' % (address))
537
538             if action.state == 'trigger':
539                 wf_service = netsvc.LocalService("workflow")
540                 model = action.wkf_model_id.model
541                 obj_pool = self.pool.get(action.model_id.model)
542                 res_id = self.pool.get(action.model_id.model).read(cr, uid, [context.get('active_id')], [action.trigger_obj_id.name])
543                 id = res_id [0][action.trigger_obj_id.name]
544                 wf_service.trg_validate(uid, model, int(id), action.trigger_name, cr)
545
546             if action.state == 'sms':
547                 #TODO: set the user and password from the system
548                 # for the sms gateway user / password
549                 api_id = ''
550                 text = action.sms
551                 to = self.get_mobile(cr, uid, action, context)
552                 #TODO: Apply message mearge with the field
553                 if tools.sms_send(user, password, api_id, text, to) == True:
554                     logger.notifyChannel('sms', netsvc.LOG_INFO, 'SMS successfully send to : %s' % (action.address))
555                 else:
556                     logger.notifyChannel('sms', netsvc.LOG_ERROR, 'Failed to send SMS to : %s' % (action.address))
557             
558             if action.state == 'other':
559                 res = None
560                 for act in action.child_ids:
561                     result = self.run(cr, uid, [act.id], context)
562                     if result:
563                         res = result
564                 return res
565             
566             if action.state == 'object_write':
567                 res = {}
568                 for exp in action.fields_lines:
569                     euq = exp.value
570                     if exp.type == 'equation':
571                         obj_pool = self.pool.get(action.model_id.model)
572                         obj = obj_pool.browse(cr, uid, context['active_id'], context=context)
573                         expr = eval(euq, {'context':context, 'object': obj, 'time':time})
574                     else:
575                         expr = exp.value
576                     res[exp.col1.name] = expr
577
578                 if not action.record_id:
579                     if not action.srcmodel_id:
580                         obj_pool = self.pool.get(action.model_id.model)
581                         obj_pool.write(cr, uid, [context.get('active_id')], res)
582                     else:
583                         obj_pool = self.pool.get(action.srcmodel_id.model)
584                         obj_pool.write(cr, uid, [context.get('active_id')], res)
585                 else:
586                     obj_pool = self.pool.get(action.srcmodel_id.model)
587                     id = self.pool.get(action.model_id.model).read(cr, uid, [context.get('active_id')], [action.record_id.name])
588                     obj_pool.write(cr, uid, [int(id[0][action.record_id.name])], res)
589             
590             if action.state == 'object_create':
591                 res = {}
592                 for exp in action.fields_lines:
593                     euq = exp.value
594                     if exp.type == 'equation':
595                         obj_pool = self.pool.get(action.model_id.model)
596                         obj = obj_pool.browse(cr, uid, context['active_id'], context=context)
597                         expr = eval(euq, {'context':context, 'object': obj, 'time':time})
598                     else:
599                         expr = exp.value
600                     res[exp.col1.name] = expr
601
602                 obj_pool = None
603                 res_id = False
604                 obj_pool = self.pool.get(action.srcmodel_id.model)
605                 res_id = obj_pool.create(cr, uid, res)
606                 self.pool.get(action.model_id.model).write(cr, uid, [context.get('active_id')], {action.record_id.name:res_id})
607
608         return False
609 actions_server()
610
611 class act_window_close(osv.osv):
612     _name = 'ir.actions.act_window_close'
613     _table = 'ir_actions'
614     _sequence = 'ir_actions_id_seq'
615     _columns = {
616         'name': fields.char('Action Name', size=64, translate=True),
617         'type': fields.char('Action Type', size=32, required=True),
618     }
619     _defaults = {
620         'type': lambda *a: 'ir.actions.act_window_close',
621     }
622 act_window_close()
623
624 # This model use to register action services.
625 # if action type is 'configure', it will be start on configuration wizard.
626 # if action type is 'service',
627 #                - if start_type= 'at once', it will be start at one time on start date
628 #                - if start_type='auto', it will be start on auto starting from start date, and stop on stop date
629 #                - if start_type="manual", it will start and stop on manually 
630 class ir_actions_todo(osv.osv):
631     _name = 'ir.actions.todo'    
632     _columns={
633         'name':fields.char('Name',size=64,required=True, select=True),
634         'note':fields.text('Text', translate=True),
635         'start_date': fields.datetime('Start Date'),
636         'end_date': fields.datetime('End Date'),
637         'action_id':fields.many2one('ir.actions.act_window', 'Action', select=True,required=True, ondelete='cascade'),
638         'sequence':fields.integer('Sequence'),
639         'active': fields.boolean('Active'),
640         'type':fields.selection([('configure', 'Configure'),('service', 'Service'),('other','Other')], string='Type', required=True),
641         'start_on':fields.selection([('at_once', 'At Once'),('auto', 'Auto'),('manual','Manual')], string='Start On'),
642         'groups_id': fields.many2many('res.groups', 'res_groups_act_todo_rel', 'act_todo_id', 'group_id', 'Groups'),
643         'users_id': fields.many2many('res.users', 'res_users_act_todo_rel', 'act_todo_id', 'user_id', 'Users'),
644         'state':fields.selection([('open', 'Not Started'),('done', 'Done'),('skip','Skipped'),('cancel','Cancel')], string='State', required=True)
645     }
646     _defaults={
647         'state': lambda *a: 'open',
648         'sequence': lambda *a: 10,
649         'active':lambda *a:True,
650         'type':lambda *a:'configure'
651     }
652     _order="sequence"
653 ir_actions_todo()
654
655 # This model to use run all configuration actions
656 class ir_actions_configuration_wizard(osv.osv_memory):
657     _name='ir.actions.configuration.wizard'
658     def next_configuration_action(self,cr,uid,context={}):
659         item_obj = self.pool.get('ir.actions.todo')
660         item_ids = item_obj.search(cr, uid, [('type','=','configure'),('state', '=', 'open'),('active','=',True)], limit=1, context=context)
661         if item_ids and len(item_ids):
662             item = item_obj.browse(cr, uid, item_ids[0], context=context)
663             return item
664         return False
665     def _get_action_name(self, cr, uid, context={}):
666         next_action=self.next_configuration_action(cr,uid,context=context)        
667         if next_action:
668             return next_action.note
669         else:
670             return "Your database is now fully configured.\n\nClick 'Continue' and enjoy your OpenERP experience..."
671         return False
672
673     def _get_action(self, cr, uid, context={}):
674         next_action=self.next_configuration_action(cr,uid,context=context)
675         if next_action:           
676             return next_action.id
677         return False
678
679     def _progress_get(self,cr,uid, context={}):
680         total = self.pool.get('ir.actions.todo').search_count(cr, uid, [], context)
681         todo = self.pool.get('ir.actions.todo').search_count(cr, uid, [('type','=','configure'),('active','=',True),('state','<>','open')], context)
682         return max(5.0,round(todo*100/total))
683
684     _columns = {
685         'name': fields.text('Next Wizard',readonly=True),
686         'progress': fields.float('Configuration Progress', readonly=True),
687         'item_id':fields.many2one('ir.actions.todo', 'Next Configuration Wizard',invisible=True, readonly=True),
688     }
689     _defaults={
690         'progress': _progress_get,
691         'item_id':_get_action,
692         'name':_get_action_name,
693     }
694     def button_next(self,cr,uid,ids,context=None):
695         user_action=self.pool.get('res.users').browse(cr,uid,uid)
696         act_obj=self.pool.get(user_action.menu_id.type)
697         action_ids=act_obj.search(cr,uid,[('name','=',user_action.menu_id.name)])
698         action_open=act_obj.browse(cr,uid,action_ids)[0]
699         if context.get('menu',False):
700             return{
701                 'view_type': action_open.view_type,
702                 'view_id':action_open.view_id and [action_open.view_id.id] or False,
703                 'res_model': action_open.res_model,
704                 'type': action_open.type,
705                 'domain':action_open.domain
706             }
707         return {'type':'ir.actions.act_window_close'}
708
709     def button_skip(self,cr,uid,ids,context=None):
710         item_obj = self.pool.get('ir.actions.todo')
711         item_id=self.read(cr,uid,ids)[0]['item_id']
712         if item_id:
713             item = item_obj.browse(cr, uid, item_id, context=context)
714             item_obj.write(cr, uid, item.id, {
715                 'state': 'skip',
716                 }, context=context)
717             return{
718                 'view_type': 'form',
719                 "view_mode": 'form',
720                 'res_model': 'ir.actions.configuration.wizard',
721                 'type': 'ir.actions.act_window',
722                 'target':'new',
723             }
724         return self.button_next(cr, uid, ids, context)
725
726     def button_continue(self, cr, uid, ids, context=None):
727         item_obj = self.pool.get('ir.actions.todo')
728         item_id=self.read(cr,uid,ids)[0]['item_id']
729         if item_id:
730             item = item_obj.browse(cr, uid, item_id, context=context)
731             item_obj.write(cr, uid, item.id, {
732                 'state': 'done',
733                 }, context=context)
734             return{
735                   'view_mode': item.action_id.view_mode,
736                   'view_type': item.action_id.view_type,
737                   'view_id':item.action_id.view_id and [item.action_id.view_id.id] or False,
738                   'res_model': item.action_id.res_model,
739                   'type': item.action_id.type,
740                   'target':item.action_id.target,
741             }
742         return self.button_next(cr, uid, ids, context)
743 ir_actions_configuration_wizard()
744
745 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
746