From b118c762d4cb5a2fbfcb8e9876b978e45a273d6f Mon Sep 17 00:00:00 2001 From: Lorenzo Battistini Date: Mon, 11 Aug 2014 10:13:50 +0200 Subject: [PATCH] [REF] select_or_create_state and tests --- base_location_geonames_import/__openerp__.py | 18 +++++-- base_location_geonames_import/test/import.yml | 27 ++++++++++ .../wizard/geonames_import.py | 50 +++++++++++-------- 3 files changed, 72 insertions(+), 23 deletions(-) create mode 100644 base_location_geonames_import/test/import.yml diff --git a/base_location_geonames_import/__openerp__.py b/base_location_geonames_import/__openerp__.py index bd5518843..7df05c801 100644 --- a/base_location_geonames_import/__openerp__.py +++ b/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 +# Copyright (C) 2014 Agile Business Group (http://www.agilebg.com) +# @author Lorenzo Battistini # # 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,14 +36,24 @@ 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 for any help or question about this module. - """, + + +Contributors +------------ + +- Alexis de Lattre +- Lorenzo Battistini +""", 'author': 'Akretion', 'website': 'http://www.akretion.com', 'depends': ['base_location'], 'external_dependencies': {'python': ['requests', 'unicodecsv']}, 'data': [ 'wizard/geonames_import_view.xml', - ], + ], + 'test': [ + 'test/import.yml' + ], 'installable': True, 'active': False, } diff --git a/base_location_geonames_import/test/import.yml b/base_location_geonames_import/test/import.yml new file mode 100644 index 000000000..a500e3db3 --- /dev/null +++ b/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'" diff --git a/base_location_geonames_import/wizard/geonames_import.py b/base_location_geonames_import/wizard/geonames_import.py index e74c42204..b4bba3924 100644 --- a/base_location_geonames_import/wizard/geonames_import.py +++ b/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 +# Copyright (C) 2014 Agile Business Group (http://www.agilebg.com) +# @author Lorenzo Battistini # # 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 '