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

class DoctorLinkInline(admin.TabularInline):
    model = DoctorLink

class DoctorQualificationInline(admin.TabularInline):
    model = DoctorQualification

class DoctorMembershipInline(admin.TabularInline):
    model = DoctorMembership

class DoctorAdmin(admin.ModelAdmin):
    formfield_overrides = {models.TextField: {'widget': RichTextEditorWidget}}
    inlines = [DoctorLinkInline, DoctorQualificationInline, DoctorMembershipInline]

class TestAdmin(admin.ModelAdmin):
    formfield_overrides = {models.TextField: {'widget': RichTextEditorWidget}}

admin.site.register(Doctor, DoctorAdmin)
admin.site.register(Partner)
admin.site.register(FoldedBlock)
admin.site.register(CallToAction)
admin.site.register(PatientFeedback)
admin.site.register(Test,TestAdmin)