from django.contrib import admin
from solo.admin import SingletonModelAdmin
from djboomin.widgets import RichTextEditorWidget
from models import *

class PageAdmin(SingletonModelAdmin):
    formfield_overrides = {models.TextField: {'widget': RichTextEditorWidget}}

class OtherPageAdmin(admin.ModelAdmin):
    list_display = ['name', 'slug', 'last_modified', 'show_in_main_nav', 'show_in_footer_nav']
    prepopulated_fields = {"slug": ("name",)}
    formfield_overrides = {models.TextField: {'widget': RichTextEditorWidget}}

admin.site.register(Homepage, PageAdmin)
admin.site.register(AboutPage, PageAdmin)
admin.site.register(NewsPage, PageAdmin)
admin.site.register(TestsPage, PageAdmin)
admin.site.register(FAQPage, PageAdmin)
admin.site.register(ContactPage, PageAdmin)
admin.site.register(Page, OtherPageAdmin)