Nuevos Parámetros incluídos en createSession

app.py

@@ -1,11 +1,10 @@
 import funciones
-from fastapi import FastAPI, Form, HTTPException
+from typing import Optional, Literal
 from fastapi.responses import JSONResponse
-from typing import Optional
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI, Form, HTTPException, status
 
 #FUTURE: Quizá en el futuro cambiarla de Form a Json con Pydantic.
-
 app = FastAPI()
 
 # Configuración de CORS
@@ -101,13 +100,29 @@ async def creaCliente(
 @app.post("/creaLinkSesion/")
 async def creaLinkSesion(
     price_id: str = Form(...),
+    unidades: int = Form(...), # <--- ¡Nuevo parámetro 'unidades' como int!
+    mode: Literal["payment", "subscription"] = Form(...), # <--- ¡Nuevo parámetro 'mode' con validación!
     customer_email: Optional[str] = Form(None),
     customer_id: Optional[str] = Form(None),
-    firebase_user: Optional[str] = Form(None) # <-- ¡Nuevo parámetro para Firebase User ID!
+    firebase_user: Optional[str] = Form(None)
 ):
     """
-    Crea una Checkout Session, ahora con la opción de buscar/crear cliente
-    usando firebase_user.
+    Crea una Checkout Session en Stripe.
+    Acepta 'unidades' (cantidad) y 'mode' ('payment' o 'subscription').
     """
-    # Llama a la función de Stripe con el nuevo parámetro
-    return funciones.create_checkout_session(price_id, customer_email, customer_id, firebase_user)
+    try:
+        # Llama a la función de Stripe con todos los parámetros, incluyendo los nuevos
+        return funciones.create_checkout_session(
+            price_id,
+            customer_email,
+            customer_id,
+            firebase_user,
+            unidades, # Pasar el nuevo parámetro
+            mode      # Pasar el nuevo parámetro
+        )
+    except Exception as e:
+        print(f"Error al crear la sesión de checkout: {e}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"Error interno al intentar crear la sesión de checkout: {e}"
+        )

funciones.py

@@ -3,7 +3,7 @@ import globales
 
 stripe.api_key = globales.llave
    
-def create_checkout_session(price_id, customer_email=None, customer_id=None, firebase_user=None):
+def create_checkout_session(price_id, customer_email=None, customer_id=None, firebase_user=None, unidades=None, mode=None):
     """
     Crea una nueva Checkout Session para un pago único.
     Busca o crea un cliente en Stripe si no se provee customer_id ni customer_email,
@@ -88,13 +88,13 @@ def create_checkout_session(price_id, customer_email=None, customer_id=None, fir
                 'price': price_id,
                 'quantity': 1, #maneja cantidades cuando es subscription
             }],
-            'mode': 'payment', #payment o subscription
+            'mode': mode, #payment o subscription
             'success_url': 'https://app.splashmix.ink/',
             'cancel_url': 'https://app.splashmix.ink/',
             'locale': 'auto',
             'client_reference_id': 'HERC',
             'metadata': {
-                'imagenes': 1000
+                'imagenes': unidades
             } # Inicializamos metadata para añadir el customer_id y firebase_user si existen
         }