golem/golem_resource_account/wizard/golem_reservation_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 Reservation Invoice Wizard"""

from odoo import models, fields, api, _
from odoo.exceptions import UserError


class GolemReservationInvoiceWizard(models.TransientModel):
    """ GOLEM Resource Reservation Invoice Wizard """
    _name = 'golem.reservation.invoice.wizard'

    reservation_ids = fields.Many2many(
        'golem.resource.reservation', required=True, string='Reservations to invoice',
        default=lambda self: self._context.get('active_ids', []))

    @api.multi
    def create_invoices(self):
        """ Invoice creations """
        self.ensure_one()

        inv_obj = self.env['account.invoice']
        partner_id = self.reservation_ids[0].partner_id
        product = self.reservation_ids[0].resource_id.product_tmpl_id

        if product.id:
            account_id = product.property_account_income_id.id
        if not account_id:
            account_id = product.categ_id.property_account_income_categ_id.id
        if not account_id:
            raise UserError(
                _('There is no income account defined for this product: "%s". \
                   You may have to install a chart of account from Accounting \
                   app, settings menu.') % (product.name,))

        lines = []

        for reservation in self.reservation_ids:
            product = reservation.resource_id.product_tmpl_id
            amount = product.standard_price
            lines.append((0, 0, {
                'origin': reservation.resource_id.name,
                'account_id': account_id,
                'price_unit': amount,
                'quantity': 1.0,
                'uom_id': product.uom_id.id,
                'product_id': product.id,
                }))
        invoice = inv_obj.create({
            'origin': self.reservation_ids[-1].name,
            'type': 'out_invoice',
            'reference': False,
            'account_id': partner_id.property_account_receivable_id.id,
            'partner_id': partner_id.id,
            'invoice_line_ids': lines,
            })
        self.reservation_ids.write({'invoice_id': invoice.id})