from django.shortcuts import render, redirect, get_object_or_404
from django.views.generic import ListView, DetailView
from django.contrib.auth.decorators import login_required
from django.utils.decorators import method_decorator
from django.contrib import messages
from .models import SaleOrder, SaleOrderItem
from .forms import SaleOrderForm

@method_decorator(login_required, name='dispatch')
class SalesListView(ListView):
    model = SaleOrder
    template_name = 'sales/sales_list.html'
    context_object_name = 'sale_orders'
    paginate_by = 20

@login_required
def sales_create(request):
    """Create a new sales order"""
    if request.method == 'POST':
        form = SaleOrderForm(request.POST)
        if form.is_valid():
            sale_order = form.save(commit=False)
            sale_order.created_by = request.user
            # Generate order number if not provided
            if not sale_order.order_number:
                # Get the next order number
                last_order = SaleOrder.objects.order_by('-id').first()
                if last_order:
                    try:
                        last_number = int(last_order.order_number.split('-')[-1])
                        new_number = last_number + 1
                    except:
                        new_number = 1
                else:
                    new_number = 1
                sale_order.order_number = f"SO-{new_number:06d}"
            sale_order.save()
            messages.success(request, f'Sales order "{sale_order.order_number}" created successfully!')
            return redirect('sales:sales_detail', pk=sale_order.pk)
    else:
        form = SaleOrderForm()
    
    return render(request, 'sales/sales_form.html', {'form': form, 'title': 'Create Sales Order'})

@login_required
def sales_delete(request, pk):
    """Delete a sales order"""
    from users.views import is_administrator
    if not is_administrator(request.user):
        messages.error(request, 'You do not have permission to delete sales orders.')
        return redirect('sales:sales_list')
    
    sale_order = get_object_or_404(SaleOrder, pk=pk)
    
    if request.method == 'POST':
        order_number = sale_order.order_number
        sale_order.delete()
        messages.success(request, f'Sales order "{order_number}" deleted successfully!')
        return redirect('sales:sales_list')
    
    return render(request, 'sales/sales_confirm_delete.html', {'sale_order': sale_order})


@method_decorator(login_required, name='dispatch')
class SalesDetailView(DetailView):
    model = SaleOrder
    template_name = 'sales/sales_detail.html'
    context_object_name = 'sale_order'