diff --git a/account/urls.py b/account/urls.py index 5ac638b..b787599 100644 --- a/account/urls.py +++ b/account/urls.py @@ -1,9 +1,9 @@ from django.urls import path, include -from rest_framework.routers import DefaultRouter from account import views +from poizonstore.utils import get_drf_router -router = DefaultRouter() +router = get_drf_router() router.register("users", views.UserViewSet) urlpatterns = [ diff --git a/core/urls.py b/core/urls.py index 970e272..8b966a6 100644 --- a/core/urls.py +++ b/core/urls.py @@ -1,9 +1,9 @@ from django.urls import path -from rest_framework.routers import DefaultRouter +from poizonstore.utils import get_drf_router from . import views -router = DefaultRouter() +router = get_drf_router() urlpatterns = [ diff --git a/poizonstore/utils.py b/poizonstore/utils.py index ef71881..ea6cabe 100644 --- a/poizonstore/utils.py +++ b/poizonstore/utils.py @@ -1,7 +1,9 @@ from functools import reduce +from django.conf import settings from django_filters import DateFromToRangeFilter as _DateFromToRangeFilter from rest_framework.fields import DecimalField +from rest_framework.routers import SimpleRouter, DefaultRouter class PriceField(DecimalField): @@ -20,3 +22,7 @@ class DateFromToRangeFilter(_DateFromToRangeFilter): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.field.widget.suffixes = ['from', 'to'] + + +def get_drf_router(): + return DefaultRouter() if settings.DEBUG else SimpleRouter() diff --git a/store/urls.py b/store/urls.py index bd4aeaf..15cc0d9 100644 --- a/store/urls.py +++ b/store/urls.py @@ -1,9 +1,9 @@ from django.urls import path -from rest_framework.routers import DefaultRouter +from poizonstore.utils import get_drf_router from store import views -router = DefaultRouter() +router = get_drf_router() router.register(r'checklist', views.ChecklistAPI, basename='checklist')