app.py

# pylint: disable=no-name-in-module
# pylint: disable=no-self-argument

from portalsdk import APIContext, APIMethodType, APIRequest
from time import sleep
from flask import Flask, jsonify, g, request
import uuid
from firebase_admin import credentials
from firebase_admin import initialize_app
from firebase_admin import firestore
import datetime
from pydantic import BaseModel
from flask_pydantic import validate
from flask_restful import Api, Resource, reqparse

cred = credentials.Certificate("./key.json")
initialize_app(cred)
# firebase database paths
firestoreDb = firestore.client()
subscriptions_ref = firestoreDb.collection('subscriptions')

# initialize a flask app
app = Flask(__name__)
api = Api(app)


class ResponseModel(BaseModel):
    responseCode: str
    output_ResponseDesc: str
    output_ThirdPartyConversationID: str


class FirestoreDataYearly(BaseModel):
    output_ConversationID: str
    output_ResponseCode: str
    output_ResponseDesc: str
    output_ThirdPartyConversationID: str
    output_TransactionID: str
    userId: str
    msisdn: str
    itemDesc: str
    amount: int
    typeOfSubscription: str
    school: str
    userName: str
    created: datetime.datetime

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        object.__setattr__(
            self, 'endDate', datetime.datetime.utcnow() + datetime.timedelta(days=365))


class FirestoreDataMonthly(BaseModel):
    output_ConversationID: str
    output_ResponseCode: str
    output_ResponseDesc: str
    output_ThirdPartyConversationID: str
    output_TransactionID: str
    userId: str
    msisdn: str
    itemDesc: str
    amount: int
    typeOfSubscription: str
    school: str
    userName: str
    created: datetime.datetime

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        object.__setattr__(
            self, 'endDate', datetime.datetime.utcnow() + datetime.timedelta(days=30))


class Silver(Resource):

    def __init__(self):
        self._uzd = ''
        super(Silver, self).__init__()

    @property
    def uzd(self):
        return self._uzd

    @uzd.setter
    def uzd(self, value=None):
        value = uuid.uuid4().hex
        self._uzd = value

    @uzd.deleter
    def uzd(self):
        del self._uzd

    def post(self):

        bodi = request.get_json(force=True)
        nu = bodi['msisdn']
        amt = bodi['amount']
        itm = bodi['itemDesc']
        usr = bodi['userId']
        typez = bodi['typeOfSubscription']
        schl = bodi['school']
        un = bodi['userName']

        # Public key on the API listener used to encrypt keys
        public_key = 'MIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEArv9yxA69XQKBo24BaF/D' \
            '+fvlqmGdYjqLQ5WtNBb5tquqGvAvG3WMFETVUSow' \
            '/LizQalxj2ElMVrUmzu5mGGkxK08bWEXF7a1DEvtVJs6nppIlFJc2SnrU14AOrIrB28ogm58JjAl5BOQawOXD5dfSk7MaAA82pVHoIqEu0FxA8BOKU+RGTihRU+ptw1j4bsAJYiPbSX6i71gfPvwHPYamM0bfI4CmlsUUR3KvCG24rB6FNPcRBhM3jDuv8ae2kC33w9hEq8qNB55uw51vK7hyXoAa+U7IqP1y6nBdlN25gkxEA8yrsl1678cspeXr+3ciRyqoRgj9RD/ONbJhhxFvt1cLBh+qwK2eqISfBb06eRnNeC71oBokDm3zyCnkOtMDGl7IvnMfZfEPFCfg5QgJVk1msPpRvQxmEsrX9MQRyFVzgy2CWNIb7c+jPapyrNwoUbANlN8adU1m6yOuoX7F49x+OjiG2se0EJ6nafeKUXw/+hiJZvELUYgzKUtMAZVTNZfT8jjb58j8GVtuS+6TM2AutbejaCV84ZK58E2CRJqhmjQibEUO6KPdD7oTlEkFy52Y1uOOBXgYpqMzufNPmfdqqqSM4dU70PO8ogyKGiLAIxCetMjjm6FCMEA3Kc8K0Ig7/XtFm9By6VxTJK1Mg36TlHaZKP6VzVLXMtesJECAwEAAQ== '
        # Create Context with API to request a Session ID
        api_context = APIContext()
        # Api key
        api_context.api_key = '7OCA9qYWyAdaajBsrm46hgwANyqeUGfG'
        # Public key
        api_context.public_key = public_key
        # Use ssl/https
        api_context.ssl = True
        # Method type (can be GET/POST/PUT)
        api_context.method_type = APIMethodType.GET
        # API address
        api_context.address = 'openapi.m-pesa.com'
        # API Port
        api_context.port = 443
        # API Path
        api_context.path = '/sandbox/ipg/v2/vodacomTZN/getSession/'

        # Add/update headers
        api_context.add_header('Origin', '3.8.36.235')

        # Parameters can be added to the call as well that on POST will be in JSON format and on GET will be URL parameters
        # api_context.add_parameter('key', 'value')

        # Do the API call and put result in a response packet
        api_request = APIRequest(api_context)

        # Do the API call and put result in a response packet
        result = None
        try:
            result = api_request.execute()
        except Exception as e:
            print('Call Failed: ' + str(e))

        if result is None:
            raise Exception(
                'SessionKey call failed to get result. Please check.')

        # Display results
        print(result.status_code)
        # print(result.headers)
        print(result.body)

        # The above call issued a sessionID which will be used as the API key in calls that needs the sessionID

        # x = WriteOnceReadWhenever()
        self.uzd = ''
        uud = self.uzd
        del self.uzd

        print("\n")
        print(" input_ThirdPartyConversationID Representation : " + uud)
        print("\n")

        input_CustomerMSISDN = nu
        input_Amount = amt
        input_PurchasedItemsDesc = itm

        # input_CustomerMSISDN = '255756882578'
        print('Customer mobile number  ' + input_CustomerMSISDN)
        print('The amount to be deducted ' + input_Amount)
        print('payment for ' + input_PurchasedItemsDesc)

        api_context = APIContext()
        api_context.api_key = result.body['output_SessionID']
        api_context.public_key = public_key
        api_context.ssl = True
        api_context.method_type = APIMethodType.POST
        api_context.address = 'openapi.m-pesa.com'
        api_context.port = 443
        api_context.path = '/sandbox/ipg/v2/vodacomTZN/c2bPayment/singleStage/'

        api_context.add_header('Origin', '3.8.36.235')

        api_context.add_parameter('input_Amount', input_Amount)
        api_context.add_parameter('input_Country', 'TZN')
        api_context.add_parameter('input_Currency', 'TZS')
        api_context.add_parameter('input_CustomerMSISDN', input_CustomerMSISDN)
        api_context.add_parameter('input_ServiceProviderCode', '000000')
        api_context.add_parameter('input_ThirdPartyConversationID', uud)
        api_context.add_parameter('input_TransactionReference', 'T1234T')
        api_context.add_parameter(
            'input_PurchasedItemsDesc', input_PurchasedItemsDesc)

        api_request = APIRequest(api_context)

        # SessionID can take up to 30 seconds to become 'live' in the system and will be invalid until it is
        # 5sec for testing
        sleep(3)

        result = None
        try:
            result = api_request.execute()
        except Exception as e:
            print('Call Failed: ' + str(e))

        if result is None:
            raise Exception('API call failed to get result. Please check.')

        try:

            if typez == "YEARLY" and result.body["output_ResponseCode"] == "INS-0":
                userDataz = FirestoreDataYearly(
                    output_ConversationID=result.body["output_ConversationID"],
                    output_ResponseCode=result.body["output_ResponseCode"],
                    output_ResponseDesc=result.body["output_ResponseDesc"],
                    output_ThirdPartyConversationID=result.body["output_ThirdPartyConversationID"],
                    output_TransactionID=result.body["output_TransactionID"],
                    userId=usr,
                    msisdn=nu,
                    itemDesc=itm,
                    typeOfSubscription=typez,
                    school=schl,
                    userName=un,
                    amount=amt,
                    created=datetime.datetime.utcnow()
                )

                subscriptions_ref.document(uud).set(userDataz.dict())
            elif typez == "MONTHLY" and result.body["output_ResponseCode"] == "INS-0":
                userDataz2 = FirestoreDataMonthly(
                    output_ConversationID=result.body["output_ConversationID"],
                    output_ResponseCode=result.body["output_ResponseCode"],
                    output_ResponseDesc=result.body["output_ResponseDesc"],
                    output_ThirdPartyConversationID=result.body["output_ThirdPartyConversationID"],
                    output_TransactionID=result.body["output_TransactionID"],
                    userId=usr,
                    msisdn=nu,
                    itemDesc=itm,
                    typeOfSubscription=typez,
                    school=schl,
                    userName=un,
                    amount=amt,
                    created=datetime.datetime.utcnow()
                )
                print(userDataz2.dict())
                subscriptions_ref.document(uud).set(userDataz2.dict())

        except Exception as e:
            print(str(e))
            return f"An Error Occured: " + str(e)

        appResponse = ResponseModel(
            responseCode=result.body["output_ResponseCode"],
            output_ResponseDesc=result.body["output_ResponseDesc"],
            output_ThirdPartyConversationID=result.body["output_ThirdPartyConversationID"],
        )

        print(result.status_code)
        # print(result.headers)
        print(result.body)
        # This carries tru or false creation ..refer to pesa api for the code from response
        rslt = result.body["output_ResponseCode"]
        print('The response code from Mpesa........' + rslt + '.........')
        return jsonify(appResponse.dict())


api.add_resource(Silver, '/silver', )