feat: adds new cc and bank functions for ReferralCustomers

Author: Justintime50

CHANGELOG.md

@@ -3,6 +3,11 @@
 ## Next Release
 
 - Drops support for Python 3.7 and 3.8
+- Adds the following functions to assist ReferralCustomers add credit cards and bank accounts:
+  - `BetaReferralCustomerService.retrieve_credit_card_client_secret`
+  - `BetaReferralCustomerService.retrieve_bank_account_client_secret`
+  - `ReferralCustomerService.add_credit_card_from_stripe`
+  - `ReferralCustomerService.add_bank_account_from_stripe`
 - Fixes the payload wrapping for updating a webhook
 - Removes deprecated `user.all_api_keys` and `user.api_keys`, use `api_key.all` and `api_key.retrieve_api_keys_for_user` respectively
 - Bumps all dev dependencies

easypost/services/beta_referral_customer_service.py

@@ -24,7 +24,7 @@ def add_payment_method(
         EasyPost, we will associate your Stripe payment method with either your primary
         or secondary EasyPost payment method.
         """
-        wrapped_params = {
+        params = {
             "payment_method": {
                 "stripe_customer_id": stripe_customer_id,
                 "payment_method_reference": payment_method_reference,
@@ -35,33 +35,58 @@ def add_payment_method(
         response = Requestor(self._client).request(
             method=RequestMethod.POST,
             url="/referral_customers/payment_method",
-            params=wrapped_params,
+            params=params,
             beta=True,
         )
 
         return convert_to_easypost_object(response=response)
 
     def refund_by_amount(self, refund_amount: int) -> dict[str, Any]:
         """Refund a ReferralCustomer wallet by specifying an amount."""
-        wrapped_params = {"refund_amount": refund_amount}
+        params = {"refund_amount": refund_amount}
 
         response = Requestor(self._client).request(
             method=RequestMethod.POST,
             url="/referral_customers/refunds",
-            params=wrapped_params,
+            params=params,
             beta=True,
         )
 
         return convert_to_easypost_object(response=response)
 
     def refund_by_payment_log(self, payment_log_id: str) -> dict[str, Any]:
         """Refund a ReferralCustomer wallet by specifying a payment log ID to completely refund."""
-        wrapped_params = {"payment_log_id": payment_log_id}
+        params = {"payment_log_id": payment_log_id}
 
         response = Requestor(self._client).request(
             method=RequestMethod.POST,
             url="/referral_customers/refunds",
-            params=wrapped_params,
+            params=params,
+            beta=True,
+        )
+
+        return convert_to_easypost_object(response=response)
+
+    def retrieve_credit_card_client_secret(self) -> Dict[str, Any]:
+        """Retrieves a client secret to use with Stripe when adding a credit card."""
+        response = Requestor(self._client).request(
+            method=RequestMethod.POST,
+            url="/setup_intents",
+            beta=True,
+        )
+
+        return convert_to_easypost_object(response=response)
+
+    def retrieve_bank_account_client_secret(
+        self, return_url: Optional[str] = None
+    ) -> Dict[str, Any]:
+        """Retrieves a client secret to use with Stripe when adding a bank account."""
+        params = {"return_url": return_url}
+
+        response = Requestor(self._client).request(
+            method=RequestMethod.POST,
+            url="/financial_connections_sessions",
+            params=params if return_url else None,
             beta=True,
         )
 

easypost/services/referral_customer_service.py

@@ -5,6 +5,7 @@
 )
 
 import requests
+
 from easypost.constant import (
     _FILTERS_KEY,
     SEND_STRIPE_DETAILS_ERROR,
@@ -69,9 +70,13 @@ def all(self, **params) -> dict[str, Any]:
 
         url = "/referral_customers"
 
-        response = Requestor(self._client).request(method=RequestMethod.GET, url=url, params=params)
+        response = Requestor(self._client).request(
+            method=RequestMethod.GET, url=url, params=params
+        )
 
-        response[_FILTERS_KEY] = filters  # Save the filters used to reference in potential get_next_page call
+        response[_FILTERS_KEY] = (
+            filters  # Save the filters used to reference in potential get_next_page call
+        )
 
         return convert_to_easypost_object(response=response)
 
@@ -102,8 +107,8 @@ def add_credit_card(
         expiration_year: int,
         cvc: str,
         priority: str = "primary",
-    ) -> dict[str, Any]:
-        """Add credit card to a referral customer.
+    ) -> Dict[str, Any]:
+        """Add a credit card to EasyPost for a ReferralCustomer without needing a Stripe account.
 
         This function requires the ReferralCustomer User's API key.
         """
@@ -128,6 +133,64 @@ def add_credit_card(
 
         return convert_to_easypost_object(response)
 
+    def add_credit_card_from_stripe(
+        self,
+        referral_api_key: str,
+        payment_method_id: str,
+        priority: str = "primary",
+    ) -> Dict[str, Any]:
+        """Add a credit card to EasyPost for a ReferralCustomer with a payment method ID from Stripe.
+
+        This function requires the ReferralCustomer User's API key.
+        """
+        params = {
+            "credit_card": {
+                "payment_method_id": payment_method_id,
+                "priority": priority,
+            }
+        }
+
+        # Override the API key to use the referral's for this single request
+        referral_client = deepcopy(self._client)
+        referral_client.api_key = referral_api_key
+
+        response = Requestor(referral_client).request(
+            method=RequestMethod.POST,
+            params=params,
+            url="/credit_cards",
+        )
+
+        return convert_to_easypost_object(response)
+
+    def add_bank_account_from_stripe(
+        self,
+        referral_api_key: str,
+        financial_connections_id: str,
+        mandate_data: Dict[str, Any],
+        priority: str = "primary",
+    ) -> Dict[str, Any]:
+        """Add a bank account to EasyPost for a ReferralCustomer.
+
+        This function requires the ReferralCustomer User's API key.
+        """
+        params = {
+            "financial_connections_id": financial_connections_id,
+            "mandate_data": mandate_data,
+            "priority": priority,
+        }
+
+        # Override the API key to use the referral's for this single request
+        referral_client = deepcopy(self._client)
+        referral_client.api_key = referral_api_key
+
+        response = Requestor(referral_client).request(
+            method=RequestMethod.POST,
+            params=params,
+            url="/bank_accounts",
+        )
+
+        return convert_to_easypost_object(response)
+
     def _retrieve_easypost_stripe_api_key(self) -> str:
         """Retrieve EasyPost's Stripe public API key."""
         public_key = Requestor(self._client).request(