Primer Commit

.gitignore

@@ -0,0 +1,3 @@
+/venv/
+/__pycache__/
+stripe.exe

app.py

@@ -0,0 +1,105 @@
+import os
+import stripe
+import uvicorn
+from fastapi import FastAPI, Request, Header
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+import time
+
+
+app = FastAPI()
+
+templates = Jinja2Templates(directory="templates")
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+stripe.api_key = os.environ["STRIPE_KEY"]
+# This is a terrible idea, only used for demo purposes!
+app.state.stripe_customer_id = None
+
+
+@app.get("/")
+def index(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request, "hasCustomer": app.state.stripe_customer_id is not None})
+
+
+@app.get("/success")
+async def success(request: Request):
+    return templates.TemplateResponse("success.html", {"request": request})
+
+
+@app.get("/cancel")
+async def cancel(request: Request):
+    return templates.TemplateResponse("cancel.html", {"request": request})
+
+
+@app.post("/create-checkout-session")
+async def create_checkout_session(request: Request):
+    data = await request.json()
+
+    if not app.state.stripe_customer_id:
+        customer = stripe.Customer.create(
+            description="Demo customer",
+        )
+        app.state.stripe_customer_id = customer["id"]
+
+    checkout_session = stripe.checkout.Session.create(
+        customer=app.state.stripe_customer_id,
+        success_url="http://localhost:8000/success?session_id={CHECKOUT_SESSION_ID}",
+        cancel_url="http://localhost:8000/cancel",
+        payment_method_types=["card"],
+        mode="subscription",
+        line_items=[{
+            "price": data["priceId"],
+            "quantity": 1
+        }],
+    )
+    return {"sessionId": checkout_session["id"]}
+
+
+@app.post("/create-portal-session")
+async def create_portal_session():
+    session = stripe.billing_portal.Session.create(
+        customer=app.state.stripe_customer_id,
+        return_url="http://localhost:8000"
+    )
+    return {"url": session.url}
+
+
+@app.post("/webhook")
+async def webhook_received(request: Request, stripe_signature: str = Header(None)):
+    webhook_secret = os.environ["STRIPE_WEBHOOK_SECRET"]
+
+    print("Entré al webhook 182...")
+    time.sleep(1)
+
+    data = await request.body()
+    try:
+        event = stripe.Webhook.construct_event(
+            payload=data,
+            sig_header=stripe_signature,
+            secret=webhook_secret
+        )
+        event_data = event['data']
+    except Exception as e:
+        return {"error": str(e)}
+
+    event_type = event['type']
+
+    print("Voy a imprimir el event type:")
+    print(event_type)
+    time.sleep(1)
+
+    if event_type == 'checkout.session.completed':
+        print('checkout session completed')
+    elif event_type == 'invoice.paid':
+        print('invoice paid')
+    elif event_type == 'invoice.payment_failed':
+        print('invoice payment failed')
+    else:
+        print(f'unhandled event: {event_type}')
+    
+    return {"status": "success"}
+
+
+if __name__ == '__main__':
+    uvicorn.run("app:app", reload=True)

ostest.py

@@ -0,0 +1,5 @@
+import os
+
+api_key = os.environ["STRIPE_WEBHOOK_SECRET"]
+
+print(api_key)

static/script.js

@@ -0,0 +1,56 @@
+var createCheckoutSession = function(priceId) {
+    return fetch("/create-checkout-session", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        priceId: priceId
+      })
+    }).then(function(result) {
+      return result.json();
+    });
+  };
+
+const PREMIUM_PRICE_ID = '';
+const BASIC_PRICE_ID = '';
+const stripe = Stripe("");
+
+document.addEventListener("DOMContentLoaded", function(event) {
+    document
+    .getElementById("checkout-premium")
+    .addEventListener("click", function(evt) {
+        createCheckoutSession(PREMIUM_PRICE_ID).then(function(data) {
+            stripe
+                .redirectToCheckout({
+                    sessionId: data.sessionId
+                });
+            });
+        });
+    
+    document
+    .getElementById("checkout-basic")
+    .addEventListener("click", function(evt) {
+        createCheckoutSession(BASIC_PRICE_ID).then(function(data) {
+            stripe
+                .redirectToCheckout({
+                    sessionId: data.sessionId
+                });
+            });
+        });
+
+    const billingButton = document.getElementById("manage-billing");
+    if (billingButton) {
+        billingButton.addEventListener("click", function(evt) {
+        fetch("/create-portal-session", {
+            method: "POST"
+        })
+            .then(function(response) {
+                return response.json()
+            })
+            .then(function(data) {
+                window.location.href = data.url;
+            });
+        })
+    }
+});

templates/cancel.html

@@ -0,0 +1,7 @@
+<head>
+    <title>Cancelled Checkout</title>
+</head>
+<body>
+    <h1>Payment was cancelled</h1>
+    <a href="/">Go home</a>
+</body>

templates/index.html

@@ -0,0 +1,13 @@
+<html>
+    <head>
+        <script src="https://js.stripe.com/v3/"></script>
+        <script src="{{ url_for('static', path='script.js') }}"></script>
+    </head>
+    <body>
+        <button id="checkout-premium">Subscribe Premium</button>
+        <button id="checkout-basic">Subscribe Basic</button>
+        {% if hasCustomer %}
+        <button id="manage-billing">Manage Billing</button>
+        {% endif %}
+    </body>
+</html>

templates/success.html

@@ -0,0 +1,7 @@
+<head>
+    <title>Checkout Successful</title>
+</head>
+<body>
+    <h1>Payment was successful</h1>
+    <a href="/">Go home</a>
+</body>