ingadhoc/odoo-argentina

# -*- coding: utf-8 -*-
from openerp import models, fields, api, _
import openerp.tools as tools
from pyafipws.iibb import IIBB
# from pyafipws.padron import PadronAFIP
from openerp.exceptions import Warning
import logging
from dateutil.relativedelta import relativedelta
_logger = logging.getLogger(__name__)


class ResCompany(models.Model):
    _inherit = "res.company"

    arba_cit = fields.Char(
        'CIT ARBA',
        help='Clave de Identificación Tributaria de ARBA',
    )

    @api.model
    def _get_arba_environment_type(self):
        """
        Function to define homologation/production environment
        First it search for a paramter "arba.ws.env.type" if exists and:
        * is production --> production
        * is homologation --> homologation
        Else
        Search for 'server_mode' parameter on conf file. If that parameter is:
        * 'test' or 'develop' -->  homologation
        * other or no parameter -->  production
        """
        parameter_env_type = self.env[
            'ir.config_parameter'].get_param('arba.ws.env.type')
        if parameter_env_type == 'production':
            environment_type = 'production'
        elif parameter_env_type == 'homologation':
            environment_type = 'homologation'
        else:
            server_mode = tools.config.get('server_mode')
            if not server_mode or server_mode == 'production':
                environment_type = 'production'
            else:
                environment_type = 'homologation'
        _logger.info(
            'Running arba WS on %s mode' % environment_type)
        return environment_type

    @api.model
    def get_arba_login_url(self, environment_type):
        if environment_type == 'production':
            arba_login_url = (
                'https://dfe.arba.gov.ar/DomicilioElectronico/'
                'SeguridadCliente/dfeServicioConsulta.do')
        else:
            arba_login_url = (
                'https://dfe.test.arba.gov.ar/DomicilioElectronico'
                '/SeguridadCliente/dfeServicioConsulta.do')
        return arba_login_url

    @api.multi
    def arba_connect(self):
        """
        Method to be called
        """
        self.ensure_one()
        cuit = self.partner_id.document_number

        if not cuit:
            raise Warning(_(
                'You must configure CUIT con company %s related partner') % (
                    self.name))
        if not self.arba_cit:
            raise Warning(_(
                'You must configure ARBA CIT on company %s') % (
                    self.name))

        ws = IIBB()
        environment_type = self._get_arba_environment_type()
        _logger.info(
            'Getting connection to ARBA on %s mode' % environment_type)

        # argumentos de conectar: self, url=None, proxy="",
        # wrapper=None, cacert=None, trace=False, testing=""
        arba_url = self.get_arba_login_url(environment_type)
        ws.Usuario = cuit
        ws.Password = self.arba_cit
        ws.Conectar(url=self.get_arba_login_url(environment_type))
        _logger.info(
            'Connection getted to ARBA with url "%s" and CUIT %s' % (
                arba_url, cuit))
        return ws

    @api.model
    def get_arba_data(self, partner, date):
        self.ensure_one()

        from_date = date + relativedelta(day=1)
        to_date = date + relativedelta(day=1, days=-1, months=+1)

        cuit = partner.document_number
        if not cuit:
            raise Warning(_('No CUIT for partner %s') % (
                partner.name))

        _logger.info(
            'Getting ARBA data for cuit %s from date %s to date %s' % (
                from_date, to_date, cuit))
        ws = self.arba_connect()
        ws.ConsultarContribuyentes(
            from_date.strftime('%Y%m%d'),
            to_date.strftime('%Y%m%d'),
            cuit)

        if ws.Excepcion:
            raise Warning("%s\nExcepcion: %s" % (
                ws.Traceback, ws.Excepcion))

        # ' Hubo error general de ARBA?
        if ws.CodigoError:
            raise Warning("%s\nError %s: %s" % (
                ws.MensajeError, ws.TipoError, ws.CodigoError))

        # ' Datos generales de la respuesta:'
        data = {
            'NumeroComprobante': ws.NumeroComprobante,
            'CodigoHash': ws.CodigoHash,
            'CuitContribuyente': ws.CuitContribuyente,
            'AlicuotaPercepcion': ws.AlicuotaPercepcion,
            'AlicuotaRetencion': ws.AlicuotaRetencion,
            'GrupoPercepcion': ws.GrupoPercepcion,
            'GrupoRetencion': ws.GrupoRetencion,
        }
        _logger.info('We get the following data: \n%s' % data)
        return data