Add API for AI Agent

tienda/api.py

@@ -0,0 +1,93 @@
+from typing import Optional
+from ninja import Router, Schema
+from django.db.models import Count
+from django.shortcuts import get_object_or_404
+from .models import Category, Product
+
+router = Router()
+
+class CategoryOut(Schema):
+    id: int
+    name: str
+    product_count: int
+
+class ImageInfo(Schema):
+    url: str
+    alt: str
+
+class ProductListOut(Schema):
+    id: int
+    name: str
+    sku: Optional[str] = None
+    briefdesc: str
+    price: float
+    price_with_vat: float
+    stock: int
+    category_id: int
+    category_name: str
+    primary_image: Optional[ImageInfo] = None
+    average_rating: float
+    reviews_count: int
+
+class ProductDetailOut(ProductListOut):
+    description: str
+    secondary_images: list[ImageInfo]
+
+
+def _image_info(img, request):
+    if not img:
+        return None
+    return ImageInfo(
+        url=request.build_absolute_uri(img.image.url),
+        alt=img.alt or img.name,
+    )
+
+def _product_to_list_out(p, request):
+    return ProductListOut(
+        id=p.id,
+        name=p.name,
+        sku=p.sku,
+        briefdesc=p.briefdesc,
+        price=p.price,
+        price_with_vat=p.get_price_with_vat(),
+        stock=p.stock,
+        category_id=p.category_id,
+        category_name=p.category.name,
+        primary_image=_image_info(p.primary_image, request),
+        average_rating=p.get_average_rating(),
+        reviews_count=p.get_reviews_count(),
+    )
+
+def _product_to_detail_out(p, request):
+    base = _product_to_list_out(p, request)
+    data = base.dict()
+    data["description"] = p.description
+    data["secondary_images"] = [
+        _image_info(img, request) for img in p.secondary_images.all()
+    ]
+    return ProductDetailOut(**data)
+
+
+@router.get("/categorias", response=list[CategoryOut])
+def listar_categorias(request):
+    qs = Category.objects.annotate(product_count=Count("product"))
+    return [
+        CategoryOut(id=c.id, name=c.name, product_count=c.product_count)
+        for c in qs
+    ]
+
+@router.get("/productos", response=list[ProductListOut])
+def listar_productos(request, categoria_id: Optional[int] = None):
+    qs = Product.objects.select_related("category", "primary_image")
+    if categoria_id:
+        qs = qs.filter(category_id=categoria_id)
+    return [_product_to_list_out(p, request) for p in qs]
+
+@router.get("/productos/{product_id}", response=ProductDetailOut)
+def detalle_producto(request, product_id: int):
+    p = get_object_or_404(
+        Product.objects.select_related("category", "primary_image")
+        .prefetch_related("secondary_images"),
+        id=product_id,
+    )
+    return _product_to_detail_out(p, request)