352 lines
12 KiB
Python
352 lines
12 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
'Address'
|
|
from string import Template
|
|
|
|
from sql.conditionals import Coalesce
|
|
from sql.functions import Substring
|
|
from sql.operators import Concat
|
|
|
|
from trytond.i18n import gettext
|
|
from trytond.model import (
|
|
ModelView, ModelSQL, MatchMixin, DeactivableMixin, fields,
|
|
sequence_ordered)
|
|
from trytond.model.exceptions import AccessError
|
|
from trytond.pyson import Eval, If
|
|
from trytond.pool import Pool
|
|
from trytond.rpc import RPC
|
|
from trytond.transaction import Transaction
|
|
from trytond.cache import Cache
|
|
from .exceptions import InvalidFormat
|
|
|
|
__all__ = ['Address', 'AddressFormat']
|
|
|
|
STATES = {
|
|
'readonly': ~Eval('active'),
|
|
}
|
|
DEPENDS = ['active']
|
|
|
|
|
|
class Address(DeactivableMixin, sequence_ordered(), ModelSQL, ModelView):
|
|
"Address"
|
|
__name__ = 'party.address'
|
|
party = fields.Many2One('party.party', 'Party', required=True,
|
|
ondelete='CASCADE', select=True, states={
|
|
'readonly': If(~Eval('active'), True, Eval('id', 0) > 0),
|
|
},
|
|
depends=['active', 'id'])
|
|
party_name = fields.Char(
|
|
"Party Name", states=STATES, depends=DEPENDS,
|
|
help="If filled, replace the name of the party for address formatting")
|
|
name = fields.Char("Building Name", states=STATES, depends=DEPENDS)
|
|
street = fields.Text("Street", states=STATES, depends=DEPENDS)
|
|
zip = fields.Char('Zip', states=STATES, depends=DEPENDS)
|
|
city = fields.Char('City', states=STATES, depends=DEPENDS)
|
|
country = fields.Many2One('country.country', 'Country',
|
|
states=STATES, depends=DEPENDS)
|
|
subdivision = fields.Many2One("country.subdivision",
|
|
'Subdivision', domain=[
|
|
('country', '=', Eval('country', -1)),
|
|
('parent', '=', None),
|
|
],
|
|
states=STATES, depends=['active', 'country'])
|
|
full_address = fields.Function(fields.Text('Full Address'),
|
|
'get_full_address')
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Address, cls).__setup__()
|
|
cls._order.insert(0, ('party', 'ASC'))
|
|
cls.__rpc__.update(
|
|
autocomplete_zip=RPC(instantiate=0, cache=dict(days=1)),
|
|
autocomplete_city=RPC(instantiate=0, cache=dict(days=1)),
|
|
)
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
cursor = Transaction().connection.cursor()
|
|
sql_table = cls.__table__()
|
|
|
|
super(Address, cls).__register__(module_name)
|
|
|
|
table = cls.__table_handler__(module_name)
|
|
|
|
# Migration from 4.0: remove streetbis
|
|
if table.column_exist('streetbis'):
|
|
value = Concat(
|
|
Coalesce(sql_table.street, ''),
|
|
Concat('\n', Coalesce(sql_table.streetbis, '')))
|
|
cursor.execute(*sql_table.update(
|
|
[sql_table.street],
|
|
[value]))
|
|
table.drop_column('streetbis')
|
|
|
|
_autocomplete_limit = 100
|
|
|
|
@fields.depends('country', 'subdivision')
|
|
def _autocomplete_domain(self):
|
|
domain = []
|
|
if self.country:
|
|
domain.append(('country', '=', self.country.id))
|
|
if self.subdivision:
|
|
domain.append(['OR',
|
|
('subdivision', 'child_of',
|
|
[self.subdivision.id], 'parent'),
|
|
('subdivision', '=', None),
|
|
])
|
|
return domain
|
|
|
|
def _autocomplete_search(self, domain, name):
|
|
pool = Pool()
|
|
Zip = pool.get('country.zip')
|
|
if domain:
|
|
records = Zip.search(domain, limit=self._autocomplete_limit)
|
|
if len(records) < self._autocomplete_limit:
|
|
return sorted({getattr(z, name) for z in records})
|
|
return []
|
|
|
|
@fields.depends('city', methods=['_autocomplete_domain'])
|
|
def autocomplete_zip(self):
|
|
domain = self._autocomplete_domain()
|
|
if self.city:
|
|
domain.append(('city', 'ilike', '%%%s%%' % self.city))
|
|
return self._autocomplete_search(domain, 'zip')
|
|
|
|
@fields.depends('zip', methods=['_autocomplete_domain'])
|
|
def autocomplete_city(self):
|
|
domain = self._autocomplete_domain()
|
|
if self.zip:
|
|
domain.append(('zip', 'ilike', '%s%%' % self.zip))
|
|
return self._autocomplete_search(domain, 'city')
|
|
|
|
def get_full_address(self, name):
|
|
pool = Pool()
|
|
AddressFormat = pool.get('party.address.format')
|
|
full_address = Template(AddressFormat.get_format(self)).substitute(
|
|
**self._get_address_substitutions())
|
|
return '\n'.join(
|
|
filter(None, (x.strip() for x in full_address.splitlines())))
|
|
|
|
def _get_address_substitutions(self):
|
|
context = Transaction().context
|
|
subdivision_code = ''
|
|
if getattr(self, 'subdivision', None):
|
|
subdivision_code = self.subdivision.code or ''
|
|
if '-' in subdivision_code:
|
|
subdivision_code = subdivision_code.split('-', 1)[1]
|
|
substitutions = {
|
|
'party_name': '',
|
|
'attn': '',
|
|
'name': getattr(self, 'name', None) or '',
|
|
'street': getattr(self, 'street', None) or '',
|
|
'zip': getattr(self, 'zip', None) or '',
|
|
'city': getattr(self, 'city', None) or '',
|
|
'subdivision': (self.subdivision.name
|
|
if getattr(self, 'subdivision', None) else ''),
|
|
'subdivision_code': subdivision_code,
|
|
'country': (self.country.name
|
|
if getattr(self, 'country', None) else ''),
|
|
'country_code': (self.country.code or ''
|
|
if getattr(self, 'country', None) else ''),
|
|
}
|
|
if context.get('address_from_country') == getattr(self, 'country', ''):
|
|
substitutions['country'] = ''
|
|
if context.get('address_with_party', False):
|
|
substitutions['party_name'] = self.party_full_name
|
|
if context.get('address_attention_party', False):
|
|
substitutions['attn'] = (
|
|
context['address_attention_party'].full_name)
|
|
for key, value in list(substitutions.items()):
|
|
substitutions[key.upper()] = value.upper()
|
|
return substitutions
|
|
|
|
@property
|
|
def party_full_name(self):
|
|
name = ''
|
|
if self.party_name:
|
|
name = self.party_name
|
|
elif self.party:
|
|
name = self.party.full_name
|
|
return name
|
|
|
|
def get_rec_name(self, name):
|
|
party = self.party_full_name
|
|
if self.street:
|
|
street = self.street.splitlines()[0]
|
|
else:
|
|
street = None
|
|
if self.country:
|
|
country = self.country.code
|
|
else:
|
|
country = None
|
|
return ', '.join(
|
|
filter(None, [
|
|
party,
|
|
self.name,
|
|
street,
|
|
self.zip,
|
|
self.city,
|
|
country]))
|
|
|
|
@classmethod
|
|
def search_rec_name(cls, name, clause):
|
|
if clause[1].startswith('!') or clause[1].startswith('not '):
|
|
bool_op = 'AND'
|
|
else:
|
|
bool_op = 'OR'
|
|
return [bool_op,
|
|
('party',) + tuple(clause[1:]),
|
|
('name',) + tuple(clause[1:]),
|
|
('street',) + tuple(clause[1:]),
|
|
('zip',) + tuple(clause[1:]),
|
|
('city',) + tuple(clause[1:]),
|
|
('country',) + tuple(clause[1:]),
|
|
]
|
|
|
|
@classmethod
|
|
def write(cls, *args):
|
|
actions = iter(args)
|
|
for addresses, values in zip(actions, actions):
|
|
if 'party' in values:
|
|
for address in addresses:
|
|
if address.party.id != values['party']:
|
|
raise AccessError(
|
|
gettext('party.msg_address_change_party',
|
|
address=address.rec_name))
|
|
super(Address, cls).write(*args)
|
|
|
|
@fields.depends('subdivision', 'country')
|
|
def on_change_country(self):
|
|
if (self.subdivision
|
|
and self.subdivision.country != self.country):
|
|
self.subdivision = None
|
|
|
|
|
|
class AddressFormat(DeactivableMixin, MatchMixin, ModelSQL, ModelView):
|
|
"Address Format"
|
|
__name__ = 'party.address.format'
|
|
country_code = fields.Char("Country Code", size=2)
|
|
language_code = fields.Char("Language Code", size=2)
|
|
format_ = fields.Text("Format", required=True,
|
|
help="Available variables (also in upper case):\n"
|
|
"- ${party_name}\n"
|
|
"- ${name}\n"
|
|
"- ${attn}\n"
|
|
"- ${street}\n"
|
|
"- ${zip}\n"
|
|
"- ${city}\n"
|
|
"- ${subdivision}\n"
|
|
"- ${subdivision_code}\n"
|
|
"- ${country}\n"
|
|
"- ${country_code}")
|
|
|
|
_get_format_cache = Cache('party.address.format.get_format')
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(AddressFormat, cls).__setup__()
|
|
cls._order.insert(0, ('country_code', 'ASC NULLS LAST'))
|
|
cls._order.insert(1, ('language_code', 'ASC NULLS LAST'))
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
pool = Pool()
|
|
Country = pool.get('country.country')
|
|
Language = pool.get('ir.lang')
|
|
country = Country.__table__()
|
|
language = Language.__table__()
|
|
table = cls.__table__()
|
|
cursor = Transaction().connection.cursor()
|
|
|
|
super().__register__(module_name)
|
|
|
|
table_h = cls.__table_handler__()
|
|
|
|
# Migration from 5.2: replace country by country_code
|
|
if table_h.column_exist('country'):
|
|
query = table.update(
|
|
[table.country_code],
|
|
country.select(
|
|
country.code,
|
|
where=country.id == table.country))
|
|
cursor.execute(*query)
|
|
table_h.drop_column('country')
|
|
|
|
# Migration from 5.2: replace language by language_code
|
|
if table_h.column_exist('language'):
|
|
query = table.update(
|
|
[table.language_code],
|
|
language.select(
|
|
Substring(language.code, 0, 2),
|
|
where=language.id == table.language))
|
|
cursor.execute(*query)
|
|
table_h.drop_column('language')
|
|
|
|
@classmethod
|
|
def default_format_(cls):
|
|
return """${party_name}
|
|
${name}
|
|
${street}
|
|
${zip} ${city}
|
|
${subdivision}
|
|
${COUNTRY}"""
|
|
|
|
@classmethod
|
|
def create(cls, *args, **kwargs):
|
|
records = super(AddressFormat, cls).create(*args, **kwargs)
|
|
cls._get_format_cache.clear()
|
|
return records
|
|
|
|
@classmethod
|
|
def write(cls, *args, **kwargs):
|
|
super(AddressFormat, cls).write(*args, **kwargs)
|
|
cls._get_format_cache.clear()
|
|
|
|
@classmethod
|
|
def delete(cls, *args, **kwargs):
|
|
super(AddressFormat, cls).delete(*args, **kwargs)
|
|
cls._get_format_cache.clear()
|
|
|
|
@classmethod
|
|
def validate(cls, formats):
|
|
super(AddressFormat, cls).validate(formats)
|
|
for format_ in formats:
|
|
format_.check_format()
|
|
|
|
def check_format(self):
|
|
pool = Pool()
|
|
Address = pool.get('party.address')
|
|
address = Address()
|
|
try:
|
|
Template(self.format_).substitute(
|
|
**address._get_address_substitutions())
|
|
except Exception as exception:
|
|
raise InvalidFormat(gettext('party.invalid_format',
|
|
format=self.format_,
|
|
exception=exception)) from exception
|
|
|
|
@classmethod
|
|
def get_format(cls, address, pattern=None):
|
|
if pattern is None:
|
|
pattern = {}
|
|
else:
|
|
pattern = pattern.copy()
|
|
pattern.setdefault(
|
|
'country_code', address.country.code if address.country else None)
|
|
pattern.setdefault('language_code', Transaction().language[:2])
|
|
|
|
key = tuple(sorted(pattern.items()))
|
|
format_ = cls._get_format_cache.get(key)
|
|
if format_ is not None:
|
|
return format_
|
|
|
|
for record in cls.search([]):
|
|
if record.match(pattern):
|
|
format_ = record.format_
|
|
break
|
|
else:
|
|
format_ = cls.default_format_()
|
|
|
|
cls._get_format_cache.set(key, format_)
|
|
return format_
|