Merge pull request #3646 from TheSpaceDevs/add-ratelimit

feat: add rate-limit

.vscode/launch.json

@@ -0,0 +1,15 @@
+{
+    "configurations": [
+        {
+            "type": "debugpy",
+            "request": "launch",
+            "name": "Launch Django App",
+            "program": "${workspaceFolder}/src/manage.py",
+            "args": [
+                "runserver",
+                "0.0.0.0:8000"
+            ],
+            "django": true
+        }
+    ]
+}

pyproject.toml

@@ -37,6 +37,7 @@ dev-dependencies = [
     "bandit[toml]>=1.7.9",
     "commitizen>=4.0.0",
     "coverage>=7.6.1",
+    "django-debug-toolbar>=4.4.6",
     "django-filter-stubs>=0.1.3",
     "djangorestframework-stubs[compatible-mypy]>=3.14.0,<3.15.0",
     "mypy>=1.7.0,<1.8.0",

src/snapy/settings.py

@@ -80,9 +80,11 @@
     "health_check.contrib.s3boto3_storage",
     "health_check.contrib.redis",
     "graphene_django",
+    "debug_toolbar",
 ]
 
 MIDDLEWARE = [
+    "debug_toolbar.middleware.DebugToolbarMiddleware",
     "corsheaders.middleware.CorsMiddleware",
     "django.middleware.security.SecurityMiddleware",
     "django.contrib.sessions.middleware.SessionMiddleware",
@@ -188,6 +190,12 @@
     "DEFAULT_FILTER_BACKENDS": ["django_filters.rest_framework.DjangoFilterBackend"],
     "DEFAULT_SCHEMA_CLASS": "drf_spectacular.openapi.AutoSchema",
     "DEFAULT_VERSIONING_CLASS": "rest_framework.versioning.NamespaceVersioning",
+    'DEFAULT_THROTTLE_CLASSES': [
+        'rest_framework.throttling.AnonRateThrottle',
+    ],
+    'DEFAULT_THROTTLE_RATES': {
+        'anon': '5/second',
+    }
 }
 
 SPECTACULAR_SETTINGS = {
@@ -230,3 +238,7 @@
         "LOCATION": env.str("REDIS_URL", "redis://localhost:6379"),
     }
 }
+
+INTERNAL_IPS = [
+    "127.0.0.1",
+]

src/snapy/urls.py

@@ -17,6 +17,7 @@
 
 """
 
+from debug_toolbar.toolbar import debug_toolbar_urls
 from django.contrib import admin
 from django.urls import include, re_path
 from django.views.decorators.csrf import csrf_exempt
@@ -33,4 +34,4 @@
     # Non v4 URLs
     re_path(r"health/", include("health_check.urls")),
     re_path(r"admin/", admin.site.urls),
-]
+] + debug_toolbar_urls()