# Part of Odoo. See LICENSE file for full copyright and licensing details.
import datetime
from dateutil.relativedelta import relativedelta

from odoo import fields, models, _

from odoo.exceptions import UserError
from odoo.tools import format_date


class ResCurrency(models.Model):
    _inherit = "res.currency"

    def _l10n_ar_get_last_business_day_rate(self, afip_ws='wsfe', from_date=None):
        """ Get last business rate from a given date """
        # Last  ultimo dia habil
        arca_rate = False
        arca_date = False
        max_tries = try_count = 15
        while not arca_rate and try_count:
            arca_date, arca_rate = self._l10n_ar_get_afip_ws_currency_rate(afip_ws, date_rate=from_date)
            if not arca_rate:
                from_date = from_date - relativedelta(days=1)
            try_count -= 1

        if try_count == 0:
            raise UserError(_(
                'Did not find any rate last %(max_tries)s days before the given date (%(given_date)s)',
                max_tries=max_tries,
                given_date=from_date or fields.Date.context_today(self),
            ))

        arca_date = format_date(
            self.env, datetime.datetime.strptime(arca_date, '%Y%m%d'), date_format='EEEE, dd MMMM YYYY')
        return arca_date, arca_rate

    def l10n_ar_action_get_afip_ws_currency_rate(self):
        date, rate = self._l10n_ar_get_last_business_day_rate()
        raise UserError(_('Last Business Day: %(date)s\nRate: %(rate)s', date=date, rate=rate))

    def _l10n_ar_get_afip_ws_currency_rate(self, afip_ws='wsfe', date_rate=None):
        """ Return the date and rate for a given currency
        This is only for the user so that he can quickly check the last rate on afip por a currency.
        This is really useful. There is a NTH for the future to integrate this with the automtaic currency rates """
        self.ensure_one()
        if not self.l10n_ar_afip_code:
            raise UserError(_('No ARCA code for currency %s. Please configure the ARCA code consulting information in ARCA page', self.name))
        if self.l10n_ar_afip_code == 'PES':
            raise UserError(_('No rate for ARS (is the base currency for ARCA)'))

        connection = self.env.company._l10n_ar_get_connection(afip_ws)
        client, auth = connection._get_client()
        if afip_ws == 'wsfe':
            req_data = {"MonId": self.l10n_ar_afip_code}
            if date_rate:
                req_data["FchCotiz"] = date_rate.strftime("%Y%m%d")

            response = client.service.FEParamGetCotizacion(auth, **req_data)
            if response.Errors:
                if response.Errors.Err[0].Code == 602:  # Not found rate for the given date
                    return date_rate, False
                raise UserError(_(
                    'The was an error obtaining the rate:\n\n * Code %(error_code)s -  %(error_message)s',
                    error_code=response.Errors.Err[0].Code,
                    error_message=response.Errors.Err[0].Msg,
                ))

            # Events None
            date = response.ResultGet.FchCotiz
            rate = response.ResultGet.MonCotiz
        elif afip_ws == 'wsfex':
            req_data = {"Mon_id": self.l10n_ar_afip_code}
            if date_rate:
                req_data["FchCotiz"] = date_rate.strftime("%Y-%m-%d")

            response = client.service.FEXGetPARAM_Ctz(auth, **req_data)
            if response.FEXErr.ErrCode != 0:  # ErrCode is always send, if == 0 everything is ok
                if response.FEXErr.ErrCode == 1800:  # Not found rate for the given date
                    return date_rate, False
                raise UserError(_(
                    'The was an error obtaining the rate:\n\n * Code %(error_code)s -  %(error_message)s',
                    error_code=response.FEXErr.ErrCode,
                    error_message=response.FEXErr[0].ErrMsg,
                ))
            date = response.FEXResultGet.Mon_fecha
            rate = float(response.FEXResultGet.Mon_ctz)  # WS returns Decimal() type
        elif afip_ws == 'wsbfe':
            req_data = {"Mon_id": self.l10n_ar_afip_code}
            if date_rate:
                req_data["FchCotiz"] = date_rate.strftime("%Y%m%d")

            response = client.service.BFEGetCotizacion(auth, **req_data)
            if response.BFEErr.ErrCode != 0:  # ErrCode is always send, if == 0 everything is ok
                if response.BFEErr.ErrCode == 4964:  # Not found rate for the given date
                    return date_rate, False
                raise UserError(_(
                    'The was an error obtaining the rate:\n\n * Code %(error_code)s -  %(error_message)s',
                    error_code=response.BFEErr.ErrCode,
                    error_message=response.BFEErr[0].ErrMsg,
                ))
            date = response.BFEResultGet.FchCotiz
            rate = float(response.BFEResultGet.MonCotiz)  # WS returns Decimal() type

        else:
            raise UserError(_(
                'Get ARCA currency rate not implemented for webservice %(afip_ws)s',
                afip_ws=afip_ws,
            ))
        return date, rate