#This file is part country_zip module for Tryton.
#The COPYRIGHT file at the top level of this repository contains
#the full copyright notices and license terms.
from trytond.pool import Pool, PoolMeta

__all__ = ['Address']
__metaclass__ = PoolMeta


class Address:
    __name__ = 'party.address'

    @classmethod
    def __setup__(cls):
        super(Address, cls).__setup__()
        if cls.zip.on_change is None:
            cls.zip.on_change = []
        for field in ('zip', 'country'):
            if not field in cls.zip.on_change:
                cls.zip.on_change.append(field)
        if cls.country.on_change is None:
            cls.country.on_change = []
        for field in ('zip', 'country'):
            if not field in cls.country.on_change:
                cls.country.on_change.append(field)

    @staticmethod
    def default_country():
        Configuration = Pool().get('party.configuration')
        config = Configuration(1)
        if config.default_country:
            return config.default_country.id

    def get_subdivision_country(self):
        res = {}
        Zip = Pool().get('country.zip')

        if self.zip and self.country:
            zips = Zip.search([
                        ('zip', '=', self.zip),
                        ('subdivision.country', '=', self.country.id),
                        ])
            if zips:
                zip_ = zips[0]
                res['city'] = zip_.city
                if zip_.subdivision:
                    res['subdivision'] = zip_.subdivision.id
        return res

    def on_change_zip(self):
        return self.get_subdivision_country()

    def on_change_country(self):
        return self.get_subdivision_country()