Django-Rest-Framework custom router that accept urls with or without slash at the end

urls.py

from rest_framework.routers import DefaultRouter

class OptionalTraillingSlashRouter(DefaultRouter):
    def __init__(self, trailing_slash=True):
        super().__init__(trailing_slash)
        self.trailing_slash = "/?"

    def get_lookup_regex(self, viewset):
        base_regex = '(?P<{lookup_field}>[^/.]+)'
        lookup_field = getattr(viewset, 'lookup_field', 'pk')
        return base_regex.format(lookup_field=lookup_field)

''' 
    router = OptionalTraillingSlashRouter()
    router.register(r'test', TestViewSet, basename='test')

    #  This router accept both url
    # test/1/
    # test/1

'''