from django.contrib import admin
from import_export.admin import ImportExportModelAdmin
from .models import ClickableArea, Element3D, Environment, Scene3D


class Scene3DAdmin(ImportExportModelAdmin, admin.ModelAdmin):
    filter_horizontal = ("elements",)

class ImportExportBtnAdmin(ImportExportModelAdmin, admin.ModelAdmin):
    pass

admin.site.register(Environment, ImportExportBtnAdmin)
admin.site.register(Scene3D, Scene3DAdmin)
admin.site.register(Element3D, ImportExportBtnAdmin)
admin.site.register(ClickableArea, ImportExportBtnAdmin)