CybroAddons/myfatoorah_payment_gateway/models/payment_transaction.py

# -*- coding: utf-8 -*-
##############################################################################
#
#    Cybrosys Technologies Pvt. Ltd.
#
#    Copyright (C) 2023-TODAY Cybrosys Technologies(<http://www.cybrosys.com>).
#    Author: Rahul CK(<https://www.cybrosys.com>)
#
#    You can modify it under the terms of the GNU AFFERO GENERAL
#    PUBLIC LICENSE (AGPL v3), Version 3.
#
#    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 (AGPL v3) for more details.
#
#    You should have received a copy of the GNU AFFERO GENERAL PUBLIC
#    LICENSE (AGPL v3) along with this program.
#    If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
import json
import logging
import requests
from odoo import _, models
from odoo.exceptions import ValidationError
# Import required libraries (make sure it is installed!)

_logger = logging.getLogger(__name__)


class PaymentTransaction(models.Model):
    """Extend and customize payment transactions for MyFatoorah integration."""
    _inherit = 'payment.transaction'

    def execute_payment(self):
        """Fetching data and Executing Payment"""
        base_api_url = self.env['payment.acquirer'].search(
            [('provider', '=', 'myfatoorah')])._myfatoorah_get_api_url()
        api_url = f"{base_api_url}v2/ExecutePayment"
        api_key = self.env['payment.acquirer'].search(
            [('provider', '=', 'myfatoorah')]).myfatoorah_token
        odoo_base_url = self.env['ir.config_parameter'].get_param(
            'web.base.url')
        sale_order = self.env['payment.transaction'].search(
            [('id', '=', self.id)]).sale_order_ids
        order_line = self.env['payment.transaction'].search(
            [('id', '=', self.id)]).sale_order_ids.order_line
        invoice_items = [
            {
                'ItemName': rec.product_id.name,
                'Quantity': int(rec.product_uom_qty),
                'UnitPrice': rec.price_unit,
            }
            for rec in order_line
        ]
        MobileCountryCode = self.partner_id.country_id.phone_code
        phone_number = self.partner_phone
        if not phone_number:
            raise ValueError("Please provide the phone number.")
        if phone_number:
            phone_number = phone_number.replace(str(MobileCountryCode), '')
            if phone_number.startswith('+'):
                phone_number = phone_number[1:]
        payment_details = {
            "PaymentMethodId": 6,
            "CustomerName": self.partner_name,
            "DisplayCurrencyIso": self.currency_id.name,
            "MobileCountryCode": MobileCountryCode,
            "CustomerMobile": phone_number,
            "CustomerEmail": self.partner_email,
            "InvoiceValue": (self.amount - sale_order.amount_tax),
            "CallBackUrl": f"{odoo_base_url}/payment/myfatoorah/_return_url",
            "ErrorUrl": f"{odoo_base_url}/payment/myfatoorah/failed",
            "Language": "en",
            "CustomerReference": self.reference,
            "CustomerAddress": {
                "Address": f'{self.partner_address},'
                           f'{self.partner_city}, {self.partner_zip},'
                           f'{self.partner_country_id.name}',
            },
            "InvoiceItems":
                invoice_items
        }
        headers = {
            'Content-Type': 'application/json',
            'Accept': 'application/json',
            'Authorization': f'Bearer {api_key}',
        }
        payload = json.dumps(payment_details)
        response = requests.request("POST", api_url, headers=headers,
                                    data=payload)
        response_data = response.json()
        payment_url = response_data.get('Data')['PaymentURL']
        payment_details['PaymentURL'] = payment_url
        return payment_details
    def _get_tx_from_notification_data(self, provider, notification_data):
        """Getting  payment status from myfatoorah"""
        api_key = self.env['payment.acquirer'].search(
            [('provider', '=', 'myfatoorah')]).myfatoorah_token
        base_api_url = self.env['payment.acquirer'].search(
            [('provider', '=', 'myfatoorah')])._myfatoorah_get_api_url()
        url = f"{base_api_url}v2/GetPaymentStatus"
        paymentid = notification_data.get('paymentId')
        payload = json.dumps({
            "Key": f"{paymentid}",
            "KeyType": "paymentId"
        })
        headers = {
            'Content-Type': 'application/json',
            'Accept': 'application/json',
            'Authorization': f'Bearer {api_key}',
        }
        response = requests.request("POST", url, headers=headers, data=payload)
        response_data = response.json()
        tx = super()._get_tx_from_notification_data(provider,
                                                    notification_data)
        if provider != 'myfatoorah' or len(tx) == 1:
            return tx
        reference = response_data["Data"]["CustomerReference"]
        tx = self.search(
            [
                ('reference', '=', reference),
                ('provider', '=', 'myfatoorah')])
        if not tx:
            raise ValidationError(
                "myfatoorah: " + _(
                    "No transaction found matching reference %s.",
                    reference)
            )
        return tx

    def _handle_notification_data(self, provider, notification_data):
        """Handle notification data from MyFatoorah and process transaction."""
        tx = self._get_tx_from_notification_data(provider,
                                                 notification_data)
        tx._process_notification_data(notification_data)
        tx._execute_callback()
        return tx

    def _process_notification_data(self, notification_data):
        """Process notification data, specifically for MyFatoorah transactions."""
        super()._process_notification_data(notification_data)
        if self.provider != 'myfatoorah':
            return
        else:
            self._set_done()