currency.py

# This file is part of the account_invoice_ar module for Tryton.
# The COPYRIGHT file at the top level of this repository contains
# the full copyright notices and license terms.
import logging
from decimal import Decimal
import datetime
from pyafipws.wsfev1 import WSFEv1
from pyafipws.wsfexv1 import WSFEXv1

from trytond.model import fields
from trytond.pool import Pool, PoolMeta
from trytond.transaction import Transaction
from trytond.exceptions import UserError
from trytond.i18n import gettext

logger = logging.getLogger(__name__)


class Currency(metaclass=PoolMeta):
    __name__ = 'currency.currency'

    afip_code = fields.Char('AFIP Code', size=3,
        help="The 3 digits AFIP currency code.")

    @classmethod
    def compute(cls, from_currency, amount, to_currency, round=True):
        pool = Pool()
        Company = pool.get('company.company')

        currency_rate = Transaction().context.get('currency_rate')
        if not currency_rate:
            return super().compute(from_currency, amount, to_currency, round)

        if to_currency == from_currency:
            if round:
                return to_currency.round(amount)
            else:
                return amount

        company = Company(Transaction().context['company'])
        if from_currency == company.currency:
            from_currency_rate = currency_rate
            currency_rate = Decimal('1.0')
        else:
            from_currency_rate = Decimal('1.0')

        if round:
            return to_currency.round(
                amount * currency_rate / from_currency_rate)
        else:
            return amount * currency_rate / from_currency_rate


class Rate(metaclass=PoolMeta):
    __name__ = 'currency.currency.rate'

    def get_afip_rate(self, service='wsfex'):
        '''
        get rate from afip webservice.
        '''
        pool = Pool()
        Company = pool.get('company.company')
        company_id = Transaction().context.get('company')
        if not company_id:
            logger.error('The company is not defined')
            raise UserError(gettext(
                'account_invoice_ar.msg_company_not_defined'))
        company = Company(company_id)
        # authenticate against AFIP:
        ta = company.pyafipws_authenticate(service=service)

        if service == 'wsfe':
            ws = WSFEv1()
            if company.pyafipws_mode_cert == 'homologacion':
                WSDL = 'https://wswhomo.afip.gov.ar/wsfev1/service.asmx?WSDL'
            elif company.pyafipws_mode_cert == 'produccion':
                WSDL = (
                    'https://servicios1.afip.gov.ar/wsfev1/service.asmx?WSDL')
        elif service == 'wsfex':
            ws = WSFEXv1()
            if company.pyafipws_mode_cert == 'homologacion':
                WSDL = 'https://wswhomo.afip.gov.ar/wsfexv1/service.asmx?WSDL'
            elif company.pyafipws_mode_cert == 'produccion':
                WSDL = (
                    'https://servicios1.afip.gov.ar/wsfexv1/service.asmx?WSDL')
        else:
            logger.critical('AFIP ws is not yet supported! %s', service)
            raise UserError(gettext(
                'account_invoice_ar.msg_webservice_not_supported',
                service=service))

        cache = Company.get_cache_dir()
        ws.LanzarExcepciones = True
        try:
            ws.Conectar(wsdl=WSDL, cache=cache, cacert=True)
        except Exception as e:
            msg = ws.Excepcion + ' ' + str(e)
            logger.error('WSAA connecting to afip: %s' % msg)
            raise UserError(gettext(
                'account_invoice_ar.msg_wsaa_error', msg=msg))
        ws.SetTicketAcceso(ta)
        ws.Cuit = company.party.vat_number

        if not self.currency.afip_code:
            logger.error('AFIP code is empty %s', self.currency.code)
            raise UserError(gettext(
                'account_invoice_ar.msg_afip_code_empty'))

        self.rate = Decimal(ws.GetParamCtz('DOL'))
        self.date = datetime.datetime.strptime(ws.FchCotiz, '%Y%m%d').date()