# coding: utf-8
from .extra_timezones import TIEMPO_DEL_CENTRO_ZIPCODES, TIEMPO_DEL_CENTRO_EN_FRONTIERA_ZIPCODES

from pytz import timezone

from odoo import api, fields, models, _
from odoo.addons.l10n_mx_edi.models.res_company import FISCAL_REGIMES_SELECTION
from odoo.addons.l10n_mx_edi.models.l10n_mx_edi_document import USAGE_SELECTION


class ResPartner(models.Model):
    _inherit = 'res.partner'

    l10n_mx_edi_addenda_ids = fields.Many2many(
        comodel_name='l10n_mx_edi.addenda',
        string='Addendas & Complementos',
        copy=False,
    )
    l10n_mx_edi_fiscal_regime = fields.Selection(
        selection=FISCAL_REGIMES_SELECTION,
        string="Fiscal Regime",
        compute="_compute_l10n_mx_edi_fiscal_regime",
        readonly=False,
        store=True,
        help="Fiscal Regime is required for all partners (used in CFDI)")
    l10n_mx_edi_ieps_breakdown = fields.Boolean(string="IEPS Breakdown")
    l10n_mx_edi_usage = fields.Selection(
        selection=USAGE_SELECTION,
        string="Usage",
        help="The code that corresponds to the use that will be made of the receipt by the recipient.",
    )
    l10n_mx_edi_payment_method_id = fields.Many2one(
        comodel_name='l10n_mx_edi.payment.method',
        string="Payment Way",
        help="Indicates the way the invoice was/will be paid, where the options could be: "
             "Cash, Nominal Check, Credit Card, etc. Leave empty if unkown and the XML will show 'Unidentified'.",
    )
    l10n_mx_edi_payment_policy = fields.Selection(
        string="Payment Policy",
        selection=[
            ('PPD', 'PPD'),
            ('PUE', 'PUE'),
        ],
    )
    l10n_mx_edi_partner_address_complete = fields.Boolean(compute='_compute_l10n_mx_edi_partner_address_complete')

    @api.depends('country_code')
    def _compute_l10n_mx_edi_fiscal_regime(self):
        for partner in self:
            if not partner.country_code:
                partner.l10n_mx_edi_fiscal_regime = None
            elif partner.country_code != 'MX':
                partner.l10n_mx_edi_fiscal_regime = '616'
            elif not partner.l10n_mx_edi_fiscal_regime:
                partner.l10n_mx_edi_fiscal_regime = '601'

    def _l10n_mx_edi_get_cfdi_timezone(self):
        self.ensure_one()
        code = self.state_id.code
        zipcode = self.zip

        # northwest area
        if code == 'BCN':
            return timezone('America/Tijuana') # UTC-8 (-7 DST)
        # Southeast area
        elif code == 'ROO':
            return timezone('America/Bogota') # UTC-5
        # East Chihuahua
        elif code == 'CHH' and zipcode in TIEMPO_DEL_CENTRO_EN_FRONTIERA_ZIPCODES:
            return timezone('America/Boise') # UTC-7 (-6 DST)
        # Tiempo del centro areas
        elif code == 'NAY' and zipcode in TIEMPO_DEL_CENTRO_ZIPCODES:
            return timezone('America/Guatemala') # UTC-6
        # Tiempo del centro en frontiera areas
        elif code in ('TAM', 'NLE', 'COA') and zipcode in TIEMPO_DEL_CENTRO_EN_FRONTIERA_ZIPCODES:
            return timezone('America/Matamoros') # UTC-6 (-5 DST)
        # Pacific area
        elif code in ('SON', 'BCS', 'SIN', 'NAY'):
            return timezone('America/Hermosillo') # UTC-7
        # By default, takes the central area timezone
        return timezone('America/Guatemala') # UTC-6

    def _get_frontend_writable_fields(self):
        frontend_writable_fields = super()._get_frontend_writable_fields()
        frontend_writable_fields.add('l10n_mx_edi_fiscal_regime')

        return frontend_writable_fields

    @api.model
    def get_partner_localisation_fields_required_to_invoice(self, country_id):
        res = super().get_partner_localisation_fields_required_to_invoice(country_id)
        if country_id.code == 'MX':
            res.extend([self.env['ir.model.fields']._get(self._name, 'l10n_mx_edi_fiscal_regime')])
        return res

    def _deduce_country_code(self):
        if self.country_code == 'MX':
            return 'MX'
        return super()._deduce_country_code()

    @api.depends('country_id', 'zip', 'vat')
    def _compute_l10n_mx_edi_partner_address_complete(self):
        for partner in self:
            partner.l10n_mx_edi_partner_address_complete = bool(partner.country_id and partner.zip and partner.vat)