GOLEM/golem_family_membership/wizard/golem_membership_invoice.py

# -*- coding: utf-8 -*-

#    Copyright 2018 Youssef El Ouahby <youssef@yaltik.com>
#    Copyright 2018 Fabien Bourgeois <fabien@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 Membership Invoice Adaptations"""

from odoo import models, fields, api

class GolemMembershipInvoice(models.TransientModel):
    """ GOLEM Membership Invoice adaptations """
    _inherit = 'golem.membership.invoice'

    family_id = fields.Many2one('golem.family', string='Family',
                                required=True, ondelete='cascade')
    member_ids = fields.Many2many('res.partner', string='Concerned members')
    on_the_name_of = fields.Many2one('res.partner', ondelete='cascade',
                                     required=True)

    @api.onchange('family_id')
    def onchange_family(self):
        """ Fill member_ids """
        for record in self:
            if record.family_id and record.family_id.member_ids:
                record.member_ids = [(6, False,
                                      record.family_id.member_ids.ids)]

    @api.onchange('member_ids')
    def onchange_members(self):
        """ On change members : custom domain for on the name of """
        record = self[0]
        if record.member_ids:
            return {
                'domain': {'on_the_name_of': [('id', 'in', record.member_ids.ids)]}
            }
        return {'domain': {'on_the_name_of': []}}

    @api.multi
    def membership_family_invoice(self):
        """ Create family membership """
        self.ensure_one()
        record = self[0]
        datas = {'membership_product_id': record.product_id.id,
                 'amount': record.member_price}
        invoice_list = record.on_the_name_of.create_membership_invoice(datas=datas)
        # Link membership lines to family and targetted members
        membership_line = record.on_the_name_of.member_lines[0] # Last one
        membership_line.family_id = record.family_id
        # Add membership lines for other family members
        for member in record.member_ids:
            if member != record.on_the_name_of:
                membership_line.copy({'partner': member.id})

        search_view_id = self.env.ref('account.view_account_invoice_filter')
        form_view_id = self.env.ref('account.invoice_form')
        return {
            'domain': [('id', 'in', invoice_list)],
            'name': 'Membership Invoices',
            'view_type': 'form',
            'view_mode': 'tree,form',
            'res_model': 'account.invoice',
            'type': 'ir.actions.act_window',
            'views': [(False, 'tree'), (form_view_id.id, 'form')],
            'search_view_id': search_view_id.id
        }