From 1d6a2662c3da759dc6f8211523c4a6ddc70427e5 Mon Sep 17 00:00:00 2001 From: Josh Washburne Date: Tue, 3 Apr 2018 16:23:17 -0400 Subject: [PATCH] Deal with optional trailing slashes. --- savepointradio/api/urls.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/savepointradio/api/urls.py b/savepointradio/api/urls.py index 8ba2f3d..5b13c18 100644 --- a/savepointradio/api/urls.py +++ b/savepointradio/api/urls.py @@ -7,7 +7,19 @@ from api.views.radio import (AlbumViewSet, ArtistViewSet, GameViewSet, SongViewSet) -router = DefaultRouter() +class OptionalSlashRouter(DefaultRouter): + ''' + Custom Router that allows for a trailing slash to be optional in the + endpoint URI. + + (Thanks to https://stackoverflow.com/questions/46163838/) + ''' + def __init__(self): + super().__init__() + self.trailing_slash = '/?' + + +router = OptionalSlashRouter() router.register(r'albums', AlbumViewSet, base_name='album') router.register(r'artists', ArtistViewSet, base_name='artist')