from django.contrib import admin
from unfold.admin import ModelAdmin

from core.apps.api.models import AdModel, AdImage, AdVariant


class AdImageInline(admin.TabularInline):
    model = AdImage
    extra = 1


class AdVariantInline(admin.TabularInline):
    model = AdVariant
    extra = 1


@admin.register(AdModel)
class AdModelAdmin(ModelAdmin):
    list_display = (
        "id",
        "__str__",
    )
    inlines = [AdImageInline, AdVariantInline]