from typing import Type

from rest_framework.routers import DefaultRouter

from dj_kaos.models.rest import GenericKaosModelViewSet


class KaosRestRouter(DefaultRouter):
    def get_default_basename(self, viewset: Type[GenericKaosModelViewSet]):
        try:
            return super().get_default_basename(viewset)
        except AssertionError:
            return viewset.model._meta.object_name.lower()