from django.forms import ModelForm, Textarea, Select, ModelMultipleChoiceField, CheckboxSelectMultiple
from main.models import Mood, Activity, Tag


class MoodModelForm(ModelForm):

    activities = ModelMultipleChoiceField(
        queryset=Activity.objects.all().order_by('name'),
        widget=CheckboxSelectMultiple(attrs={"class": "checkbox"}),
        required=False
    )
    tags = ModelMultipleChoiceField(
        queryset=Tag.objects.all().order_by('name'),
        widget=CheckboxSelectMultiple(attrs={"class": "checkbox"}),
        required=False
    )

    class Meta:
        model = Mood
        fields = ['moment', 'comment', 'mood', 'activities', 'tags']
        labels = {'comment': 'Notiz',
                  'moment': 'Zeit',
                  'mood': 'Stimmung'}
        widgets = {
            'comment': Textarea(attrs={'cols': 24, 'rows': 4}),
        }