add level property in Profile & add in Profile view

0seoYun · Aug 17, 2023 · b0b1897 · b0b1897
1 parent 2097f13
commit b0b1897
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/accounts/models.py b/accounts/models.py
@@ -20,6 +20,12 @@ class Profile(BaseModel):
     info = models.JSONField(default={})
     completed_cnt = models.IntegerField(default=0)
 
+    @property
+    def level(self):
+        level = self.completed_cnt // 3 + 1
+        level = 5 if (level > 5) else level
+        return level
+
 
 class ProfileReport(ReportBaseModel):
     profile = models.ForeignKey(

diff --git a/accounts/serializers.py b/accounts/serializers.py
@@ -20,6 +20,7 @@ class Meta:
             "type",
             "user_id",
             "joined_at",
+            "level",
         ]
         read_only_fields = [
             "id",
@@ -29,6 +30,7 @@ class Meta:
             "completed_cnt",
             "type",
             "user_id",
+            "level",
         ]
 
     def get_following_cnt(self, instance):

diff --git a/core/serializers.py b/core/serializers.py
@@ -7,12 +7,7 @@ class WriterSerializer(serializers.ModelSerializer):
     profile_id = serializers.IntegerField(source="profile.id")
     nickname = serializers.CharField(source="profile.nickname")
     type = serializers.CharField(source="profile.type")
-    level = serializers.SerializerMethodField(read_only=True)
-
-    def get_level(self, instance):
-        level = instance.profile.completed_cnt // 3 + 1
-        level = 5 if (level > 5) else level
-        return level
+    level = serializers.IntegerField(source="profile.level")
 
     class Meta:
         model = User