# -*- coding: utf-8 -*-
##############################################################################
#
#    datenpol gmbh
#    Copyright (C) 2013-TODAY datenpol gmbh (<http://www.datenpol.at/>)
#
#    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/>.
#
##############################################################################
from odoo import api, fields, models, _
from odoo.exceptions import UserError


class AccountInvoiceLine(models.Model):
    _inherit = 'account.invoice.line'

    intrastat_id = fields.Many2one(comodel_name='report.intrastat.code', string='Intrastat Code',
                                   compute="_compute_intrastat_id", inverse='_set_intrastat_id')
    lot_id = fields.Many2one(comodel_name='stock.production.lot', string='Lot')
    weight = fields.Float(string='Gewicht', compute='_compute_weight')

    @api.model
    def create(self, vals):
        if vals.get('product_id', False) and not vals.get('intrastat_id', False):
            vals.update(intrastat_id=self.env['product.template'].browse(vals['product_id']).intrastat_id.id)
        return super(AccountInvoiceLine, self).create(vals)

    @api.multi
    def action_show_lot(self):
        self.ensure_one()
        action = self.env.ref('stock.action_production_lot_form').read()[0]
        action['res_id'] = self.lot_id.id
        action['view_mode'] = 'form'
        action['views'] = [(False, 'form')]

        return action

    def _compute_weight(self):
        for record in self:
            record.weight = record.lot_id.weight or record.product_id.weight

    def _compute_intrastat_id(self):
        for record in self:
            record.intrastat_id = record.lot_id.intrastat_id.id or record.product_id.intrastat_id.id

    def _set_intrastat_id(self):
        for record in self:
            if record.lot_id:
                record.lot_id.intrastat_id = record.intrastat_id.id
            else:
                raise UserError(_('Der Intrastrat Code kann nur gesetzt werden wenn ein Lot angegeben wurde.'))


class AccountPaymentTerm(models.Model):
    _inherit = 'account.payment.term'

    code = fields.Char(string='Code')


class AccountInvoice(models.Model):
    _inherit = 'account.invoice'

    positions = fields.Integer(string='Positionen', compute='_compute_positions')
    num_items = fields.Integer(string='Anzahl der Artikel', compute='_compute_num_items')
    weight_total = fields.Float(string='Gesamtgewicht', compute='_compute_weight_total')

    @api.multi
    def _compute_weight_total(self):
        for record in self:
            _sum = 0
            for line in record.invoice_line_ids:
                _sum += line.weight * line.quantity
            record.weight_total = _sum

    @api.multi
    def _compute_positions(self):
        for record in self:
            record.positions = len(record.invoice_line_ids)

    @api.multi
    def _compute_num_items(self):
        for record in self:
            num_items = 0
            for line in record.invoice_line_ids:
                if line.uom_id == self.env.ref('product.product_uom_unit'):  # wenn die Mengeneinheit Stk. ist
                    num_items += line.quantity
            record.num_items = num_items

    @api.multi
    def invoice_print(self):
        self.ensure_one()
        self.sent = True
        return self.env.ref('account.account_invoices_without_payment').report_action(self)