octaflop · August 3, 2011 22:11 · Aug 3, 2011
diff --git a/forms.py b/forms.py
@@ -0,0 +1,38 @@
+from satchless.forms.widgets import DecimalInput
+from satchless.product.forms import BaseVariantForm
+from . import models
+
+class ProductPriceForm(forms.ModelForm):
+    class Meta:
+        widgets = {
+            'price': DecimalInput(min_decimal_places=2),
+        }
+
+class ShoeVariantForm(BaseVariantForm):
+    def __init__(self, *args, **kwargs):
+        super(ShoeVariantForm, self).__init__(*args, **kwargs):
+            self.fields['color'].choices = 
+            [
+                (k,v) for k,v in models.ShoeVariant.COLOR_CHOICES if k in
+                self.product.variants.values_list('color', flat=True).distinct()
+                ]
+
+    color = forms.CharField(
+            max_length=10,
+            widget=forms.Select(choices=models.ShoeVariant.COLOR_CHOICES))
+    size = forms.CharField(
+            max_length=6,
+            widget=forms.Select(choices=models.ShoeVariant.SIZE_CHOICES))
+    def _get_variant_queryset(self):
+        return models.ShoeVariant.objects.filter(
+                product=self.product,
+                color=self.cleaned_data['color'],
+                size=self.cleaned_data['size'],
+            )
+    def clean(self):
+        if not self._get_variant_queryset().exists():
+            raise forms.ValidationError("Variant does not exist")
+        return self.cleaned_data
+
+    def get_variant(self):
+        return self._get_variant_queryset().get()