GOLEM/golem_activity_registration/wizard/golem_activity_registration_analysis_wizard.py

# -*- coding: utf-8 -*-
#
#    Copyright 2019 Fabien Bourgeois <fabien@yaltik.com>
#    Copyright 2019 Youssef El Ouahby <youssef@yaltik.com>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero 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 Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.

""" GOLEM Activity Registration Analysis Wizard Management """

import logging
from odoo import models, fields, api, _
_LOGGER = logging.getLogger(__name__)


class GolemActivityRegistrationAnalysisWizard(models.TransientModel):
    """ GOLEM Activity Registration Analysis Wizard Management  """
    _name = 'golem.activity.registration.analysis.wizard'
    _description = 'GOLEM Activity Registration Analysis Wizard Management '

    def default_season(self):
        """ Get default season """
        domain = [('is_default', '=', True)]
        return self.env['golem.season'].search(domain, limit=1)

    name = fields.Char(default='/')
    season_id = fields.Many2one('golem.season', 'Season', required=True,
                                default=default_season, ondelete='cascade')
    season_start = fields.Date(related='season_id.date_start')
    season_end = fields.Date(related='season_id.date_end')
    analyse_type = fields.Selection([('activity_area', 'By area and activity')],
                                    default='activity_area', required=True)
    age_computing = fields.Selection([('season_start', 'At season start'),
                                      ('season_end', 'At season stop'),
                                      ('now', 'Now')], required=True,
                                     default='season_start', string='Age')

    @api.multi
    def get_pivot_view(self):
        """ get pivot view """
        self.ensure_one()
        rec = self[0]
        ref = u''
        if rec.age_computing == 'season_start':
            if rec.analyse_type == 'activity_area':
                ref = (u'golem_activity_registration.'
                       'golem_member_analysis_pivot_age_start_activity_area')
        elif rec.age_computing == 'season_end':
            if rec.analyse_type == 'activity_area':
                ref = (u'golem_activity_registration.'
                       'golem_member_analysis_pivot_age_end_activity_area')
        else: # now
            if rec.analyse_type == 'activity_area':
                ref = (u'golem_activity_registration.'
                       'golem_member_analysis_pivot_age_now_activity_area')
        return self.env.ref(ref, False)

    @api.multi
    def get_registrations(self):
        """ Returns registrations """
        self.ensure_one()
        return self.env['golem.activity.registration'].search(
            [('season_id', '=', self[0].season_id.id)]
        )


    @api.multi
    def registration_analyse(self):
        """ registration analyse """
        self.ensure_one()
        analysis_rec = self[0]
        registrations = analysis_rec.get_registrations()
        if registrations:
            for registration in registrations:
                self.env['golem.member.analysis'].create({
                    'season_id': analysis_rec.season_id.id,
                    'member_id': registration.member_id.id,
                    'activity_id': registration.activity_id.id,
                    'resgistration_analysis_id': analysis_rec.id
                    })
        analyse_pivot = analysis_rec.get_pivot_view()
        return {
            'name': _('Registration Analysis'),
            'type': 'ir.actions.act_window',
            'res_model': 'golem.member.analysis',
            'domain': [('resgistration_analysis_id', '=', analysis_rec.id)],
            'view_type': 'pivot',
            'view_mode': 'pivot',
            'view_id': analyse_pivot.id,
            'target': 'current',
        }