from wagtail.admin.viewsets.model import ModelViewSet as BaseModelViewSet
from wagtail.snippets.views.snippets import SnippetViewSet as BaseSnippetViewSet

from .helpers import LazyModelViewSetMixin


class ModelAdmin(LazyModelViewSetMixin, BaseSnippetViewSet):
    pass


class SnippetViewSet(LazyModelViewSetMixin, BaseSnippetViewSet):
    pass


class ModelViewSet(LazyModelViewSetMixin, BaseModelViewSet):
    pass
