#63 Migrate to new API.

This commit is contained in:
Jared Kipe
2018-12-30 09:40:32 -08:00
parent fb70888119
commit 042d141702
2 changed files with 14 additions and 14 deletions

View File

@@ -1,9 +1,9 @@
from odoo import api, fields, models from odoo import api, fields, models
try: try:
from uszipcode import ZipcodeSearchEngine from uszipcode import SearchEngine
except ImportError: except ImportError:
ZipcodeSearchEngine = None SearchEngine = None
class Partner(models.Model): class Partner(models.Model):
@@ -13,13 +13,13 @@ class Partner(models.Model):
def geo_localize(self): def geo_localize(self):
# We need country names in English below # We need country names in English below
for partner in self.with_context(lang='en_US'): for partner in self.with_context(lang='en_US'):
if ZipcodeSearchEngine and partner.zip: if SearchEngine and partner.zip:
with ZipcodeSearchEngine() as search: with SearchEngine() as search:
zipcode = search.by_zipcode(str(self.zip).split('-')[0]) zipcode = search.by_zipcode(str(self.zip).split('-')[0])
if zipcode and zipcode['Latitude']: if zipcode and zipcode.lat:
partner.write({ partner.write({
'partner_latitude': zipcode['Latitude'], 'partner_latitude': zipcode.lat,
'partner_longitude': zipcode['Longitude'], 'partner_longitude': zipcode.lng,
'date_localization': fields.Date.context_today(partner), 'date_localization': fields.Date.context_today(partner),
}) })
else: else:

View File

@@ -8,10 +8,10 @@ from logging import getLogger
_logger = getLogger(__name__) _logger = getLogger(__name__)
try: try:
from uszipcode import ZipcodeSearchEngine from uszipcode import SearchEngine
except ImportError: except ImportError:
_logger.warn('module "uszipcode" cannot be loaded, falling back to Google API') _logger.warn('module "uszipcode" cannot be loaded, falling back to Google API')
ZipcodeSearchEngine = None SearchEngine = None
from odoo import api, fields, models, tools from odoo import api, fields, models, tools
from odoo.addons.base_geolocalize.models.res_partner import geo_find, geo_query_address from odoo.addons.base_geolocalize.models.res_partner import geo_find, geo_query_address
@@ -58,12 +58,12 @@ class FakePartner():
if not hasattr(self, 'date_localization') and self.date_localization: if not hasattr(self, 'date_localization') and self.date_localization:
self.date_localization = 'TODAY!' self.date_localization = 'TODAY!'
# The fast way. # The fast way.
if ZipcodeSearchEngine and self.zip: if SearchEngine and self.zip:
with ZipcodeSearchEngine() as search: with SearchEngine() as search:
zipcode = search.by_zipcode(str(self.zip).split('-')[0]) zipcode = search.by_zipcode(str(self.zip).split('-')[0])
if zipcode and zipcode['Latitude']: if zipcode and zipcode.lat:
self.partner_latitude = zipcode['Latitude'] self.partner_latitude = zipcode.lat
self.partner_longitude = zipcode['Longitude'] self.partner_longitude = zipcode.lng
return self.date_localization return self.date_localization
# The slow way. # The slow way.