la-chariotte/la_chariotte/order/forms.py
2023-07-07 15:07:01 +02:00

85 lines
2.7 KiB
Python

import datetime
from django import forms
from django.contrib.auth import get_user_model
from django.core.exceptions import NON_FIELD_ERRORS, ValidationError
from django.db import IntegrityError
from la_chariotte import settings
from la_chariotte.order.models import GroupedOrder, Item
class GroupedOrderForm(forms.ModelForm):
deadline_date = forms.DateField(
label="Date limite de commande", widget=forms.DateInput(attrs={"type": "date"})
)
deadline_time = forms.TimeField(
label="Heure limite de commande",
widget=forms.TimeInput(attrs={"type": "time"}),
initial=datetime.time(hour=23, minute=59, second=59),
)
class Meta:
model = GroupedOrder
fields = [
"name",
"deadline_date",
"deadline_time",
"delivery_date",
"place",
"description",
]
widgets = {
"name": forms.TextInput(
attrs={"placeholder": "ex : Des oranges pour Noël"}
),
"delivery_date": forms.DateInput(attrs={"type": "date"}),
"place": forms.TextInput(attrs={"placeholder": "(facultatif)"}),
"description": forms.Textarea(
attrs={
"placeholder": "Plus d'infos sur la commande groupée ? (facultatif)"
}
),
}
def __init__(self, *args, **kwargs):
self.user = kwargs.pop("user")
super().__init__(*args, **kwargs)
def save(self, commit=True):
self.instance.orga = get_user_model().objects.get(id=self.user.pk)
return super().save(commit=commit)
def clean(self):
cleaned_data = self.cleaned_data
self.instance.deadline = datetime.datetime.combine(
self.cleaned_data["deadline_date"], self.cleaned_data["deadline_time"]
)
try:
GroupedOrder.objects.get(
name=cleaned_data["name"],
delivery_date=cleaned_data["delivery_date"],
orga=get_user_model().objects.get(id=self.user.pk),
)
except GroupedOrder.DoesNotExist:
pass
else:
raise ValidationError(
"Une commande groupée avec le même titre et la même date existe déjà"
)
return cleaned_data
class ItemCreateForm(forms.ModelForm):
class Meta:
model = Item
fields = ["name", "price", "max_limit"]
def __init__(self, *args, **kwargs):
self.grouped_order = kwargs.pop("grouped_order") # type: GroupedOrder
super().__init__(*args, **kwargs)
def save(self, commit=True):
self.instance.grouped_order = GroupedOrder.objects.get(id=self.grouped_order.pk)
return super().save(commit=commit)