from django.contrib import admin

from .models import Audience, Document, DocumentType, Site, Structure


class DocumentAdmin(admin.ModelAdmin):
    list_filter = ('type', 'audiences', 'sites', 'structures')


class DocumentTypeAdmin(admin.ModelAdmin):
    list_display = ('__str__', 'level')


admin.site.register(Audience)
admin.site.register(Site)
admin.site.register(Structure)
admin.site.register(Document, DocumentAdmin)
admin.site.register(DocumentType, DocumentTypeAdmin)