[REF] select_or_create_state

and tests

base_location_geonames_import/__openerp__.py

@@ -4,6 +4,8 @@
 #    Base Location Geonames Import module for OpenERP
 #    Copyright (C) 2014 Akretion (http://www.akretion.com)
 #    @author Alexis de Lattre <alexis.delattre@akretion.com>
+#    Copyright (C) 2014 Agile Business Group (http://www.agilebg.com)
+#    @author Lorenzo Battistini <lorenzo.battistini@agilebg.com>
 #
 #    This program is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU Affero General Public License as
@@ -23,7 +25,7 @@
 
 {
     'name': 'Base Location Geonames Import',
-    'version': '0.1',
+    'version': '0.2',
     'category': 'Extra Tools',
     'license': 'AGPL-3',
     'summary': 'Import better zip entries from Geonames',
@@ -34,6 +36,13 @@ Base Location Geonames Import
 This module adds a wizard to import better zip entries from Geonames (http://download.geonames.org/export/zip/). When you start the wizard, it will ask you to select a country ; then, for the selected country, it will delete all the current better zip entries, download the latest version of the list of cities from geonames.org and create new better zip entries.
 
 Please contact Alexis de Lattre from Akretion <alexis.delattre@akretion.com> for any help or question about this module.
+
+
+Contributors
+------------
+
+- Alexis de Lattre <alexis.delattre@akretion.com>
+- Lorenzo Battistini <lorenzo.battistini@agilebg.com>
 """,
     'author': 'Akretion',
     'website': 'http://www.akretion.com',
@@ -42,6 +51,9 @@ Please contact Alexis de Lattre from Akretion <alexis.delattre@akretion.com> for
     'data': [
         'wizard/geonames_import_view.xml',
         ],
+    'test': [
+        'test/import.yml'
+        ],
     'installable': True,
     'active': False,
 }

base_location_geonames_import/test/import.yml

@@ -0,0 +1,27 @@
+-
+  I create the wizard
+-
+  !record {model: better.zip.geonames.import, id: import_wizard_1, view: better_zip_geonames_import_form}:
+    country_id: base.it
+-
+  I run the import
+-
+  !python {model: better.zip.geonames.import}: |
+    self.run_import(cr, uid, [ref('import_wizard_1')], context=context)
+-
+  I check the data
+-
+  !python {model: res.better.zip}: |
+    state_obj = self.pool['res.country.state']
+    state_ids = state_obj.search(cr, uid, [
+        ('code', '=', 'LG'),
+        ('country_id', '=', ref('base.it')),
+        ], context=context)
+    assert len(state_ids) == 1, "There must be 1 LG"
+    zip_ids = self.search(cr, uid, [
+        ('name', '=', '16017'),
+        ('city', '=', 'Isola Del Cantone'),
+        ('state_id', '=', state_ids[0]),
+        ('country_id', '=', ref('base.it'))
+        ], context=context)
+    assert len(zip_ids) == 1, "There must be 1 'Isola Del Cantone'"

base_location_geonames_import/wizard/geonames_import.py

@@ -4,6 +4,8 @@
 #    Base Location Geonames Import module for OpenERP
 #    Copyright (C) 2014 Akretion (http://www.akretion.com)
 #    @author Alexis de Lattre <alexis.delattre@akretion.com>
+#    Copyright (C) 2014 Agile Business Group (http://www.agilebg.com)
+#    @author Lorenzo Battistini <lorenzo.battistini@agilebg.com>
 #
 #    This program is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU Affero General Public License as
@@ -41,22 +43,19 @@ class better_zip_geonames_import(models.TransientModel):
     country_id = fields.Many2one('res.country', 'Country', required=True)
 
     @api.model
-    def _prepare_better_zip(self, row, country_id, states):
-        '''This function is designed to be inherited'''
-        state_id = False
-        if states and row[4] and row[4] in states:
-            state_id = states[row[4].upper()]
+    def _prepare_better_zip(self, row, country_id):
+        state = self.select_or_create_state(row, country_id)
         vals = {
             'name': row[1],
             'city': row[2],
-            'state_id': state_id,
+            'state_id': state.id,
             'country_id': country_id,
             }
         return vals
 
     @api.model
     def create_better_zip(
-            self, row, country_id, country_code, states):
+            self, row, country_id, country_code):
         bzip_id = False
         if row[0] != country_code:
             raise Warning(
@@ -66,11 +65,32 @@ class better_zip_geonames_import(models.TransientModel):
                 % (row[0], country_code))
         logger.debug('ZIP = %s - City = %s' % (row[1], row[2]))
         if row[1] and row[2]:
-            vals = self._prepare_better_zip(row, country_id, states)
+            vals = self._prepare_better_zip(row, country_id)
             if vals:
                 bzip_id = self.env['res.better.zip'].create(vals)
         return bzip_id
 
+    @api.model
+    def select_or_create_state(
+        self, row, country_id, code_row_index=4, name_row_index=3
+    ):
+        states = self.env['res.country.state'].search([
+            ('country_id', '=', country_id),
+            ('code', '=', row[code_row_index]),
+            ])
+        if len(states) > 1:
+            raise Warning(
+                _("Too many states with code %s for counrty %s")
+                % (row[code_row_index], country_id))
+        if len(states) == 1:
+            return states[0]
+        else:
+            return self.env['res.country.state'].create({
+                'name': row[name_row_index],
+                'code': row[code_row_index],
+                'country_id': country_id
+                })
+
     @api.one
     def run_import(self):
         bzip_obj = self.env['res.better.zip']
@@ -90,15 +110,6 @@ class better_zip_geonames_import(models.TransientModel):
             logger.info(
                 '%d better zip entries deleted for country %s'
                 % (len(bzip_ids_to_delete), self.country_id.name))
-        state_ids = self.env['res.country.state'].search(
-            [('country_id', '=', country_id)])
-        states = {}
-        # key = code of the state ; value = ID of the state in OpenERP
-        if state_ids:
-            states_r = self.env['res.country.state'].read(
-                state_ids, ['code', 'country_id'])
-            for state in states_r:
-                states[state['code'].upper()] = state['id']
         f_geonames = zipfile.ZipFile(StringIO.StringIO(res_request.content))
         tempdir = tempfile.mkdtemp(prefix='openerp')
         f_geonames.extract('%s.txt' % country_code, tempdir)
@@ -106,11 +117,10 @@ class better_zip_geonames_import(models.TransientModel):
         data_file = open(os.path.join(tempdir, '%s.txt' % country_code), 'r')
         data_file.seek(0)
         logger.info(
-            'Starting to create the better zip entries %s state information'
-            % (states and 'with' or 'without'))
+            'Starting to create the better zip entries')
         for row in unicodecsv.reader(
                 data_file, encoding='utf-8', delimiter='	'):
-            self.create_better_zip(row, country_id, country_code, states)
+            self.create_better_zip(row, country_id, country_code)
         data_file.close()
         logger.info(
             'The wizard to create better zip entries from geonames '