nk20/apps/food/forms.py

# Copyright (C) 2018-2024 by BDE ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later

from random import shuffle

from django import forms
from django.utils.translation import gettext_lazy as _
from django.utils import timezone
from member.models import Club
from note_kfet.inputs import Autocomplete, DatePickerInput, DateTimePickerInput
from note_kfet.middlewares import get_current_request
from permission.backends import PermissionBackend

from .models import QR_code, Allergen, BasicFood, TransformedFood


class BasicFoodForms(forms.ModelForm):
    """
    Form for add non-transformed food
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['name'].widget.attrs.update({"autofocus": "autofocus"})
        self.fields['name'].required = True
        self.fields['owner'].required = True
        self.fields['label'].help_text = _('The lot number must be contained in the picture')

        # Some example
        self.fields['name'].widget.attrs.update({"placeholder": _("pasta")})
        clubs = list(Club.objects.filter(PermissionBackend.filter_queryset(get_current_request(), Club, "change")).all())
        shuffle(clubs)
        self.fields['owner'].widget.attrs["placeholder"] = ", ".join(club.name for club in clubs[:4]) + ", ..."
    def clean_dlm_or_dlc(self):
        is_dlc = self.cleaned_data["is_DLC"]
        is_ddm = self.cleaned_data["is_DDM"]
        if is_dlc and is_ddm:
            self.add_error("is_ddm", _("the product cannot be a DLC and a DDM"))
        return is_ddm

    class Meta:
        model = BasicFood
        fields = ('name', 'owner', 'is_DLC', 'is_DDM', 'expiry_date', 'label')        
        widgets = {
            "owner": Autocomplete(
                model = Club,
                attrs = {"api_url": "/api/members/club/"},
            ),
            'expiry_date': DatePickerInput(),
        }



class TransformedFoodForms(forms.ModelForm):
    """
    Form for add transformed food
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['name'].widget.attrs.update({"autofocus": "autofocus"})
        self.fields['name'].required = True
        self.fields['owner'].required = True
        self.fields['creation_date'].required = True
        self.fields['creation_date'].initial = timezone.now
        self.fields['is_active'].initial = True

        # Some example
        self.fields['name'].widget.attrs.update({"placeholder": _("lasagna")})
        clubs = list(Club.objects.filter(PermissionBackend.filter_queryset(get_current_request(), Club, "change")).all())
        shuffle(clubs)
        self.fields['owner'].widget.attrs["placeholder"] = ", ".join(club.name for club in clubs[:4]) + ", ..."

    
    class Meta:
        model = TransformedFood
        fields = ('name', 'creation_date', 'owner', 'is_active',)
        widgets = {
            "owner": Autocomplete(
                model = Club,
                attrs = {"api_url": "/api/members/club/"},
            ),
            'creation_date': DateTimePickerInput(),
        }
class AllergenForms(forms.ModelForm):
    """
    Form for allergen
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
    
    class Meta:
        model = Allergen
        exclude = ['basic_food', 'transformed_food']