feat: send notification when payment of auto-recharge/susbcription invoice fails

daras_ai_v2/send_email.py

@@ -76,7 +76,7 @@ def send_email_via_postmark(
     html_body: str = "",
     text_body: str = "",
     message_stream: typing.Literal[
-        "outbound", "gooey-ai-workflows", "announcements"
+        "outbound", "gooey-ai-workflows", "announcements", "billing"
     ] = "outbound",
 ):
     if is_running_pytest:

daras_ai_v2/settings.py

@@ -11,7 +11,6 @@
 """
 
 import os
-import json
 from pathlib import Path
 
 import sentry_sdk
@@ -265,6 +264,7 @@
 ADMIN_EMAILS = config("ADMIN_EMAILS", cast=Csv(), default="")
 SUPPORT_EMAIL = "Gooey.AI Support <[email protected]>"
 SALES_EMAIL = "Gooey.AI Sales <[email protected]>"
+PAYMENT_EMAIL = "Gooey.AI Payments <[email protected]>"
 SEND_RUN_EMAIL_AFTER_SEC = config("SEND_RUN_EMAIL_AFTER_SEC", 5)
 
 DISALLOWED_TITLE_SLUGS = config("DISALLOWED_TITLE_SLUGS", cast=Csv(), default="") + [

payments/tasks.py

@@ -39,6 +39,34 @@ def send_monthly_spending_notification_email(user_id: int):
     user.subscription.save(update_fields=["monthly_spending_notification_sent_at"])
 
 
+@app.task
+def send_payment_failed_email_with_invoice(
+    uid: str,
+    invoice_url: str,
+    dollar_amt: float,
+    subject: str,
+):
+    from routers.account import account_route
+
+    user = AppUser.objects.get(uid=uid)
+    if not user.email:
+        logger.error(f"User doesn't have an email: {user=}")
+        return
+
+    send_email_via_postmark(
+        from_address=settings.PAYMENT_EMAIL,
+        to_address=user.email,
+        subject=subject,
+        html_body=templates.get_template("auto_payment_failed_email.html").render(
+            user=user,
+            dollar_amt=f"{dollar_amt:.2f}",
+            invoice_url=invoice_url,
+            account_url=get_app_route_url(account_route),
+        ),
+        message_stream="billing",
+    )
+
+
 def send_monthly_budget_reached_email(user: AppUser):
     from routers.account import account_route
 

payments/webhooks.py

@@ -12,7 +12,10 @@
 from daras_ai_v2 import paypal
 from .models import Subscription
 from .plans import PricingPlan
-from .tasks import send_monthly_spending_notification_email
+from .tasks import (
+    send_monthly_spending_notification_email,
+    send_payment_failed_email_with_invoice,
+)
 
 
 class PaypalWebhookHandler:
@@ -189,6 +192,28 @@ def handle_subscription_cancelled(cls, uid: str):
             external_id=None,
         )
 
+    @classmethod
+    def handle_invoice_failed(cls, uid: str, data: dict):
+        if stripe.Charge.list(payment_intent=data["payment_intent"], limit=1).has_more:
+            # we must have already sent an invoice for this to the user. so we should just ignore this event
+            logger.info("Charge already exists for this payment intent")
+            return
+
+        if data.get("metadata", {}).get("auto_recharge"):
+            send_payment_failed_email_with_invoice.delay(
+                uid=uid,
+                invoice_url=data["hosted_invoice_url"],
+                dollar_amt=data["amount_due"] / 100,
+                subject="Payment failure on your Gooey.AI auto-recharge",
+            )
+        elif data.get("subscription_details", {}):
+            send_payment_failed_email_with_invoice.delay(
+                uid=uid,
+                invoice_url=data["hosted_invoice_url"],
+                dollar_amt=data["amount_due"] / 100,
+                subject="Payment failure on your Gooey.AI subscription",
+            )
+
 
 def add_balance_for_payment(
     *,

routers/stripe.py

@@ -42,6 +42,8 @@ def webhook_received(request: Request, payload: bytes = fastapi_request_body):
     match event["type"]:
         case "invoice.paid":
             StripeWebhookHandler.handle_invoice_paid(uid, data)
+        case "invoice.payment_failed":
+            StripeWebhookHandler.handle_invoice_failed(uid, data)
         case "checkout.session.completed":
             StripeWebhookHandler.handle_checkout_session_completed(uid, data)
         case "customer.subscription.created" | "customer.subscription.updated":

templates/auto_payment_failed_email.html

@@ -0,0 +1,38 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css">
+</head>
+
+<body class="d-flex flex-column min-vh-100">
+<div class="flex-fill">
+
+  <p>Hi {{ user.first_name() }},</p>
+
+  <p>We attempted to process your payment for ${{ dollar_amt }} but your payment method was declined.</p>
+
+  <p>
+    Please <a href="{{ invoice_url }}">make a payment</a> on Gooey.AI for continued service or update
+    your payment method on <a href="{{ account_url }}">your account</a>.
+  </p>
+
+  <p>
+    <a href="{{ invoice_url }}">
+      <button class="btn btn-dark text-light">Make Payment</button>
+    </a>
+  </p>
+
+  <p>
+    Cheers,<br />
+    The Gooey.AI team
+  </p>
+
+</div>
+
+</body>
+
+</html>