Merge pull request #355 from EasyPost/new_partner_white_label_billing

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.create_credit_card_client_secret`
+  - `BetaReferralCustomerService.create_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

@@ -1,5 +1,6 @@
 from typing import (
     Any,
+    Optional,
 )
 
 from easypost.easypost_object import convert_to_easypost_object
@@ -24,7 +25,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 +36,56 @@ 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 create_credit_card_client_secret(self) -> dict[str, Any]:
+        """Creates 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 create_bank_account_client_secret(self, return_url: Optional[str] = None) -> dict[str, Any]:
+        """Creates 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

@@ -103,7 +103,7 @@ def add_credit_card(
         cvc: str,
         priority: str = "primary",
     ) -> dict[str, Any]:
-        """Add credit card to a referral customer.
+        """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 +128,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(