~altegra/pepess/pepess_dev_carlos_scripts

#!/usr/bin/env python

#-*- coding:utf-8 -*-

 

#############################################################################

#    Module Writen to OpenERP, Open Source Management Solution

#    CopyLeft 2012 - http://www.grupoaltegra.com

#    You are free to share, copy, distribute, transmit, adapt and use for commercial purpose

#    More information about license: http://www.gnu.org/licenses/agpl.html

#    info Grupo Altegra (openerp@grupoaltegra.com)

#

#############################################################################

#

#    Coded by: Carlos Blanco (carlos.blanco@grupoaltegra.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/>.

#

#############################################################################

 

import re

from openerp.osv import fields, osv

from openerp.tools.translate import _

import string

 

_ref_vat = {

    'at': 'ATU12345675',

    'be': 'BE0477472701',

    'bg': 'BG1234567892',

    'ch': 'CHE-123.456.788 TVA or CH TVA 123456', #Swiss by Yannick Vaucher @ Camptocamp

    'cy': 'CY12345678F',

    'cz': 'CZ12345679',

    'de': 'DE123456788',

    'dk': 'DK12345674',

    'ee': 'EE123456780',

    'el': 'EL12345670',

    'es': 'ESA12345674',

    'fi': 'FI12345671',

    'fr': 'FR32123456789',

    'gb': 'GB123456782',

    'gr': 'GR12345670',

    'hu': 'HU12345676',

    'hr': 'HR01234567896', # Croatia, contributed by Milan Tribuson 

    'ie': 'IE1234567T',

    'it': 'IT12345670017',

    'lt': 'LT123456715',

    'lu': 'LU12345613',

    'lv': 'LV41234567891',

    'mt': 'MT12345634',

    'mx': 'MXABC123456T1B',

    'nl': 'NL123456782B90',

    'no': 'NO123456785',

    'pl': 'PL1234567883',

    'pt': 'PT123456789',

    'ro': 'RO1234567897',

    'se': 'SE123456789701',

    'si': 'SI12345679',

    'sk': 'SK0012345675',

}

 

class res_partner(osv.Model):

    _inherit = 'res.partner'

    

    def check_vat(self, cr, uid, ids, context=None):

        vat_list = []

        #~ return all( [ self.check_vat_mx( partner.vat ) for partner in self.browse(cr, uid, ids, context=context) ] )

        # Se Verifica si tiene el codigo de pais agregado en el RFC

        for partner in self.browse(cr, uid, ids, context=context):

            vat = partner.vat

            if partner.vat:

                country_code = partner.vat[:2]

                # Si tiene el codigo de pais se quita y solo se toma n cuenta el RFC

                if country_code.count('MX'):

                    vat = partner.vat[2:]

                # Si no si manda todo el RFC sin quitarne nada para validarlo

                else:

                    vat = partner.vat

                vat_list.append(self.check_vat_mx( vat ))

            else:

                vat_list.append(True)

        return all(vat_list)

    

    def _construct_constraint_msg(self, cr, uid, ids, context=None):

        # Se construye el mensaje que se mandara si el RFC es incorrecto

        def default_vat_check(cn, vn):

            # by default, a VAT number is valid if:

            #  it starts with 2 letters

            #  has more than 3 characters

            return cn[0] in string.ascii_lowercase and cn[1] in string.ascii_lowercase

        vat_country, vat_number = self._split_vat(self.browse(cr, uid, ids)[0].vat)

        vat_no = "'CC##' (CC=Country Code, ##=VAT Number)"

        if default_vat_check(vat_country, vat_number):

            vat_no = _ref_vat[vat_country] if vat_country in _ref_vat else vat_no

        return '\n' + _('This VAT number does not seem to be valid.\nNote: the expected format is %s') % vat_no

    

    _constraints = [(check_vat, _construct_constraint_msg, ["vat"])]

    

    def write(self, cr, uid, ids, vals, context=None):

        """

        Se redefine el metodo write para ponerle el codigo de pais al principio del RFC

        """

        if context is None:

            context = {}

        if 'vat' in vals:

            if vals['vat']:

                vals['vat'] = vals['vat'] and vals['vat'].replace(u'-', u'').replace(u' ', u'').replace(u'.', u'').replace(u'/', u'').replace(u'ñ', u'Ñ').upper() or vals['vat']

                country_code = vals['vat'][:2]

                if not country_code.count('MX'):

                    vals['vat'] = 'MX'+vals['vat']

        return super(res_partner, self).write(cr, uid, ids, vals, context=context)