Coverage for website/sales/forms.py: 34.78%

1from django import forms 

2 

3from sales.models.product import ProductListItem 

4 

5 

6class ProductItemField(forms.IntegerField): 

7 def get_productlistitem(self): 

8 return self._productlistitem 

9 

10 def get_product(self): 

11 return self._productlistitem.product 

12 

13 def set_productlistitem(self, product_list_item): 

14 self._productlistitem = product_list_item 

15 

16 

17class ProductOrderForm(forms.Form): 

18 def __init__(self, product_list, is_adult, *args, **kwargs): 

19 super().__init__(*args, **kwargs) 

20 product_list_items = ProductListItem.objects.filter(product_list=product_list) 

21 for item in product_list_items: 

22 field = ProductItemField( 

23 label=item.product.name, 

24 required=False, 

25 min_value=0, 

26 ) 

27 field.set_productlistitem(item) 

28 if not is_adult and item.product.age_restricted: 

29 field.disabled = True 

30 self.fields[f"product_{item.product.pk}"] = field