from django.contrib import admin

from .models import AboutUs, Branch, Description


class DescriptionInline(admin.StackedInline):
    model = Description
    fields = ['desc_title', 'desc_text', ]
    extra = 0


class BranchInline(admin.StackedInline):
    model = Branch
    fields = ['branch_name', 'image', ]
    extra = 0


class AboutUsAdmin(admin.ModelAdmin):
    inlines = [
        DescriptionInline,
        BranchInline
    ]


admin.site.register(AboutUs, AboutUsAdmin)