Changed onboarding stages nums, added OnboardingStage enum

Author: yakser

users/constants.py

@@ -1,5 +1,15 @@
+from enum import Enum
+
 from django.conf import settings
 
+
+class OnboardingStage(Enum):
+    intro = 0
+    skills = 1
+    account_type = 2
+    completed = None
+
+
 ADMIN = 0
 MEMBER = 1
 MENTOR = 2

users/models.py

@@ -11,6 +11,7 @@
     MENTOR,
     VERBOSE_ROLE_TYPES,
     VERBOSE_USER_TYPES,
+    OnboardingStage,
 )
 from users.managers import (
     CustomUserManager,
@@ -88,9 +89,9 @@ class CustomUser(AbstractUser):
         null=True,
         blank=True,
         editable=False,
-        default=0,
+        default=OnboardingStage.intro.value,
         verbose_name="Стадия онбординга",
-        help_text="0 - не пройден, 1, 2 и 3 - номера стадий онбординга, null(пустое) - онбординг пройден",
+        help_text="0, 1, 2 - номера стадий онбординга, null(пустое) - онбординг пройден",
     )
     verification_date = models.DateField(
         null=True,

users/views.py

@@ -35,6 +35,7 @@
     VERBOSE_ROLE_TYPES,
     VERBOSE_USER_TYPES,
     VERIFY_EMAIL_REDIRECT_URL,
+    OnboardingStage,
 )
 from users.models import UserAchievement, LikesOnProject
 from users.permissions import IsAchievementOwnerOrReadOnly
@@ -354,7 +355,10 @@ def put(self, request: Request, pk):
                     data={"error": "Wrong onboarding stage number!"},
                 )
             # if the user was on the last stage and passed it
-            if request.user.onboarding_stage == 3 and new_stage is None:
+            if (
+                request.user.onboarding_stage == OnboardingStage.account_type.value
+                and new_stage == OnboardingStage.completed.value
+            ):
                 VerificationTasks.create(request.user)
 
             request.user.onboarding_stage = new_stage