Lucas Perais (lpe)
9645a64d0e
Bafore this commit, fields that were not written during on change were kept leading to having a wrong overall address when changing vat when the return of the API did not contain the same fields as before. Also, ZIP from Netherlands were not supported After this commit, when changing the VAT number, fields that were not set by this operation are reset to False Also, NL zip codes compute fine OPW 807666
95 lines
3.6 KiB
Python
95 lines
3.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
import logging
|
|
import re
|
|
|
|
from odoo import api, models
|
|
|
|
from suds.client import Client
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
import stdnum.eu.vat as stdnum_vat
|
|
except ImportError:
|
|
_logger.warning('Python `stdnum` library not found, unable to call VIES service to detect address based on VAT number.')
|
|
stdnum_vat = None
|
|
|
|
|
|
class ResPartner(models.Model):
|
|
_inherit = 'res.partner'
|
|
|
|
@api.onchange('vat')
|
|
def vies_vat_change(self):
|
|
def _check_city(lines, country='BE'):
|
|
if country == 'GB':
|
|
ukzip = '[A-Z]{1,2}[0-9][0-9A-Z]?\s?[0-9][A-Z]{2}'
|
|
if re.match(ukzip, lines[-1]):
|
|
cp = lines.pop()
|
|
city = lines.pop()
|
|
return (cp, city)
|
|
else:
|
|
result = re.match('((?:L-|AT-)?[0-9\-]+[A-Z]{,2}) (.+)', lines[-1])
|
|
if result:
|
|
lines.pop()
|
|
return (result.group(1), result.group(2))
|
|
return False
|
|
|
|
def _set_address_field(partner, field, value):
|
|
partner[field] = value
|
|
non_set_address_fields.remove(field)
|
|
|
|
if stdnum_vat is None:
|
|
return {}
|
|
|
|
for partner in self:
|
|
# If a field is non set in this algorithm
|
|
# wipe it anyway
|
|
non_set_address_fields = set(['street', 'street2', 'city', 'zip', 'state_id', 'country_id'])
|
|
if not partner.vat:
|
|
return {}
|
|
if len(partner.vat) > 5 and partner.vat[:2].lower() in stdnum_vat.country_codes:
|
|
# Equivalent to stdnum_vat.check_vies(partner.vat).
|
|
# However, we want to add a custom timeout to the suds.client
|
|
# because by default, it's 120 seconds and this is to long.
|
|
try:
|
|
client = Client(stdnum_vat.vies_wsdl, timeout=5)
|
|
partner_vat = stdnum_vat.compact(partner.vat)
|
|
result = client.service.checkVat(partner_vat[:2], partner_vat[2:])
|
|
except:
|
|
# Avoid blocking the client when the service is unreachable/unavailable
|
|
return {}
|
|
|
|
if not result['valid']:
|
|
return {}
|
|
|
|
if (not partner.name) and (result['name'] != '---'):
|
|
partner.name = result['name']
|
|
|
|
#parse the address from VIES and fill the partner's data
|
|
if result['address'] == '---': return {}
|
|
|
|
lines = [x for x in result['address'].split("\n") if x]
|
|
if len(lines) == 1:
|
|
lines = [x.strip() for x in lines[0].split(',') if x]
|
|
if len(lines) == 1:
|
|
lines = [x.strip() for x in lines[0].split(' ') if x]
|
|
|
|
_set_address_field(partner, 'street', lines.pop(0))
|
|
|
|
if len(lines) > 0:
|
|
res = _check_city(lines, result['countryCode'])
|
|
if res:
|
|
_set_address_field(partner, 'zip', res[0])
|
|
_set_address_field(partner, 'city', res[1])
|
|
if len(lines) > 0:
|
|
_set_address_field(partner, 'street2', lines.pop(0))
|
|
|
|
country = self.env['res.country'].search([('code', '=', result['countryCode'])], limit=1)
|
|
_set_address_field(partner, 'country_id', country and country.id or False)
|
|
|
|
for field in non_set_address_fields:
|
|
if partner[field]:
|
|
partner[field] = False
|