add create_stripe_account_link

move modify_stripe_account_capability into Flask_SaaS class

Author: chrisjsimpson

src/flask_saas/blueprints/stripe_connect.py

@@ -19,13 +19,6 @@
 )  # noqa: E501
 
 
-def modify_stripe_account_capability(account_id):
-    """Request (again) card_payments capability after kyc onboarding
-    is complete"""
-    log.debug("Called modify_stripe_account_capability")
-    stripe.Account.modify_capability(account_id, "card_payments", requested=True)
-
-
 def _generate_account_link(account_id):
     """
     From the Stripe Docs:
@@ -50,7 +43,6 @@ def _generate_account_link(account_id):
 @stripe_connect.route("/stripe-set-livemode", methods=["POST"])
 def set_stripe_livemode():
     log.debug("Called set_stripe_livemode")
-    breakpoint()
     flask_saas: Flask_SaaS = current_app.config["flask_saas"]
     livemode = request.data.decode("utf-8")
     if livemode == "0" or livemode == "1":
@@ -90,7 +82,7 @@ def index() -> str:
         # Attempt to Updates an existing Account Capability to accept card payments
         try:
             account = flask_saas.get_stripe_connect_account()
-            modify_stripe_account_capability(account.id)
+            flask_saas.modify_stripe_account_capability(account.id)
         except Exception as e:
             log.error(f"Could not update card_payments capability for account. {e}")
 
@@ -113,7 +105,6 @@ def index() -> str:
 @stripe_connect.route("/stripe-onboard", methods=["POST"])
 def stripe_onboarding() -> jsonify:
     log.debug("called stripe_onboarding")
-
     flask_saas: Flask_SaaS = current_app.config["flask_saas"]
     stripe.api_key = flask_saas.get_stripe_secret_key()
 

src/flask_saas/flask_saas.py

@@ -88,3 +88,30 @@ def create_stripe_connect_account(self) -> stripe.Account:
         )
 
         return account
+
+    def create_stripe_account_link(
+        self, account_id: stripe.Account, refresh_url: str, return_url: str
+    ) -> str:
+        """
+        From the Stripe Docs: https://stripe.com/docs/api/account_links/create
+        A user that is redirected to your return_url might not have completed the
+        onboarding process. Use the /v1/accounts endpoint to retrieve the user’s
+        account and check for charges_enabled. If the account is not fully onboarded,
+        provide UI prompts to allow the user to continue onboarding later. The user
+        can complete their account activation through a new account link (generated
+        by your integration). You can check the state of the details_submitted
+        parameter on their account to see if they’ve completed the onboarding process.
+        """
+        account_link = stripe.AccountLink.create(
+            type="account_onboarding",
+            account=account_id,
+            refresh_url=refresh_url,
+            return_url=return_url,
+        )
+        return account_link.url
+
+    def modify_stripe_account_capability(self, account_id: str) -> None:
+        """Request (again) card_payments capability after kyc onboarding
+        is complete"""
+        log.debug("Called modify_stripe_account_capability")
+        stripe.Account.modify_capability(account_id, "card_payments", requested=True)