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')