# Copyright 2016 Vauxoo
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import os

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


class CompanyCountryConfigSettings(models.TransientModel):
    _name = 'company.country.config.settings'
    _description = 'Company Country Configuration Settings'

    @api.model
    def load_company_country(self, country_code=None):
        if not country_code:
            country_code = os.environ.get('COUNTRY')
        if country_code == "":
            self.env.ref('base.main_company').write({'country_id': None})
            return
        if not country_code:
            l10n_to_install = self.env['ir.module.module'].search([
                ('state', '=', 'to install'),
                ('name', '=like', 'l10n_%')], limit=1)
            if not l10n_to_install:
                raise ValidationError(
                    _('Error COUNTRY environment variable with country code '
                        'not defined and no localization found in pool.'))
            country_code = l10n_to_install.name.split('l10n_')[1][:2].upper()

        country = self.env['res.country'].search([
            ('code', 'ilike', country_code)], limit=1)
        if not country:
            raise ValidationError(
                _('Country code %s not found. Use ISO 3166 codes 2 letters'))
        self.env.ref('base.main_company').write({'country_id': country.id})