~credativ/openobject-addons/addons-extra-syrup

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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# -*- encoding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
#    $Id$
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

from osv import osv
from osv import fields
import pooler
import time
import tools
from tools.translate import _

class training_session_create_wizard(osv.osv_memory):
    _name = 'training.session.create.wizard'

    _columns = {
        'date': fields.datetime('Date', required=True),
        'name': fields.char('Name', size=64, required=True),
    }

    def _default_get_name(self, cr, uid, context=None):
        if context is None:
            return ''
        active_id = context.get('active_id')
        if not active_id:
            return ''
        offer = self.pool.get('training.offer').browse(cr, uid, active_id, context=context)
        return offer.name

    _defaults = {
        'date': lambda *a: time.strftime('%Y-%m-%d 08:30:00'),
        'name': _default_get_name,
    }

    def create_session(self, cr, uid, ids, context=None):
        session_ids = []
        pool = pooler.get_pool(cr.dbname)
        proxy = pool.get('training.offer')
        offer = proxy.browse(cr, uid, context['active_id'], context=context)
        for form in self.read(cr, uid, ids, [], context=context):
            if not offer.can_be_planned:
                raise osv.except_osv(_('Warning'), _("You can not create a session with a non-validated offer") )

            proxy = pool.get('training.session')
            session_id = proxy.create(cr, uid,
                                      {
                                          'offer_id' : form['id'],
                                          'date' : form['date'],
                                          'name' : "%s (%s)" % (form['name'], time.strftime("%Y-%m-%d", time.strptime(form['date'], '%Y-%m-%d %H:%M:%S'))),
                                          'format_id' : offer.format_id.id,
                                      }
                                     )
            vals = proxy.on_change_offer(cr, uid, None, context['active_id'])
            if isinstance(vals, dict):
                # we have specified a custom name on session creation,
                # don't let on_change overwrite it now.
                try:
                    if 'name' in vals['value']:
                        del vals['value']['name']
                except KeyError:
                    pass
                proxy.write(cr, uid, [session_id], vals['value'], context=context)
            session_ids.append(session_id)
        number_of_sessions = len(session_ids)
        if not number_of_sessions:
            return {}
        elif number_of_sessions == 1:
            return {
                'res_id': int(session_ids[0]),
                'name': 'Sessions',
                'view_type': 'form',
                "view_mode": 'form,tree',
                'res_model': 'training.session',
                'view_id': pool.get('ir.ui.view').search(cr,uid,[('name','=','training.session.form')]),
                'type': 'ir.actions.act_window',
                'target': 'current',
            }
        else:
            return {
                'domain' : "[('id', 'in', [%s])]" % ','.join(map(str,session_ids)),
                'name' : 'Sessions',
                'view_type' : 'form',
                'view_mode' : 'tree,form',
                'res_model' : 'training.session',
                'view_id' : False,
                'type' : 'ir.actions.act_window',
                'target': 'current',
            }

training_session_create_wizard()