from django.shortcuts import render, get_object_or_404, redirect
from .models import Category, Project
import json

import logging
logger = logging.getLogger("django")


def home(request):
    projects_data = []

    projects = Project.objects.select_related('category').prefetch_related('images')

    for project in projects:
        projects_data.append({
            'id': project.id,
            'title': project.title,
            'category': project.category.name,
            'short_description': project.short_description,
            'description1': project.description1,
            'description2': project.description2,
            'description3': project.description3,
            'main_image': project.main_image.url if project.main_image else '',
            'extra_images': [img.image.url for img in project.images.all()]
        })

    categories = Category.objects.values_list('name', flat=True).distinct()

    context = {
        'projects_json': json.dumps(projects_data),
        'categories_json': json.dumps(list(categories))
    }

    return render(request, 'website/pages/index.html', context)