#63 Migrate to new API.

l10n_us_partner_zipcode/res_partner.py

@@ -3,10 +3,10 @@ import logging
 _logger = logging.getLogger(__name__)
 
 try:
-    from uszipcode import ZipcodeSearchEngine
+    from uszipcode import SearchEngine
 except ImportError:
     _logger.warn('module "uszipcode" cannot be loaded, you will be unable to detect Cities and States by ZIP')
-    ZipcodeSearchEngine = None
+    SearchEngine = None
 
 from odoo import api, fields, models
 
@@ -16,24 +16,24 @@ class Partner(models.Model):
 
     @api.onchange('zip')
     def _zip_to_city_state(self):
-        if ZipcodeSearchEngine and self.zip and not self.city:
+        if SearchEngine and self.zip and not self.city:
             country_us = self.env['res.country'].search([('code', '=', 'US')], limit=1)
             state_obj = self.env['res.country.state']
             if not self.country_id or self.country_id.id == country_us.id:
-                with ZipcodeSearchEngine() as search:
+                with SearchEngine() as search:
                     zipcode = search.by_zipcode(self.zip)
                     if zipcode:
                         if not self.country_id:
                             self.country_id = country_us
 
-                        self.city = zipcode['City']
+                        self.city = zipcode.city
                         self.state_id = state_obj.search([
-                            ('code', '=', zipcode['State']),
+                            ('code', '=', zipcode.state),
                             ('country_id', '=', country_us.id),
                         ], limit=1)
 
                         if hasattr(self, 'partner_latitude') and not self.partner_latitude:
-                            self.partner_latitude = zipcode['Latitude']
+                            self.partner_latitude = zipcode.lat
-                            self.partner_longitude = zipcode['Longitude']
+                            self.partner_longitude = zipcode.lng
                             self.date_localization = fields.Date.context_today(self)
         return {}