from django import forms
from .models import GENRE_CHOICES, CD


class ExchangeForm(forms.Form):
    name = forms.CharField(max_length=100)
    email = forms.EmailField()
    title = forms.CharField(max_length=100)
    artist = forms.CharField(max_length=40)
    genre = forms.ChoiceField(choices=GENRE_CHOICES)
    price = forms.DecimalField(decimal_places=1, required=False)
    comment = forms.CharField(widget=forms.Textarea, required=False)

    def clean_artist(self):
        data = self.cleaned_data['artist']
        artists = CD.objects.filter(artist__icontains=data.lower())
        if not artists.exists():
            raise forms.ValidationError("Артист не найден!")
        return data