from django.contrib.gis import admin
from mptt.admin import DraggableMPTTAdmin, TreeRelatedFieldListFilter
from utils.admin import BaseModelAdminMixin

from collection import models


@admin.register(models.Collection)
class CollectionAdmin(admin.ModelAdmin):
    """Collection admin."""


@admin.register(models.Guide)
class GuideAdmin(admin.ModelAdmin):
    """Guide admin."""


@admin.register(models.GuideElementType)
class GuideElementType(admin.ModelAdmin):
    """Guide element admin."""


@admin.register(models.GuideElement)
class GuideElementAdmin(DraggableMPTTAdmin, BaseModelAdminMixin, admin.ModelAdmin):
    """Guide element admin."""
    raw_id_fields = [
        'guide_element_type', 'establishment', 'review',
        'wine_region', 'product', 'city',
        'wine_color_section', 'section', 'guide',
        'parent',
    ]
    # list_filter = (
    #     ('parent', TreeRelatedFieldListFilter),
    # )