From 8fb4e461e960112d3377833c31b6883b195e7164 Mon Sep 17 00:00:00 2001 From: phzhik Date: Thu, 30 May 2024 20:59:46 +0400 Subject: [PATCH] * SimpleRouter for non-debug environment --- account/urls.py | 4 ++-- core/urls.py | 4 ++-- poizonstore/utils.py | 6 ++++++ store/urls.py | 4 ++-- 4 files changed, 12 insertions(+), 6 deletions(-) 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')