from django import forms

from lists.models import ListItem, Tag, Entity


class EntityForm(forms.ModelForm):
    class Meta:
        model = Entity
        fields = ['mbid', 'entity_type']


class ListItemForm(forms.ModelForm):
    class Meta:
        model = ListItem
        fields = ['entity', 'tags', 'stars', 'opinion']


class TagForm(forms.ModelForm):
    class Meta:
        model = Tag
        fields = ['name']