22 lines
615 B
Python
22 lines
615 B
Python
|
from typing import Optional
|
||
|
|
|
||
|
|
from django.urls import URLPattern, re_path
|
||
|
|
from django.views import View
|
||
|
|
from rest_framework import routers
|
||
|
|
|
||
|
|
|
||
|
|
class OptionalSlashRouter(routers.SimpleRouter):
|
||
|
|
"""
|
||
|
|
A router with optional trailing slash at the end
|
||
|
|
APIRouter().register("users", ...) will match both "users" and "users/"
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
self.trailing_slash = "/?"
|
||
|
|
|
||
|
|
|
||
|
|
def optional_slash_path(route: str, view: View, name: Optional[str] = None) -> URLPattern:
|
||
|
|
regex_route = "^{}/?$".format(route)
|
||
|
|
return re_path(route=regex_route, view=view, name=name)
|