23 lines
721 B
Python
23 lines
721 B
Python
from django import forms
|
|
from django.contrib.auth.forms import UserCreationForm
|
|
from django.contrib.auth.models import User
|
|
from .models import Consumption
|
|
|
|
class ConsumptionForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Consumption
|
|
fields = ['drink_type', 'size']
|
|
labels = {
|
|
'drink_type': 'Drink Type',
|
|
'size': 'Size'
|
|
}
|
|
widgets = {
|
|
'drink_type': forms.Select(attrs={'class': 'form-select form-select-lg mb-3'}),
|
|
'size': forms.Select(attrs={'class': 'form-select form-select-lg mb-3'}),
|
|
}
|
|
|
|
class SignUpForm(UserCreationForm):
|
|
class Meta(UserCreationForm.Meta):
|
|
model = User
|
|
fields = ('username', 'email')
|