from django.contrib import admin
from django import forms
from .models import Element, ElementDescription, ElementSeparator


class ElementDescriptionAdminForm(forms.ModelForm):
    class Meta:
        model = ElementDescription
        widgets = {
            'description': forms.widgets.Textarea,
        }
        fields = '__all__'


class ElementDescriptionInline(admin.StackedInline):
    model = ElementDescription
    form = ElementDescriptionAdminForm
    extra = 2


class ElementAdmin(admin.ModelAdmin):
    list_display = ('__str__', 'parent', 'link', 'weight')
    ordering = ('parent', 'weight')
    inlines = [ElementDescriptionInline]


admin.site.register(Element, ElementAdmin)
admin.site.register(ElementSeparator)