from django import forms
from django.forms import modelformset_factory

from shop.models import OrderProductRelation


class OrderProductRelationForm(forms.ModelForm):
    class Meta:
        model = OrderProductRelation
        fields = ["quantity"]

    def clean_quantity(self):
        product = self.instance.product
        new_quantity = self.cleaned_data["quantity"]

        if product.stock_amount and product.left_in_stock < new_quantity:
            raise forms.ValidationError(
                "Only {} left in stock.".format(product.left_in_stock)
            )

        return new_quantity


OrderProductRelationFormSet = modelformset_factory(
    OrderProductRelation, form=OrderProductRelationForm, extra=0
)