82 lines
3.0 KiB
Python
82 lines
3.0 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\-]+) (.+)', lines[-1])
|
|
if result:
|
|
lines.pop()
|
|
return (result.group(1), result.group(2))
|
|
return False
|
|
|
|
if stdnum_vat is None:
|
|
return {}
|
|
|
|
for partner in self:
|
|
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]
|
|
partner.street = lines.pop(0)
|
|
if len(lines) > 0:
|
|
res = _check_city(lines, result['countryCode'])
|
|
if res:
|
|
partner.zip = res[0]
|
|
partner.city = res[1]
|
|
if len(lines) > 0:
|
|
partner.street2 = lines.pop(0)
|
|
|
|
country = self.env['res.country'].search([('code', '=', result['countryCode'])], limit=1)
|
|
partner.country_id = country and country.id or False
|