← Back to branch summary

~openerp-spain-team/openerp-spain/6.0-git

« back to all changes in this revision

Viewing changes to l10n_es_toponyms/l10n_es_toponyms.py

  • Committer: Borja L.S.
  • Date: 2010-10-18 10:04:25 UTC
  • Revision ID: git-v1:271c47a993616dbba60585d48b8b98d603199d93
[REF] *: Refactorización para portar a 6.0 - Paso 1.

- Se han renombrado los módulos para usar la nomenclatura propuesta
  por OpenERP: l10n_es para el módulo base de localización (plan de 
  cuentas), l10n_es_* para el resto de módulos.

- Se eliminan los módulos extra_addons/* que deberían moverse a 
  los extra-addons genéricos (no son específicos de España).

- Se renombran los __terp__.py por __openerp__.py

Show diffs side-by-side

added added

removed removed

Lines of Context:
l10n_es_toponyms/l10n_es_toponyms.py
 
1
# -*- encoding: utf-8 -*-
 
2
##############################################################################
 
3
#
 
4
#    OpenERP, Open Source Management Solution
 
5
#    Copyright (c) 2009 Zikzakmedia S.L. (http://zikzakmedia.com) All Rights Reserved.
 
6
#                       Jordi Esteve <jesteve@zikzakmedia.com>
 
7
#    $Id$
 
8
#
 
9
#    This program is free software: you can redistribute it and/or modify
 
10
#    it under the terms of the GNU General Public License as published by
 
11
#    the Free Software Foundation, either version 3 of the License, or
 
12
#    (at your option) any later version.
 
13
#
 
14
#    This program is distributed in the hope that it will be useful,
 
15
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
16
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
17
#    GNU General Public License for more details.
 
18
#
 
19
#    You should have received a copy of the GNU General Public License
 
20
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
21
#
 
22
##############################################################################
 
23
 
 
24
from osv import fields,osv
 
25
import os
 
26
import ir
 
27
import pooler
 
28
import tools
 
29
import threading
 
30
 
 
31
class config_ES_toponyms(osv.osv_memory):
 
32
    _name='config.ES.toponyms'
 
33
 
 
34
    def _city_module_default(self, cr, uid, context=None):
 
35
        cr.execute('select * from ir_module_module where name=%s and state=%s', ('city','installed'))
 
36
        if cr.fetchone():
 
37
            return 'installed'
 
38
        else:
 
39
            return 'uninstalled'
 
40
 
 
41
    def _city_info_recover_default(self, cr, uid, context=None):
 
42
        # City info can be selected if zip field exists in the database. That will happen only
 
43
        # if module city wasn't installed by the initial profile.
 
44
        cr.execute("select * from information_schema.columns where table_name='res_partner_address' and table_schema='public' and column_name='zip'")
 
45
        if cr.fetchone():
 
46
            return 'yes'
 
47
        else:
 
48
            return 'no'
 
49
 
 
50
    _columns = {
 
51
        'name':fields.char('Name', size=64),
 
52
        'state': fields.selection([('official','Official'),('spanish','Spanish'),('both','Both')], 'State names', required=True, help="Toponym version of the Spanish states. For example: Official (Girona), Spanish (Gerona), Both (Gerona / Girona)"),
 
53
        'city_module': fields.selection([('installed','Installed'),('uninstalled','Uninstalled')], 'City module', readonly=True, help="City module is optional and groups state, city and zip code in one entity."),
 
54
        'city_info': fields.selection([('yes','Yes'),('no','No')], 'City information', required=True, help="Do you want to add city and state information associated to the zip codes for all the Spanish cities? This allows to fill automatically the city and states fields of partner and contact forms from the zip code."),
 
55
        'city_info_recover': fields.selection([('yes','Yes'),('no','No')], 'City information recover', required=True, help="If the city module has been installed, do you want to recover the location data (city information) from the zip code there was in the partner addresses before installing the city module?"),
 
56
    }
 
57
 
 
58
    _defaults={
 
59
        'state': lambda *args: 'official',
 
60
        'city_module': _city_module_default,
 
61
        'city_info': lambda *args: 'yes',
 
62
        'city_info_recover': _city_info_recover_default,
 
63
    }
 
64
 
 
65
    def onchange_city_info(self, cr, uid, ids, city_info, city_module):
 
66
        if city_module == 'uninstalled':
 
67
            v = {'city_info_recover': 'no'}
 
68
        elif city_info == 'yes':
 
69
            v = {
 
70
                'city_info_recover': self._city_info_recover_default(cr, uid)
 
71
            }
 
72
        return {'value':v}
 
73
 
 
74
    def _onchange_city_info(self, cr, uid, ids, city_info, city_module):
 
75
        """onchange_city_info alias for backwards compatibility"""
 
76
        return self.onchange_city_info(cr, uid, ids, city_info, city_module)
 
77
 
 
78
    def _create_defaults(self, cr, uid, context):
 
79
        # Creates default values of state and city res.partner.address fields linked to zip codes
 
80
        from municipios_cpostal import cod_postales
 
81
        pool = pooler.get_pool(cr.dbname)
 
82
        idc = pool.get('res.country').search(cr, uid, [('code', '=', 'ES'),])
 
83
        if not idc:
 
84
            return
 
85
        idc = idc[0]
 
86
        for m in cod_postales:
 
87
            ids = pool.get('res.country.state').search(cr, uid, [('country_id', '=', idc), ('code', '=', m[0][:2]),])
 
88
            if ids:
 
89
                ir.ir_set(cr, uid, 'default', 'zip='+m[0], 'state_id', [('res.partner.address', False)], ids[0])
 
90
            ir.ir_set(cr, uid, 'default', 'zip='+m[0], 'city', [('res.partner.address', False)], m[1])
 
91
        return {}
 
92
 
 
93
 
 
94
    def _recover_zipcodes(self, cr, uid, context):
 
95
        # Recovers the location data (city info) from the zip code there was in the partner addresses before installing the city module
 
96
        cr.execute("select id, zip from res_partner_address where location IS NULL")
 
97
        zipcodes = cr.dictfetchall()
 
98
        cont = 0
 
99
        for zipcode in zipcodes:
 
100
            if zipcode['zip']:
 
101
                cr.execute("select id from city_city where zipcode = '%s'" %zipcode['zip'])
 
102
                city_id = cr.fetchall()
 
103
                if len(city_id) > 0:
 
104
                    cr.execute("update res_partner_address SET location = %i WHERE id = %i" %(city_id[0][0], zipcode['id']))
 
105
                    cont += 1
 
106
        return cont
 
107
 
 
108
 
 
109
    def create_zipcodes(self, db_name, uid, ids, res, context):
 
110
        # Import Spanish cities and zip codes (15000 zip codes can take several minutes)
 
111
        db, pool = pooler.get_db_and_pool(db_name)
 
112
        cr = db.cursor()
 
113
        if res['city_module'] == 'uninstalled': # city module no installed
 
114
            self._create_defaults(cr, uid, context)
 
115
        else:                                   # city module installed
 
116
            try:
 
117
                fp = tools.file_open(os.path.join('l10n_ES_toponyms', 'l10n_ES_toponyms_zipcodes.xml'))
 
118
            except IOError, e:
 
119
                fp = None
 
120
            if fp:
 
121
                idref = {}
 
122
                tools.convert_xml_import(cr, 'l10n_ES_toponyms', fp,  idref, 'init', noupdate=True)
 
123
                if res['city_info_recover'] == 'yes':
 
124
                    res= self._recover_zipcodes(cr, uid, context)
 
125
                    #print res
 
126
        cr.commit()
 
127
        cr.close()
 
128
        return {}
 
129
 
 
130
 
 
131
    def action_set(self, cr, uid, ids, context=None):
 
132
        res = self.read(cr, uid, ids)[0]
 
133
        #print res
 
134
 
 
135
        # Import Spanish states (official, Spanish or both)
 
136
        file_name = 'l10n_ES_toponyms_states_'+res['state']+'.xml'
 
137
        try:
 
138
            fp = tools.file_open(os.path.join('l10n_ES_toponyms', file_name))
 
139
        except IOError, e:
 
140
            fp = None
 
141
        if fp:
 
142
            idref = {}
 
143
            tools.convert_xml_import(cr, 'l10n_ES_toponyms', fp,  idref, 'init', noupdate=True)
 
144
 
 
145
        # Import Spanish cities and zip codes in other thread (15000 zip codes can take several minutes)
 
146
        if res['city_info'] == 'yes':
 
147
            cr.commit()
 
148
            thread1 = threading.Thread(target=self.create_zipcodes, args=(cr.dbname, uid, ids, res, context))
 
149
            thread1.start()
 
150
 
 
151
        return {
 
152
                'view_type': 'form',
 
153
                "view_mode": 'form',
 
154
                'res_model': 'ir.actions.configuration.wizard',
 
155
                'type': 'ir.actions.act_window',
 
156
                'target':'new',
 
157
            }
 
158
 
 
159
 
 
160
    def action_cancel(self, cr, uid, ids, conect=None):
 
161
        return {
 
162
                'view_type': 'form',
 
163
                "view_mode": 'form',
 
164
                'res_model': 'ir.actions.configuration.wizard',
 
165
                'type': 'ir.actions.act_window',
 
166
                'target':'new',
 
167
         }
 
168
 
 
169
config_ES_toponyms()