|
|
# -*- coding: utf-8 -*-# © 2015 Antiun Ingeniería S.L. - Antonio Espinosa# © 2015 Antiun Ingeniería S.L. - Jairo Llopis# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from openerp import models, fields, api
class ResPartner(models.Model): _inherit = 'res.partner'
nuts1_id = fields.Many2one( comodel_name='res.partner.nuts', domain=[('level', '=', 1)], string="NUTS L1") nuts2_id = fields.Many2one( comodel_name='res.partner.nuts', domain=[('level', '=', 2)], string="NUTS L2", oldname="region") nuts3_id = fields.Many2one( comodel_name='res.partner.nuts', domain=[('level', '=', 3)], string="NUTS L3", oldname="substate") nuts4_id = fields.Many2one( comodel_name='res.partner.nuts', domain=[('level', '=', 4)], string="NUTS L4")
@api.multi def _onchange_nuts(self, level): field = self["nuts%d_id" % level] country_id = field.country_id.id state_id = field.state_id.id if country_id and self.country_id.id != country_id: self.country_id = country_id if state_id and self.state_id.id != state_id: self.state_id = state_id if (level - 1) > 0: parent_id = field.parent_id.id if parent_id: parent_field = 'nuts%d_id' % (level - 1) if self[parent_field].id != parent_id: self[parent_field] = parent_id result = dict() if country_id and level < 4: result['domain'] = dict() while level < 4: parent_field = 'nuts%d_id' % level domain_field = 'nuts%d_id' % (level + 1) parent_id = self[parent_field].id if parent_id: result['domain'][domain_field] = [ ('parent_id', '=', parent_id), ] level += 1 return result
@api.multi @api.onchange('nuts4_id') def _onchange_nuts4_id(self): return self._onchange_nuts(4)
@api.multi @api.onchange('nuts3_id') def _onchange_nuts3_id(self): return self._onchange_nuts(3)
@api.multi @api.onchange('nuts2_id') def _onchange_nuts2_id(self): return self._onchange_nuts(2)
@api.multi @api.onchange('nuts1_id') def _onchange_nuts1_id(self): return self._onchange_nuts(1)
@api.multi @api.onchange('country_id') def _onchange_country_id(self): """Sensible values and domains for related fields.""" fields = ['state_id', 'nuts1_id', 'nuts2_id', 'nuts3_id', 'nuts4_id'] country_domain = ([('country_id', '=', self.country_id.id)] if self.country_id else []) domain = dict() for field in fields: if self.country_id and self[field].country_id != self.country_id: self[field] = False domain[field] = list(country_domain) # Using list() to copy fields.remove('state_id') for field in fields: level = int(field[:5][-1]) if level: domain[field].append(('level', '=', level)) if self.country_id: nuts1 = self.env['res.partner.nuts'].search([ ('level', '=', 1), ('country_id', '=', self.country_id.id), ], limit=1) if self.nuts1_id.id != nuts1.id: self.nuts1_id = nuts1.id return { 'domain': domain, }
@api.multi def onchange_state(self, state_id): result = super(ResPartner, self).onchange_state(state_id) state = self.env['res.country.state'].browse(state_id) if state.country_id.state_level: nuts_state = self.env['res.partner.nuts'].search([ ('level', '=', state.country_id.state_level), ('state_id', '=', state.id) ], limit=1) if nuts_state: field = 'nuts%d_id' % state.country_id.state_level result.setdefault("value", dict()) result['value'][field] = nuts_state.id return result
@api.model def _address_fields(self): fields = super(ResPartner, self)._address_fields() if fields: fields += ['nuts1_id', 'nuts2_id', 'nuts3_id', 'nuts4_id'] return fields
|
