Update main.py

main.py

@@ -377,67 +377,58 @@ def guardar_transaccion():
         flash(f"Error al guardar la transacción: {str(e)}", "error")
         return redirect(url_for("home"))  # Redirigir al formulario principal
 
-
-# Modelos Pydantic
 class SubcategoriaCreate(BaseModel):
     nombre_categoria: str
     subcategorias: List[str]
 
-
-@app.post("/subcategorias")
-def insertar_subcategorias(datos: SubcategoriaCreate):
-    db = SessionLocal()
+@app.route("/subcategorias", methods=["POST"])
+def insertar_subcategorias():
     try:
-        # Verificar si la categoría ya existe
-        categoria = db.query(Categoria).filter(Categoria.nombre_categoria == datos.nombre_categoria).first()
+        # Obtener los datos de la solicitud
+        datos = SubcategoriaCreate.parse_obj(request.get_json())
+        
+        db = SessionLocal()
+        with db.begin():  # Manejo automático de transacciones
+            # Verificar si la categoría ya existe
+            categoria = db.query(Categoria).filter(Categoria.nombre_categoria == datos.nombre_categoria).first()
+
+            # Crear la categoría si no existe
+            if not categoria:
+                categoria = Categoria(nombre_categoria=datos.nombre_categoria)
+                db.add(categoria)
+                db.commit()
+                db.refresh(categoria)
+
+            # Insertar subcategorías
+            subcategorias_existentes = {sub.nombre_subcategoria for sub in categoria.subcategorias}
+            subcategorias_nuevas = []
+
+            for nombre_subcategoria in datos.subcategorias:
+                if nombre_subcategoria not in subcategorias_existentes:
+                    nueva_subcategoria = Subcategoria(
+                        id_categoria=categoria.id_categoria,
+                        nombre_subcategoria=nombre_subcategoria,
+                    )
+                    db.add(nueva_subcategoria)
+                    subcategorias_nuevas.append(nombre_subcategoria)
 
-        # Crear la categoría si no existe
-        if not categoria:
-            categoria = Categoria(nombre_categoria=datos.nombre_categoria)
-            db.add(categoria)
             db.commit()
-            db.refresh(categoria)
-
-        # Insertar subcategorías
-        subcategorias_nuevas = []
-        for nombre_subcategoria in datos.subcategorias:
-            # Verificar si ya existe una subcategoría con el mismo nombre
-            subcategoria_existente = (
-                db.query(Subcategoria)
-                .filter(
-                    Subcategoria.id_categoria == categoria.id_categoria,
-                    Subcategoria.nombre_subcategoria == nombre_subcategoria,
-                )
-                .first()
-            )
-            if not subcategoria_existente:
-                nueva_subcategoria = Subcategoria(
-                    id_categoria=categoria.id_categoria,
-                    nombre_subcategoria=nombre_subcategoria,
-                )
-                db.add(nueva_subcategoria)
-                subcategorias_nuevas.append(nombre_subcategoria)
-
-        db.commit()
 
+        # Respuesta final
         if subcategorias_nuevas:
-            return JSONResponse(
-                status_code=201,
-                content={
-                    "message": "Subcategorías añadidas exitosamente",
-                    "nuevas_subcategorias": subcategorias_nuevas,
-                },
-            )
+            return jsonify({
+                "message": "Subcategorías añadidas exitosamente",
+                "nuevas_subcategorias": subcategorias_nuevas,
+            }), 201
         else:
-            return JSONResponse(
-                status_code=200,
-                content={
-                    "message": "No se añadieron subcategorías nuevas (ya existían).",
-                },
-            )
+            return jsonify({
+                "message": "No se añadieron subcategorías nuevas (ya existían).",
+            }), 200
+
+    except ValidationError as e:
+        return jsonify({"error": e.errors()}), 400
     except Exception as e:
-        db.rollback()
-        return JSONResponse(status_code=500, content={"error": str(e)})
+        return jsonify({"error": str(e)}), 500
     finally:
         db.close()