from django.shortcuts import render, get_object_or_404, redirect
from django.http import HttpResponse
from django.utils import timezone
from .models import Brigade, Order, Client

def index(request):
    return redirect('cleaning:order_list')

def order_list(request):
    orders = Order.objects.order_by('date')
    return render(request, 'cleaning/order_list.html', {'orders': orders})

def order_detail(request, order_id):
    order = get_object_or_404(Order.objects.prefetch_related('completedwork_set__service'), pk=order_id)
    return render(request, 'cleaning/order_detail.html', {'order': order})

# def brigade_detail(request, pk):
#     brigade = get_object_or_404(Brigade, pk=pk)
#     return render(request, 'cleaning/brigade_detail.html', {'brigade': brigade})
def brigade_detail(request, pk):
    brigade = get_object_or_404(Brigade, pk=pk)
    today = timezone.now().date()
    active = brigade.orders.filter(date__gte=today).order_by('date') # type: ignore
    completed = brigade.orders.filter(date__lt=today).order_by('-date') # type: ignore
    return render(request, 'cleaning/brigade_detail.html', {
        'brigade': brigade,
        'active': active,
        'completed': completed,
    })

def client_detail(request, pk):
    client = get_object_or_404(Client, pk=pk)
    orders = client.orders.order_by('date') # type: ignore
    return render(request, 'cleaning/client_detail.html', {'client': client})