from wagtail import hooks

from .admin import event_sidebar_viewset, event_explorer_viewset
from .views import event_organizer_chooser_viewset


@hooks.register("register_admin_viewset")
def register_viewset():
    return event_organizer_chooser_viewset


@hooks.register("register_admin_viewset")
def register_event_sidebar_viewset():
    return event_sidebar_viewset


@hooks.register("register_admin_viewset")
def register_event_explorer_viewset():
    return event_explorer_viewset