from django import forms
from .models import Dinosaur


class DinosaurForm(forms.ModelForm):
    class Meta:
        model = Dinosaur
        fields = [
            'name', 'era', 'diet', 'description', 'image',
            'speed', 'weight', 'intelligence', 'height',
        ]
        widgets = {
            'name': forms.TextInput(attrs={'class': 'input', 'placeholder': 'T. rex'}),
            'era': forms.Select(attrs={'class': 'input'}),
            'diet': forms.Select(attrs={'class': 'input'}),
            'description': forms.Textarea(attrs={'class': 'textarea', 'rows': 3}),
            'image': forms.ClearableFileInput(attrs={'class': 'file-input'}),
            'speed': forms.NumberInput(attrs={'class': 'input', 'min': 0, 'max': 100}),
            'weight': forms.NumberInput(attrs={'class': 'input', 'min': 0, 'max': 100}),
            'intelligence': forms.NumberInput(attrs={'class': 'input', 'min': 0, 'max': 100}),
            'height': forms.NumberInput(attrs={'class': 'input', 'min': 0, 'max': 100}),
        }