# -*- 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
from odoo.exceptions import ValidationError


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

    intrastat_id = fields.Many2one(comodel_name='report.intrastat.code', string='Intrastat Code')
    lot_id = fields.Many2one(comodel_name='stock.production.lot', string='Lot')
    weight = fields.Float(string='Gewicht', compute='_compute_weight')
    hide_intrastat_code = fields.Boolean('ZV', help='Zolltarifnummer verbergen')
    dealer_discount = fields.Boolean(string='Händlerrabatt')

    @api.model
    def create(self, vals):
        if not vals.get('intrastat_id', False):
            if vals.get('lot_id', False):
                vals.update(intrastat_id=self.env['stock.production.lot'].browse(vals['lot_id']).intrastat_id.id)
            elif vals.get('product_id', False):
                vals.update(intrastat_id=self.env['product.product'].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 write(self, vals):
        res = super(AccountInvoiceLine, self).write(vals)
        for record in self:
            lot_id = record.lot_id if 'lot_id' not in vals else vals.get('lot_id', False)
            if vals.get('intrastat_id', False) and lot_id:
                self.env['stock.production.lot'].browse([lot_id.id]).write({
                    'intrastat_id': vals.get('intrastat_id')
                })
            # elif vals.get('intrastat_id', False) and not lot_id:
            #     raise UserError(_('Der Intrastrat Code kann nur gesetzt werden wenn ein Lot angegeben wurde.'))
        return res


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')

    layout_category_id = fields.Many2one('sale.layout_category', related='invoice_line_ids.layout_category_id', string='Section')

    editor_id = fields.Many2one('res.users', string='Rechnungsbearbeiter',
                                default=lambda self: self.env.user if self.env.user.editor_name else '',
                                domain=[('editor_name', '!=', '')],track_visibility='onchange')


    @api.multi
    @api.onchange('partner_shipping_id')
    def onchange_partner_shipping_id(self):
#        res = super(AccountInvoice, self).onchange_partner_shipping_id()
        if not self.env.user.has_group('dp_custom.group_allow_third_country_sale'):
            if self.partner_id.country_id != self.partner_shipping_id.country_id:
                message = _('Rechnungs- und Lieferland passen nicht zusammen: %s <-> %s!') % \
                          (self.partner_id.country_id.name, self.partner_shipping_id.country_id.name)

                raise ValidationError(message)

    @api.multi
    def action_invoice_open(self):
        to_open_invoices = self.filtered(lambda inv: inv.state != 'open')
        if to_open_invoices.filtered(lambda inv: inv.amount_total != 0) or self.env.context.get('confirmed', False):
            res = super(AccountInvoice, self).action_invoice_open()
            return res

        action = self.env.ref('dp_custom.action_wizard_confirm_null_invoice').read()[0]
        action['context'] = str({"invoice_ids": self.env.context.get("active_ids")})
        return action

    @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 if not line.dealer_discount else 0.0
            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)

    @api.model
    def _formatLang(self, value, currency=True):
        lang = self.partner_id.lang
        lang_objs = self.env['res.lang'].search([('code', '=', lang)])
        if not lang_objs:
            lang_objs = self.env['res.lang'].search([], limit=1)
        lang_obj = lang_objs[0]

        res = lang_obj.format('%.' + str(2) + 'f', value, grouping=True, monetary=True)
        currency_obj = self.currency_id

        if currency_obj and currency_obj.symbol and currency:
            if currency_obj.position == 'after':
                res = '%s %s' % (res, currency_obj.symbol)
            elif currency_obj and currency_obj.position == 'before':
                res = '%s %s' % (currency_obj.symbol, res)
        return res

    @api.multi
    def print_invoice_short(self):
        self.ensure_one()
        return self.env.ref('dp_reports_account.account_invoices_short').report_action(self)

    @api.multi
    def print_intrastat_invoice(self):
        self.ensure_one()
        info = "Bei folgenden Produkten fehlt die Zolltarifnummer:\n"
        intrastrat_not_set = False
        position = 1
        for invoice_line in self.invoice_line_ids:
            if invoice_line.product_id and invoice_line.product_id.type != 'service':
                if not invoice_line.intrastat_id and not invoice_line.hide_intrastat_code:
                    info += "Zeile: "+str(position)
                    if invoice_line.product_id.default_code:
                        info += "\tProduktcode: " + invoice_line.product_id.default_code
                    info += "\n"
                    intrastrat_not_set = True
            position += 1
        if intrastrat_not_set:
            action = self.env.ref('dp_custom.action_wizard_confirm_print_invoice').read()[0]
            action['context'] = str({"default_info": info, "invoice_ids": self.env.context.get("active_ids")})
            return action
        return self.env.ref('dp_reports_account.account_invoices_with_intrastat').report_action(self)

    @api.multi
    def print_intrastat_invoice_short(self):
        self.ensure_one()
        info = "Bei folgenden Produkten fehlt die Zolltarifnummer:\n"
        intrastrat_not_set = False
        position = 1
        for invoice_line in self.invoice_line_ids:
            if invoice_line.product_id and invoice_line.product_id.type != 'service':
                if not invoice_line.intrastat_id and not invoice_line.hide_intrastat_code:
                    info += "Zeile: "+str(position)
                    if invoice_line.product_id.default_code:
                        info += "\tProduktcode: " + invoice_line.product_id.default_code
                    info += "\n"
                    intrastrat_not_set = True
            position += 1
        if intrastrat_not_set:
            action = self.env.ref('dp_custom.action_wizard_confirm_print_invoice_short').read()[0]
            action['context'] = str({"default_info": info, "invoice_ids": self.env.context.get("active_ids")})
            return action
        return self.env.ref('dp_reports_account.account_invoices_with_intrastat_short').report_action(self)