from admin_ordering.admin import OrderableAdmin
from django.contrib import admin
from . import models


@admin.register(models.Package)
class PackageAdmin(admin.ModelAdmin):
    raw_id_fields = ['books']


class MediaInsertTextInline(OrderableAdmin, admin.TabularInline):
    model = models.MediaInsertText
    extra = 0


class PriceLevelInline(OrderableAdmin, admin.TabularInline):
    model = models.PriceLevel
    extra = 0


@admin.register(models.Site)
class SiteAdmin(admin.ModelAdmin):
    inlines = [
        MediaInsertTextInline,
        PriceLevelInline,
    ]

@admin.register(models.SiteBookPublish)
class SiteBookPublishAdmin(admin.ModelAdmin):
    list_display = ['created_at', 'site_book', 'user', 'status', 'started_at', 'finished_at']
    list_filter = ['status', 'site_book__site']
    search_fields = ['book', 'user']
    date_hierarchy = 'started_at'
    raw_id_fields = ['site_book']