from odoo import models, api, fields
from odoo.exceptions import AccessError, UserError, ValidationError
import logging

_logger = logging.getLogger(__name__)


class DunningRun(models.Model):
    _name = 'dunning.run'
    _description = 'Dunning Run'
    _order = 'dunning_date desc, id desc'

    number = fields.Char(string='Document No', index=True, copy=False, readonly=True)
    dunning_date = fields.Date(string='Dunning Date', required=True)
    partner_id = fields.Many2one(
        'res.partner', string='Customer',
        required=True, change_default=True, index=True, tracking=1)
    dunning_line = fields.One2many('dunning.run.line', 'dunning_id', string='Dunning Lines', auto_join=True)
    # dunning_level = fields.Integer(string='Dunning Level', default=30, help='30 hari sebelum jatuh tempo invoice')

    def generate_dunning_line(self):
        if self.partner_id.parent_id:
            raise UserError('Harus pilih parent company')

        partners = []
        partners += self.partner_id.child_ids
        partners.append(self.partner_id)

        for partner in partners:
            query = [
                ('move_type', '=', 'out_invoice'),
                ('partner_id', '=', partner.id),
                ('outstanding_amount', '>', 0),
            ]
            invoices = self.env['account.move'].search(query, order='invoice_date')
            for invoice in invoices:
                parameter_line = {
                    'dunning_id': self.id,
                    'partner_id': invoice.partner_id,
                    'invoice_id': invoice.id,
                    'date_invoice': invoice.invoice_date,
                    'efaktur_id': invoice.efaktur_id,
                    'reference': invoice.ref,
                    'open_amt': invoice.outstanding_amount
                }
            self.env['dunning.run.line'].create([parameter_line])

    @api.model
    def create(self, vals):
        vals['number'] = self.env['ir.sequence'].next_by_code('dunning.run') or '0'
        result = super(DunningRun, self).create(vals)
        return result

    def generate_dunning_line(self):
        # validation
        if not self.partner_id:
            raise UserError('Customer harus diisi')

        invoices = self.env['account.move'].search([
            ('amount_residual_signed', '>', 0),
            ('partner_id', '=', self.partner_id.id),
            ('move_type', '=', 'out_invoice'),
            ('state', '=', 'posted'),
        ])
        count = 0
        for invoice in invoices:
            self.env['dunning.run.line'].create([{
                'dunning_id': self.id,
                'partner_id': invoice.partner_id.id,
                'invoice_id': invoice.id,
                'date_invoice': invoice.invoice_date,
                'efaktur_id': invoice.efaktur_id.id,
                'reference': invoice.ref,
                'total_amt': invoice.amount_total,
                'open_amt': invoice.amount_residual_signed,
                'due_date': invoice.invoice_date_due
            }])
            count += 1
        _logger.info("Dunning Line generated %s" % count)


class DunningRunLine(models.Model):
    _name = 'dunning.run.line'
    _description = 'Dunning Run Line'
    _order = 'dunning_id, id'

    dunning_id = fields.Many2one('dunning.run', string='Dunning Ref', required=True, ondelete='cascade', index=True, copy=False)
    partner_id = fields.Many2one('res.partner', string='Customer')
    invoice_id = fields.Many2one('account.move', string='Invoice')
    date_invoice = fields.Date(string='Invoice Date')
    # due_date = fields.Date(string='Due Date')
    efaktur_id = fields.Many2one('vit.efaktur', string='Faktur Pajak')
    reference = fields.Char(string='Reference')
    total_amt = fields.Float(string='Total Amount')
    open_amt = fields.Float(string='Open Amount')
    due_date = fields.Date(string='Due Date')