from django.shortcuts import render, get_object_or_404
from .models import PortfolioItem, PortfolioCategory
from django.db.models import Count

def portfolio_list(request):
    items = PortfolioItem.objects.all().order_by('-date')
    categories = PortfolioCategory.objects.annotate(item_count=Count('portfolioitem')).filter(item_count__gt=0)
    return render(request, 'portfolio/portfolio_list.html', {
        'items': items,
        'categories': categories
    })


def portfolio_detail(request, slug):
    item = get_object_or_404(PortfolioItem, slug=slug)
    return render(request, 'portfolio/portfolio_detail.html', {
        'item': item
    })
