OpenConceptLab/ocl_issues#2121 | throttling | lite --> standard

Author: snyaggarwal

core/common/throttling.py

@@ -11,12 +11,12 @@ class GuestDayThrottle(UserRateThrottle):
     scope = 'guest_day'
 
 
-class LiteMinuteThrottle(UserRateThrottle):
-    scope = 'lite_minute'
+class StandardMinuteThrottle(UserRateThrottle):
+    scope = 'standard_minute'
 
 
-class LiteDayThrottle(UserRateThrottle):
-    scope = 'lite_day'
+class StandardDayThrottle(UserRateThrottle):
+    scope = 'standard_day'
 
 
 class ThrottleUtil:
@@ -42,4 +42,4 @@ def get_throttles_by_user_plan(user):
         # order is important, first one has to be minute throttle
         if get(user, 'api_rate_limit.is_guest') or not get(user, 'is_authenticated'):
             return [GuestMinuteThrottle(), GuestDayThrottle()]
-        return [LiteMinuteThrottle(), LiteDayThrottle()]
+        return [StandardMinuteThrottle(), StandardDayThrottle()]

core/integration_tests/tests_users.py

@@ -792,7 +792,7 @@ def setUp(self):
 
     def test_put_bad_request(self):
         user = UserProfileFactory()
-        self.assertEqual(user.api_rate_limit.rate_plan, 'lite')
+        self.assertEqual(user.api_rate_limit.rate_plan, 'standard')
 
         response = self.client.put(
             f'/users/{user.username}/rate-limit/',
@@ -805,12 +805,12 @@ def test_put_bad_request(self):
             response.data,
             {
                 'detail': ErrorDetail(
-                    string='"rate_plan" needs to be one of "guest" or "lite"', code='bad_request')
+                    string='"rate_plan" needs to be one of "guest" or "standard"', code='bad_request')
             }
         )
 
         user.refresh_from_db()
-        self.assertEqual(user.api_rate_limit.rate_plan, 'lite')
+        self.assertEqual(user.api_rate_limit.rate_plan, 'standard')
 
         response = self.client.put(
             f'/users/{user.username}/rate-limit/',
@@ -821,7 +821,7 @@ def test_put_bad_request(self):
         self.assertEqual(response.status_code, 400)
         self.assertEqual(response.data, {'rate_plan': ["Value 'blah' is not a valid choice."]})
 
-        self.assertEqual(user.api_rate_limit.rate_plan, 'lite')
+        self.assertEqual(user.api_rate_limit.rate_plan, 'standard')
 
         response = self.client.put(
             f'/users/{user.username}/rate-limit/',
@@ -833,7 +833,7 @@ def test_put_bad_request(self):
 
     def test_put_204(self):
         user = UserProfileFactory()
-        self.assertEqual(user.api_rate_limit.rate_plan, 'lite')
+        self.assertEqual(user.api_rate_limit.rate_plan, 'standard')
 
         response = self.client.put(
             f'/users/{user.username}/rate-limit/',

core/settings.py

@@ -182,8 +182,8 @@
     REST_FRAMEWORK['DEFAULT_THROTTLE_RATES'] = {
         'guest_minute': '400/minute',
         'guest_day': '10000/day',
-        'lite_minute': '500/minute',
-        'lite_day': '20000/day',
+        'standard_minute': '500/minute',
+        'standard_day': '20000/day',
     }
     MIDDLEWARE = [*MIDDLEWARE, 'core.middlewares.middlewares.ThrottleHeadersMiddleware']
 

core/users/migrations/0033_auto_20250318_0212.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.2.16 on 2025-03-18 02:12
+
+from django.db import migrations
+
+
+def update_api_rate_limit_plan(apps, schema_editor):
+    UserRateLimit = apps.get_model('users', 'UserRateLimit')
+    UserRateLimit.objects.update(rate_plan='standard')
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('users', '0032_alter_userratelimit_rate_plan'),
+    ]
+
+    operations = [
+        migrations.RunPython(update_api_rate_limit_plan),
+    ]

core/users/models.py

@@ -44,34 +44,34 @@ def uri(self):
 
 
 class UserRateLimit(BaseModel):
-    LITE_PLAN = 'lite'
+    STANDARD_PLAN = 'standard'
     GUEST_PLAN = 'guest'
 
     class Meta:
         db_table = 'user_api_rate_limits'
 
     RATE_PLANS = (
-        (LITE_PLAN, LITE_PLAN),
+        (STANDARD_PLAN, STANDARD_PLAN),
         (GUEST_PLAN, GUEST_PLAN),
     )
 
     user = models.OneToOneField('users.UserProfile', on_delete=models.CASCADE, related_name='api_rate_limit')
-    rate_plan = models.CharField(choices=RATE_PLANS, max_length=100, default=LITE_PLAN)
+    rate_plan = models.CharField(choices=RATE_PLANS, max_length=100, default=STANDARD_PLAN)
     public_access = None
     uri = None
     extras = None
 
     @property
-    def is_lite(self):
-        return self.rate_plan == self.LITE_PLAN
+    def is_standard(self):
+        return self.rate_plan == self.STANDARD_PLAN
 
     @property
     def is_guest(self):
         return self.rate_plan == self.GUEST_PLAN
 
     def clean(self):
         if not self.rate_plan:
-            self.rate_plan = self.LITE_PLAN
+            self.rate_plan = self.STANDARD_PLAN
         super().clean()
 
     @classmethod

core/users/views.py

@@ -475,7 +475,7 @@ def update(self, request, *args, **kwargs):
         rate_plan = self.request.data.get('rate_plan', None)
 
         if not rate_plan:
-            raise Http400('"rate_plan" needs to be one of "guest" or "lite"')
+            raise Http400('"rate_plan" needs to be one of "guest" or "standard"')
         try:
             UserRateLimit.upsert(self.get_object(), rate_plan, request.user)
         except ValidationError as ex: