Browse Source
Add module base_location_geonames_import
Add module base_location_geonames_import
Add support for states (if states are already present in res.country.state). Add POT file and FR translation. Allow to skip entries in the _prepare method. Insist on the fact that the wizard deletes current better zip entries in the module description and in the wizard itself. Better error message when the country code inside the file is wrong. Add a FOR UPDATE NOWAIT at the beginning of the transaction. Remove France-specific code ; moved to new module l10n_fr_base_location_geonames_import on lp:openerp-french-localization Add module base_location_geonames_import Add support for states (if states are already present in res.country.state). Add POT file and FR translation. Allow to skip entries in the _prepare method. Insist on the fact that the wizard deletes current better zip entries in the module description and in the wizard itself. Better error message when the country code inside the file is wrong. Add a FOR UPDATE NOWAIT at the beginning of the transaction. Remove France-specific code ; moved to new module l10n_fr_base_location_geonames_import on lp:openerp-french-localizationpull/643/head
Alexis de Lattre
10 years ago
committed by
Pedro M. Baeza
Pedro M. Baeza
7 changed files with 413 additions and 0 deletions
-
23base_location_geonames_import/__init__.py
-
60base_location_geonames_import/__openerp__.py
-
68base_location_geonames_import/i18n/base_location_geonames_import.pot
-
68base_location_geonames_import/i18n/fr.po
-
23base_location_geonames_import/wizard/__init__.py
-
127base_location_geonames_import/wizard/geonames_import.py
-
44base_location_geonames_import/wizard/geonames_import_view.xml
@ -0,0 +1,23 @@ |
|||
# -*- encoding: utf-8 -*- |
|||
############################################################################## |
|||
# |
|||
# Base Location Geonames Import module for OpenERP |
|||
# Copyright (C) 2014 Akretion (http://www.akretion.com) |
|||
# @author Alexis de Lattre <alexis.delattre@akretion.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 |
|||
# published by the Free Software Foundation, either version 3 of the |
|||
# License, or (at your option) any later version. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU Affero General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU Affero General Public License |
|||
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|||
# |
|||
############################################################################## |
|||
|
|||
from . import wizard |
|||
@ -0,0 +1,60 @@ |
|||
# -*- encoding: utf-8 -*- |
|||
############################################################################## |
|||
# |
|||
# Base Location Geonames Import module for OpenERP |
|||
# Copyright (C) 2014 Akretion (http://www.akretion.com) |
|||
# @author Alexis de Lattre <alexis.delattre@akretion.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 |
|||
# published by the Free Software Foundation, either version 3 of the |
|||
# License, or (at your option) any later version. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU Affero General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU Affero General Public License |
|||
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|||
# |
|||
############################################################################## |
|||
|
|||
|
|||
{ |
|||
'name': 'Base Location Geonames Import', |
|||
'version': '0.1', |
|||
'category': 'Extra Tools', |
|||
'license': 'AGPL-3', |
|||
'summary': 'Import better zip entries from Geonames', |
|||
'description': """ |
|||
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,Odoo Community Association (OCA)", |
|||
'website': 'http://www.akretion.com', |
|||
'depends': ['base_location'], |
|||
'external_dependencies': {'python': ['requests', 'unicodecsv']}, |
|||
'data': [ |
|||
'wizard/geonames_import_view.xml', |
|||
], |
|||
'installable': True, |
|||
'active': False, |
|||
} |
|||
@ -0,0 +1,68 @@ |
|||
# Translation of OpenERP Server. |
|||
# This file contains the translation of the following modules: |
|||
# * base_location_geonames_import |
|||
# |
|||
msgid "" |
|||
msgstr "" |
|||
"Project-Id-Version: OpenERP Server 7.0\n" |
|||
"Report-Msgid-Bugs-To: \n" |
|||
"POT-Creation-Date: 2014-04-11 20:55+0000\n" |
|||
"PO-Revision-Date: 2014-04-11 20:55+0000\n" |
|||
"Last-Translator: <>\n" |
|||
"Language-Team: \n" |
|||
"MIME-Version: 1.0\n" |
|||
"Content-Type: text/plain; charset=UTF-8\n" |
|||
"Content-Transfer-Encoding: \n" |
|||
"Plural-Forms: \n" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
msgid "Cancel" |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: field:better.zip.geonames.import,country_id:0 |
|||
msgid "Country" |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:66 |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:90 |
|||
#, python-format |
|||
msgid "Error:" |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
msgid "For the country selected above, this wizard 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." |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:91 |
|||
#, python-format |
|||
msgid "Got an error %d when trying to download the file %s." |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
msgid "Import" |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: model:ir.model,name:base_location_geonames_import.model_better_zip_geonames_import |
|||
msgid "Import Better Zip from Geonames" |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
#: model:ir.actions.act_window,name:base_location_geonames_import.better_zip_geonames_import_action |
|||
#: model:ir.ui.menu,name:base_location_geonames_import.better_zip_geonames_import_menu |
|||
msgid "Import Geonames" |
|||
msgstr "" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:67 |
|||
#, python-format |
|||
msgid "The country code inside the file (%s) doesn't correspond to the selected country (%s)." |
|||
msgstr "" |
|||
|
|||
@ -0,0 +1,68 @@ |
|||
# Translation of OpenERP Server. |
|||
# This file contains the translation of the following modules: |
|||
# * base_location_geonames_import |
|||
# |
|||
msgid "" |
|||
msgstr "" |
|||
"Project-Id-Version: OpenERP Server 7.0\n" |
|||
"Report-Msgid-Bugs-To: \n" |
|||
"POT-Creation-Date: 2014-04-11 20:56+0000\n" |
|||
"PO-Revision-Date: 2014-04-11 20:56+0000\n" |
|||
"Last-Translator: <>\n" |
|||
"Language-Team: \n" |
|||
"MIME-Version: 1.0\n" |
|||
"Content-Type: text/plain; charset=UTF-8\n" |
|||
"Content-Transfer-Encoding: \n" |
|||
"Plural-Forms: \n" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
msgid "Cancel" |
|||
msgstr "Annuler" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: field:better.zip.geonames.import,country_id:0 |
|||
msgid "Country" |
|||
msgstr "Pays" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:66 |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:90 |
|||
#, python-format |
|||
msgid "Error:" |
|||
msgstr "Erreur :" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
msgid "For the country selected above, this wizard 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." |
|||
msgstr "Pour le pays sélectionné ci-dessus, cet assistant va SUPPRIMER TOUTES LES ENTREES BETTER ZIP, télécharger la dernière version de la liste des villes depuis geonames.org et créer de nouveaux enregistrements better zip." |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:91 |
|||
#, python-format |
|||
msgid "Got an error %d when trying to download the file %s." |
|||
msgstr "Erreur %d reçue suite à la tentative de téléchargement du fichier %s." |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
msgid "Import" |
|||
msgstr "Importer" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: model:ir.model,name:base_location_geonames_import.model_better_zip_geonames_import |
|||
msgid "Import Better Zip from Geonames" |
|||
msgstr "Import Better Zip from Geonames" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: view:better.zip.geonames.import:0 |
|||
#: model:ir.actions.act_window,name:base_location_geonames_import.better_zip_geonames_import_action |
|||
#: model:ir.ui.menu,name:base_location_geonames_import.better_zip_geonames_import_menu |
|||
msgid "Import Geonames" |
|||
msgstr "Importer Geonames" |
|||
|
|||
#. module: base_location_geonames_import |
|||
#: code:addons/base_location_geonames_import/wizard/geonames_import.py:67 |
|||
#, python-format |
|||
msgid "The country code inside the file (%s) doesn't correspond to the selected country (%s)." |
|||
msgstr "Le code pays utilisé à l'intérieur du fichier (%s) ne correspond pas au pays sélectionné (%s)." |
|||
|
|||
@ -0,0 +1,23 @@ |
|||
# -*- encoding: utf-8 -*- |
|||
############################################################################## |
|||
# |
|||
# Base Location Geonames Import module for OpenERP |
|||
# Copyright (C) 2014 Akretion (http://www.akretion.com) |
|||
# @author Alexis de Lattre <alexis.delattre@akretion.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 |
|||
# published by the Free Software Foundation, either version 3 of the |
|||
# License, or (at your option) any later version. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU Affero General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU Affero General Public License |
|||
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|||
# |
|||
############################################################################## |
|||
|
|||
from . import geonames_import |
|||
@ -0,0 +1,127 @@ |
|||
# -*- encoding: utf-8 -*- |
|||
############################################################################## |
|||
# |
|||
# Base Location Geonames Import module for OpenERP |
|||
# Copyright (C) 2014 Akretion (http://www.akretion.com) |
|||
# @author Alexis de Lattre <alexis.delattre@akretion.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 |
|||
# published by the Free Software Foundation, either version 3 of the |
|||
# License, or (at your option) any later version. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU Affero General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU Affero General Public License |
|||
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|||
# |
|||
############################################################################## |
|||
|
|||
from openerp.osv import orm, fields |
|||
from openerp.tools.translate import _ |
|||
import requests |
|||
import tempfile |
|||
import StringIO |
|||
import unicodecsv |
|||
import zipfile |
|||
import os |
|||
import logging |
|||
|
|||
logger = logging.getLogger(__name__) |
|||
|
|||
|
|||
class better_zip_geonames_import(orm.TransientModel): |
|||
_name = 'better.zip.geonames.import' |
|||
_description = 'Import Better Zip from Geonames' |
|||
|
|||
_columns = { |
|||
'country_id': fields.many2one('res.country', 'Country', required=True), |
|||
} |
|||
|
|||
def _prepare_better_zip( |
|||
self, cr, uid, row, country_id, states, context=None): |
|||
'''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()] |
|||
vals = { |
|||
'name': row[1], |
|||
'city': row[2], |
|||
'state_id': state_id, |
|||
'country_id': country_id, |
|||
} |
|||
return vals |
|||
|
|||
def create_better_zip( |
|||
self, cr, uid, row, country_id, country_code, states, |
|||
context=None): |
|||
bzip_id = False |
|||
if row[0] != country_code: |
|||
raise orm.except_orm( |
|||
_('Error:'), |
|||
_("The country code inside the file (%s) doesn't " |
|||
"correspond to the selected country (%s).") |
|||
% (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( |
|||
cr, uid, row, country_id, states, context=context) |
|||
if vals: |
|||
bzip_id = self.pool['res.better.zip'].create( |
|||
cr, uid, vals, context=context) |
|||
return bzip_id |
|||
|
|||
def run_import(self, cr, uid, ids, context=None): |
|||
assert len(ids) == 1, 'Only one ID for the better zip import wizard' |
|||
bzip_obj = self.pool['res.better.zip'] |
|||
wizard = self.browse(cr, uid, ids[0], context=context) |
|||
country_id = wizard.country_id.id |
|||
country_code = wizard.country_id.code.upper() |
|||
url = 'http://download.geonames.org/export/zip/%s.zip' % country_code |
|||
logger.info('Starting to download %s' % url) |
|||
res_request = requests.get(url) |
|||
if res_request.status_code != requests.codes.ok: |
|||
raise orm.except_orm( |
|||
_('Error:'), |
|||
_('Got an error %d when trying to download the file %s.') |
|||
% (res_request.status_code, url)) |
|||
bzip_ids_to_delete = bzip_obj.search( |
|||
cr, uid, [('country_id', '=', country_id)], context=context) |
|||
if bzip_ids_to_delete: |
|||
cr.execute('SELECT id FROM res_better_zip WHERE id in %s ' |
|||
'FOR UPDATE NOWAIT', (tuple(bzip_ids_to_delete), )) |
|||
bzip_obj.unlink(cr, uid, bzip_ids_to_delete, context=context) |
|||
logger.info( |
|||
'%d better zip entries deleted for country %s' |
|||
% (len(bzip_ids_to_delete), wizard.country_id.name)) |
|||
state_ids = self.pool['res.country.state'].search( |
|||
cr, uid, [('country_id', '=', country_id)], context=context) |
|||
states = {} |
|||
# key = code of the state ; value = ID of the state in OpenERP |
|||
if state_ids: |
|||
states_r = self.pool['res.country.state'].read( |
|||
cr, uid, state_ids, ['code', 'country_id'], context=context) |
|||
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) |
|||
logger.info('The geonames zipfile has been decompressed') |
|||
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')) |
|||
for row in unicodecsv.reader( |
|||
data_file, encoding='utf-8', delimiter=' '): |
|||
self.create_better_zip( |
|||
cr, uid, row, country_id, country_code, states, |
|||
context=context) |
|||
data_file.close() |
|||
logger.info( |
|||
'The wizard to create better zip entries from geonames ' |
|||
'has been successfully completed.') |
|||
return True |
|||
@ -0,0 +1,44 @@ |
|||
<?xml version="1.0" encoding="utf-8"?> |
|||
<!-- |
|||
Copyright (C) 2014 Akretion (http://www.akretion.com/) |
|||
@author: Alexis de Lattre <alexis.delattre@akretion.com> |
|||
The licence is in the file __openerp__.py |
|||
--> |
|||
|
|||
<openerp> |
|||
<data> |
|||
|
|||
<record id="better_zip_geonames_import_form" model="ir.ui.view"> |
|||
<field name="name">asterisk.server.company</field> |
|||
<field name="model">better.zip.geonames.import</field> |
|||
<field name="arch" type="xml"> |
|||
<form string="Import Geonames" version="7.0"> |
|||
<group name="main"> |
|||
<field name="country_id"/> |
|||
<label string="For the country selected above, this wizard 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." |
|||
colspan="2" name="import-help"/> |
|||
</group> |
|||
<footer> |
|||
<button name="run_import" type="object" |
|||
class="oe_highlight" string="Import"/> |
|||
<button special="cancel" string="Cancel" class="oe_link"/> |
|||
</footer> |
|||
</form> |
|||
</field> |
|||
</record> |
|||
|
|||
<record id="better_zip_geonames_import_action" model="ir.actions.act_window"> |
|||
<field name="name">Import Geonames</field> |
|||
<field name="res_model">better.zip.geonames.import</field> |
|||
<field name="view_type">form</field> |
|||
<field name="view_mode">form</field> |
|||
<field name="target">new</field> |
|||
</record> |
|||
|
|||
<menuitem id="better_zip_geonames_import_menu" |
|||
action="better_zip_geonames_import_action" |
|||
parent="base.menu_localisation" |
|||
sequence="50"/> |
|||
|
|||
</data> |
|||
</openerp> |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue