Create subscription.py

routes/subscription.py

@@ -0,0 +1,91 @@
+import stripe
+from fastapi import APIRouter, HTTPException, Request
+from pydantic import BaseModel
+
+router = APIRouter()
+
+stripe.api_key = "sk_test_51N6K5JB9VMe0qzbOjlJvMEsfdQyrFgV49vRaeErtmhrzHV3Cu3f5jMDJmrhKdI5uqvpHubjkmwDQgMOtCEmz19t800AouH7W6g"
+stripe.api_version = "2023-10-16"
+
+### MODELOS DE DADOS ###
+class SubscriptionRequest(BaseModel):
+    user_id: str
+    estilista_id: str
+    amount: int  # Valor em centavos (ex: R$50,00 -> 5000)
+
+class CancelSubscriptionRequest(BaseModel):
+    subscription_id: str
+
+### 1. CRIAR PRODUTO E PREÇO DINÂMICO ###
+def create_price(amount_in_cents: int):
+    try:
+        product = stripe.Product.create(name="Stylist Subscription")
+        price = stripe.Price.create(
+            unit_amount=amount_in_cents,
+            currency="brl",
+            recurring={"interval": "month"},
+            product=product.id,
+        )
+        return price.id
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+### 2. INICIAR O CHECKOUT ###
+@router.post("/create_checkout_session")
+def create_checkout_session(data: SubscriptionRequest):
+    price_id = create_price(data.amount)  # Criar um preço dinâmico
+
+    try:
+        session = stripe.checkout.Session.create(
+            success_url="http://localhost:4242/success",
+            cancel_url="http://localhost:4242/cancel",
+            payment_method_types=["card"],
+            mode="subscription",
+            line_items=[{"price": price_id, "quantity": 1}],
+            metadata={
+                "user_id": data.user_id,
+                "estilista_id": data.estilista_id,
+            },
+        )
+        return {"url": session.url}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+### 3. WEBHOOK PARA PROCESSAR PAGAMENTO BEM-SUCEDIDO ###
+@router.post("/webhook")
+async def stripe_webhook(request: Request):
+    payload = await request.body()
+    sig_header = request.headers.get("Stripe-Signature")
+
+    try:
+        event = stripe.Webhook.construct_event(payload, sig_header, "seu_webhook_secret")
+    except ValueError:
+        raise HTTPException(status_code=400, detail="Invalid payload")
+    except stripe.error.SignatureVerificationError:
+        raise HTTPException(status_code=400, detail="Invalid signature")
+
+    if event["type"] == "invoice.payment_succeeded":
+        invoice = event["data"]["object"]
+        estilista_id = invoice["metadata"]["estilista_id"]
+        user_id = invoice["metadata"]["user_id"]
+        amount_paid = invoice["amount_paid"]
+
+        # 80% do valor fica pendente para o estilista
+        amount_for_stylist = int(amount_paid * 0.8)
+        
+        # Aqui você salvaria o saldo pendente do estilista no banco de dados
+        save_pending_balance(estilista_id, amount_for_stylist)
+
+    return {"success": True}
+
+### 4. CANCELAR UMA ASSINATURA ###
+@router.post("/cancel_subscription")
+def cancel_subscription(data: CancelSubscriptionRequest):
+    try:
+        subscription = stripe.Subscription.modify(
+            data.subscription_id,
+            cancel_at_period_end=True,  # Cancela no final do ciclo atual
+        )
+        return {"status": "Subscription will be canceled at period end"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))