from registrasion import forms
from registrasion import models as rego
from registrasion.controllers.cart import CartController
from registrasion.controllers.invoice import InvoiceController

from django.contrib.auth.decorators import login_required
from django.core.exceptions import ObjectDoesNotExist
from django.db import transaction
from django.forms import formset_factory
from django.shortcuts import redirect
from django.shortcuts import render
from functools import partial, wraps


@login_required
def product_category(request, category_id):
    ''' Registration selections form for a specific category of items '''

    category_id = int(category_id)  # Routing is [0-9]+
    category = rego.Category.objects.get(pk=category_id)

    ProductItemFormForCategory = (
        wraps(forms.ProductItemForm)
        (partial(forms.ProductItemForm, category=category)))
    ProductItemFormSet = formset_factory(ProductItemFormForCategory, extra=0)

    if request.method == "POST":
        formset = ProductItemFormSet(request.POST, request.FILES)
        if formset.is_valid():
            current_cart = CartController.for_user(request.user)
            with transaction.atomic():
                for form in formset.forms:
                    data = form.cleaned_data
                    # TODO set form error instead of failing completely
                    current_cart.set_quantity(
                        data["product"], data["quantity"], batched=True)
                current_cart.end_batch()
    else:
        # Create initial data for each of products in category
        initial = []
        products = rego.Product.objects.filter(category=category)
        items = rego.ProductItem.objects.filter(product__category=category)
        products = products.order_by("order")
        for product in products:
            try:
                quantity = items.get(product=product).quantity
            except ObjectDoesNotExist:
                quantity = 0
            data = {"product": product, "quantity": quantity}
            initial.append(data)

        formset = ProductItemFormSet(initial=initial)

    data = {
        "category": category,
        "formset": formset,
    }

    return render(request, "product_category.html", data)

@login_required
def checkout(request):
    ''' Runs checkout for the current cart of items, ideally generating an
    invoice. '''

    current_cart = CartController.for_user(request.user)
    current_invoice = InvoiceController.for_cart(current_cart.cart)

    return redirect("invoice", current_invoice.invoice.id)


@login_required
def invoice(request, invoice_id):
    ''' Displays an invoice for a given invoice id. '''

    invoice_id = int(invoice_id)
    inv = rego.Invoice.objects.get(pk=invoice_id)
    current_invoice = InvoiceController(inv)

    data = {
        "invoice": current_invoice.invoice,
    }

    return render(request, "invoice.html", data)