# -*- coding:utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

import re

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


class L10nChLocationUnit(models.Model):
    _name = 'l10n.ch.location.unit'
    _description = 'Work Place - Swiss Payroll'
    _rec_name = 'partner_id'

    active = fields.Boolean(default=True)
    company_id = fields.Many2one('res.company', required=True, default=lambda self: self.env.company)
    partner_id = fields.Many2one('res.partner', string="Address", required=True)
    bur_ree_number = fields.Char(
        string="BUR-REE-Number",
        required=False,
        help="Depending on the structure of the company and the number of workplaces, there are one or more REE numbers.")
    in_house_id = fields.Char(string="InHouseID")

    canton = fields.Selection([
        ('AG', 'Argovie'),
        ('AI', 'Appenzell Rhodes-Intérieures'),
        ('AR', 'Appenzell Rhodes-Extérieures'),
        ('BE', 'Berne'),
        ('BL', 'Bâle-Campagne'),
        ('BS', 'Bâle-Ville'),
        ('FR', 'Fribourg'),
        ('GE', 'Genève'),
        ('GL', 'Glaris'),
        ('GR', 'Grisons'),
        ('JU', 'Jura'),
        ('LU', 'Lucerne'),
        ('NE', 'Neuchâtel'),
        ('NW', 'Nidwald'),
        ('OW', 'Obwald'),
        ('SG', 'Saint-Gall'),
        ('SH', 'Schaffhouse'),
        ('SO', 'Soleure'),
        ('SZ', 'Schwytz'),
        ('TG', 'Thurgovie'),
        ('TI', 'Tessin'),
        ('UR', 'Uri'),
        ('VD', 'Vaud'),
        ('VS', 'Valais'),
        ('ZG', 'Zoug'),
        ('ZH', 'Zurich'),
    ], required=False, compute="_compute_autocomplete_canton_municipality", store=True, readonly=False)
    dpi_number = fields.Char('DPI Number', required=False)  # Equivalent to SSL nummer
    municipality = fields.Char(string="Municipality ID", required=False, compute="_compute_autocomplete_canton_municipality", store=True, readonly=False)
    weekly_hours = fields.Float(string="Weekly Hours", default=40)
    weekly_lessons = fields.Float(string="Weekly Lessons")

    _unique = models.Constraint(
        'unique (company_id, partner_id)',
        "A work location cannot be set more than once for the same company and partner.",
    )

    @api.constrains('bur_ree_number')
    def _check_bur_ree_number(self):
        pattern = r'[A-Z][0-9]{8}'
        for location in self:
            if location.bur_ree_number:
                if re.fullmatch(pattern, location.bur_ree_number):
                    if not self.env['res.company']._l10n_ch_modulo_11_checksum(location.bur_ree_number, 7):
                        raise ValidationError(_("BUR-REE-Number checksum is not correct"))
                else:
                    raise ValidationError(_("BUR-REE-Number does not match the right format"))

    @api.depends("partner_id.zip")
    def _compute_autocomplete_canton_municipality(self):
        ZIP_DATA = self.env['hr.rule.parameter']._get_parameter_from_code("l10n_ch_bfs_municipalities", fields.Date.today(), raise_if_not_found=False)
        if ZIP_DATA:
            for record in self:
                if record.partner_id.zip:
                    data = ZIP_DATA.get(record.partner_id.zip)
                    if data:
                        record.municipality = data[1]
                        record.canton = data[2]