from odoo import _, api, models, tools, modules
from odoo.exceptions import ValidationError


class ResCompany(models.Model):
    _inherit = 'res.company'

    @api.constrains('country_id')
    def _check_country_change_holidays(self):
        if tools.config['test_enable'] or modules.module.current_test:
            return
        for record in self:
            conflict_domain = [
                ('employee_company_id', '=', record.id),
                ('work_entry_type_id.country_id', '!=', False),
                ('work_entry_type_id.country_id', '!=', record.country_id.id),
            ]
            if (
                self.env['hr.leave'].search_count(conflict_domain, limit=1)
                or self.env['hr.leave.allocation'].search_count(conflict_domain, limit=1)
            ):
                raise ValidationError(_(
                    "The company country cannot be changed while time off leaves "
                    "or allocations with the country exist."
                ))