Add submit_status to note and summary

api/models.py

@@ -2,6 +2,7 @@
 
 from django.contrib.auth.models import AbstractUser
 from django.db import models
+from django.utils.translation import gettext_lazy as _
 
 
 class MerlinBaseModel(models.Model):
@@ -209,6 +210,16 @@ def default(cls):
         return cls.GOAL
 
 
+class SubmitStatus(models.IntegerChoices):
+    DRAFT = 0, _("درفت")
+    INITIAL_SUBMIT = 1, _("ثبت اولیه")
+    FINAL_SUBMIT = 2, _("ثبت نهایی")
+
+    @classmethod
+    def default(cls):
+        return cls.DRAFT
+
+
 class Note(MerlinBaseModel):
     owner = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="نویسنده")
     title = models.CharField(max_length=512, verbose_name="عنوان")
@@ -233,6 +244,11 @@ class Note(MerlinBaseModel):
     linked_notes = models.ManyToManyField(
         "Note", related_name="connected_notes", blank=True, verbose_name="پیوندها"
     )
+    submit_status = models.IntegerField(
+        choices=SubmitStatus.choices,
+        default=SubmitStatus.default(),
+        verbose_name="وضعیت",
+    )
 
     class Meta:
         verbose_name = "یادداشت"
@@ -277,6 +293,11 @@ class Summary(MerlinBaseModel):
     committee_date = models.DateField(
         blank=True, null=True, verbose_name="تاریخ برگزاری جلسه‌ی کمیته"
     )
+    submit_status = models.IntegerField(
+        choices=SubmitStatus.choices,
+        default=SubmitStatus.default(),
+        verbose_name="وضعیت",
+    )
 
     class Meta:
         verbose_name = "جمع‌بندی"

api/serializers.py

@@ -85,6 +85,7 @@ class NoteSerializer(serializers.ModelSerializer):
     )
     read_status = serializers.SerializerMethodField()
     access_level = serializers.SerializerMethodField()
+    submit_status_name = serializers.CharField(source='get_submit_status_display')
 
     class Meta:
         model = Note
@@ -104,6 +105,8 @@ class Meta:
             "linked_notes",
             "read_status",
             "access_level",
+            "submit_status",
+            "submit_status_name",
         )
         read_only_fields = [
             "uuid",
@@ -168,6 +171,7 @@ def create(self, validated_data):
 
 class SummarySerializer(serializers.ModelSerializer):
     note = serializers.SlugRelatedField(read_only=True, slug_field="uuid")
+    submit_status_name = serializers.CharField(source='get_submit_status_display')
 
     class Meta:
         model = Summary
@@ -180,6 +184,8 @@ class Meta:
             "bonus",
             "salary_change",
             "committee_date",
+            "submit_status",
+            "submit_status_name",
         )
         read_only_fields = ["uuid"]