Feat/async

mpesakit/account_balance/__init__.py

@@ -1,3 +1,4 @@
+from .account_balance import AccountBalance, AsyncAccountBalance
 from .schemas import (
     AccountBalanceIdentifierType,
     AccountBalanceRequest,
@@ -7,9 +8,9 @@
     AccountBalanceTimeoutCallback,
     AccountBalanceTimeoutCallbackResponse,
 )
-from .account_balance import AccountBalance
 
 __all__ = [
+    "AsyncAccountBalance",
     "AccountBalance",
     "AccountBalanceRequest",
     "AccountBalanceResponse",

mpesakit/account_balance/account_balance.py

@@ -5,8 +5,9 @@
 """
 
 from pydantic import BaseModel, ConfigDict
-from mpesakit.auth import TokenManager
-from mpesakit.http_client import HttpClient
+
+from mpesakit.auth import AsyncTokenManager, TokenManager
+from mpesakit.http_client import AsyncHttpClient, HttpClient
 
 from .schemas import (
     AccountBalanceRequest,
@@ -43,5 +44,40 @@ def query(self, request: AccountBalanceRequest) -> AccountBalanceResponse:
             "Authorization": f"Bearer {self.token_manager.get_token()}",
             "Content-Type": "application/json",
         }
-        response_data = self.http_client.post(url, json=request.model_dump(by_alias=True), headers=headers)
+        response_data = self.http_client.post(
+            url, json=request.model_dump(by_alias=True), headers=headers
+        )
+        return AccountBalanceResponse(**response_data)
+
+
+class AsyncAccountBalance(BaseModel):
+    """Represents the async Account Balance API client for M-Pesa operations.
+
+    Attributes:
+        http_client (AsyncHttpClient): Async HTTP client for making requests to the M-Pesa API.
+        token_manager (AsyncTokenManager): Async token manager for authentication.
+    """
+
+    http_client: AsyncHttpClient
+    token_manager: AsyncTokenManager
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    async def query(self, request: AccountBalanceRequest) -> AccountBalanceResponse:
+        """Queries the account balance asynchronously.
+
+        Args:
+            request (AccountBalanceRequest): The account balance query request.
+
+        Returns:
+            AccountBalanceResponse: Response from the M-Pesa API.
+        """
+        url = "/mpesa/accountbalance/v1/query"
+        headers = {
+            "Authorization": f"Bearer {await self.token_manager.get_token()}",
+            "Content-Type": "application/json",
+        }
+        response_data = await self.http_client.post(
+            url, json=request.model_dump(by_alias=True), headers=headers
+        )
         return AccountBalanceResponse(**response_data)

mpesakit/b2b_express_checkout/__init__.py

@@ -1,13 +1,13 @@
+from .b2b_express_checkout import AsyncB2BExpressCheckout, B2BExpressCheckout
 from .schemas import (
+    B2BExpressCallbackResponse,
+    B2BExpressCheckoutCallback,
     B2BExpressCheckoutRequest,
     B2BExpressCheckoutResponse,
-    B2BExpressCheckoutCallback,
-    B2BExpressCallbackResponse,
 )
 
-from .b2b_express_checkout import B2BExpressCheckout
-
 __all__ = [
+    "AsyncB2BExpressCheckout",
     "B2BExpressCheckout",
     "B2BExpressCheckoutRequest",
     "B2BExpressCheckoutResponse",

mpesakit/b2b_express_checkout/b2b_express_checkout.py

@@ -5,8 +5,9 @@
 """
 
 from pydantic import BaseModel, ConfigDict
-from mpesakit.auth import TokenManager
-from mpesakit.http_client import HttpClient
+
+from mpesakit.auth import AsyncTokenManager, TokenManager
+from mpesakit.http_client import AsyncHttpClient, HttpClient
 
 from .schemas import (
     B2BExpressCheckoutRequest,
@@ -49,3 +50,38 @@ def ussd_push(
             url, json=request.model_dump(mode="json"), headers=headers
         )
         return B2BExpressCheckoutResponse(**response_data)
+
+
+class AsyncB2BExpressCheckout(BaseModel):
+    """Represents the async B2B Express Checkout API client for M-Pesa operations.
+
+    Attributes:
+        http_client (AsyncHttpClient): Async HTTP client for making requests to the M-Pesa API.
+        token_manager (AsyncTokenManager): Async token manager for authentication.
+    """
+
+    http_client: AsyncHttpClient
+    token_manager: AsyncTokenManager
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    async def ussd_push(
+        self, request: B2BExpressCheckoutRequest
+    ) -> B2BExpressCheckoutResponse:
+        """Initiates a B2B Express Checkout USSD Push transaction asynchronously.
+
+        Args:
+            request (B2BExpressCheckoutRequest): The B2B Express Checkout request data.
+
+        Returns:
+            B2BExpressCheckoutResponse: Response from the M-Pesa API.
+        """
+        url = "/v1/ussdpush/get-msisdn"
+        headers = {
+            "Authorization": f"Bearer {await self.token_manager.get_token()}",
+            "Content-Type": "application/json",
+        }
+        response_data = await self.http_client.post(
+            url, json=request.model_dump(mode="json"), headers=headers
+        )
+        return B2BExpressCheckoutResponse(**response_data)

mpesakit/b2c/__init__.py

@@ -1,16 +1,17 @@
+from .b2c import B2C, AsyncB2C
 from .schemas import (
     B2CCommandIDType,
     B2CRequest,
     B2CResponse,
-    B2CResultParameter,
-    B2CResultMetadata,
     B2CResultCallback,
+    B2CResultMetadata,
+    B2CResultParameter,
     B2CTimeoutCallback,
     B2CTimeoutCallbackResponse,
 )
-from .b2c import B2C
 
 __all__ = [
+    "AsyncB2C",
     "B2C",
     "B2CCommandIDType",
     "B2CRequest",

mpesakit/b2c/b2c.py

@@ -5,8 +5,9 @@
 """
 
 from pydantic import BaseModel, ConfigDict
-from mpesakit.auth import TokenManager
-from mpesakit.http_client import HttpClient
+
+from mpesakit.auth import AsyncTokenManager, TokenManager
+from mpesakit.http_client import AsyncHttpClient, HttpClient
 
 from .schemas import (
     B2CRequest,
@@ -43,5 +44,40 @@ def send_payment(self, request: B2CRequest) -> B2CResponse:
             "Authorization": f"Bearer {self.token_manager.get_token()}",
             "Content-Type": "application/json",
         }
-        response_data = self.http_client.post(url, json=request.model_dump(by_alias=True), headers=headers)
+        response_data = self.http_client.post(
+            url, json=request.model_dump(by_alias=True), headers=headers
+        )
+        return B2CResponse(**response_data)
+
+
+class AsyncB2C(BaseModel):
+    """Represents the async B2C API client for M-Pesa Business to Customer operations.
+
+    Attributes:
+        http_client (AsyncHttpClient): Async HTTP client for making requests to the M-Pesa API.
+        token_manager (AsyncTokenManager): Async token manager for authentication.
+    """
+
+    http_client: AsyncHttpClient
+    token_manager: AsyncTokenManager
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    async def send_payment(self, request: B2CRequest) -> B2CResponse:
+        """Initiates a B2C payment request asynchronously.
+
+        Args:
+            request (B2CRequest): The payment request details.
+
+        Returns:
+            B2CResponse: Response from the M-Pesa API after payment initiation.
+        """
+        url = "/mpesa/b2c/v3/paymentrequest"
+        headers = {
+            "Authorization": f"Bearer {await self.token_manager.get_token()}",
+            "Content-Type": "application/json",
+        }
+        response_data = await self.http_client.post(
+            url, json=request.model_dump(by_alias=True), headers=headers
+        )
         return B2CResponse(**response_data)

mpesakit/b2c_account_top_up/__init__.py

@@ -1,14 +1,15 @@
+from .b2c_account_top_up import AsyncB2CAccountTopUp, B2CAccountTopUp
 from .schemas import (
-    B2CAccountTopUpRequest,
-    B2CAccountTopUpResponse,
     B2CAccountTopUpCallback,
     B2CAccountTopUpCallbackResponse,
+    B2CAccountTopUpRequest,
+    B2CAccountTopUpResponse,
     B2CAccountTopUpTimeoutCallback,
     B2CAccountTopUpTimeoutCallbackResponse,
 )
-from .b2c_account_top_up import B2CAccountTopUp
 
 __all__ = [
+    "AsyncB2CAccountTopUp",
     "B2CAccountTopUp",
     "B2CAccountTopUpRequest",
     "B2CAccountTopUpResponse",

mpesakit/b2c_account_top_up/b2c_account_top_up.py

@@ -5,8 +5,9 @@
 """
 
 from pydantic import BaseModel, ConfigDict
-from mpesakit.auth import TokenManager
-from mpesakit.http_client import HttpClient
+
+from mpesakit.auth import AsyncTokenManager, TokenManager
+from mpesakit.http_client import AsyncHttpClient, HttpClient
 
 from .schemas import (
     B2CAccountTopUpRequest,
@@ -47,3 +48,37 @@ def topup(self, request: B2CAccountTopUpRequest) -> B2CAccountTopUpResponse:
             url, json=request.model_dump(mode="json"), headers=headers
         )
         return B2CAccountTopUpResponse(**response_data)
+
+
+class AsyncB2CAccountTopUp(BaseModel):
+    """Represents the async B2C Account TopUp API client for M-Pesa operations.
+
+    Attributes:
+        http_client (AsyncHttpClient): Async HTTP client for making requests to the M-Pesa API.
+        token_manager (AsyncTokenManager): Async token manager for authentication.
+    """
+
+    http_client: AsyncHttpClient
+    token_manager: AsyncTokenManager
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    async def topup(self, request: B2CAccountTopUpRequest) -> B2CAccountTopUpResponse:
+        """Initiates a B2C Account TopUp transaction asynchronously.
+
+        Args:
+            request (B2CAccountTopUpRequest): The B2C Account TopUp request data.
+
+        Returns:
+            B2CAccountTopUpResponse: Response from the M-Pesa API.
+        """
+        url = "/mpesa/b2b/v1/paymentrequest"
+        token = await self.token_manager.get_token()
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "Content-Type": "application/json",
+        }
+        response_data = await self.http_client.post(
+            url, json=request.model_dump(mode="json"), headers=headers
+        )
+        return B2CAccountTopUpResponse(**response_data)

mpesakit/bill_manager/__init__.py

@@ -1,25 +1,26 @@
+from .bill_manager import AsyncBillManager, BillManager
 from .schemas import (
-    BillManagerOptInRequest,
-    BillManagerOptInResponse,
-    BillManagerUpdateOptInRequest,
-    BillManagerUpdateOptInResponse,
-    BillManagerSingleInvoiceRequest,
-    BillManagerSingleInvoiceResponse,
     BillManagerBulkInvoiceRequest,
     BillManagerBulkInvoiceResponse,
-    BillManagerCancelSingleInvoiceRequest,
     BillManagerCancelBulkInvoiceRequest,
     BillManagerCancelInvoiceResponse,
-    BillManagerPaymentNotificationRequest,
-    BillManagerPaymentNotificationResponse,
+    BillManagerCancelSingleInvoiceRequest,
+    BillManagerOptInRequest,
+    BillManagerOptInResponse,
     BillManagerPaymentAcknowledgmentRequest,
     BillManagerPaymentAcknowledgmentResponse,
+    BillManagerPaymentNotificationRequest,
+    BillManagerPaymentNotificationResponse,
+    BillManagerSingleInvoiceRequest,
+    BillManagerSingleInvoiceResponse,
+    BillManagerUpdateOptInRequest,
+    BillManagerUpdateOptInResponse,
     InvoiceItem,
 )
 
-from .bill_manager import BillManager
-
 __all__ = [
+    "AsyncBillManager",
+    "BillManager",
     "BillManagerOptInRequest",
     "BillManagerOptInResponse",
     "BillManagerUpdateOptInRequest",
@@ -35,6 +36,5 @@
     "BillManagerPaymentNotificationResponse",
     "BillManagerPaymentAcknowledgmentRequest",
     "BillManagerPaymentAcknowledgmentResponse",
-    "BillManager",
     "InvoiceItem",
 ]