From d9a9f09bf3f234e063c1761d6739c28d0237c93c Mon Sep 17 00:00:00 2001
From: shahzeydyfah <shahzeydyfah>
Date: Thu, 14 Mar 2024 14:52:02 +0330
Subject: [PATCH] feat(project):add sts

---
 app/models/project.py               | 9 ++++++++-
 app/repositories/design_progress.py | 4 ++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/app/models/project.py b/app/models/project.py
index cdd3460..601e9de 100644
--- a/app/models/project.py
+++ b/app/models/project.py
@@ -3,7 +3,7 @@ import uuid
 
 import pydantic
 from pydantic import Field
-from typing import Optional, List
+from typing import Optional, List, Literal
 
 from core.models.base import BaseModel
 class Budget(pydantic.BaseModel):
@@ -93,6 +93,13 @@ class Design(pydantic.BaseModel):
     designer:Optional[List[str]] = Field(description="نام طراح")
     status_agreement:Optional[str] = Field(description="وضعیت موافقتنامه")
     description:Optional[str] = Field(description="توضیحات")
+    sts:Literal[1,2,3,4] = Field(description="""
+        1.شروع نشده
+        2.جاری
+        3.متوقف
+        4.خاتمه یافته
+    """)
+    reason_stop = Optional[str] = Field(description="دلیل موقف کردن پروژه در این فیلد قرار می گیرد")
 class ProjectModel(BaseModel):
     id: Optional[uuid.UUID] = Field(alias='_id')
     user_id: Optional[int] = Field(description='کد شناسایی کاربر در این فیلد قرار می گیرد')
diff --git a/app/repositories/design_progress.py b/app/repositories/design_progress.py
index e9f6841..f6ef972 100644
--- a/app/repositories/design_progress.py
+++ b/app/repositories/design_progress.py
@@ -54,6 +54,10 @@ class DesignProgressRepository(BaseRepository[CreateDesignProgress]):
         await use_collection(COLLECTION).update_one({"_id":_id},{"$set": {
             "start_at":start_at,"complete_at":_inp.complete_at
         }})
+        if design_progress.type == 1:
+            await use_collection('project').update_one({"_id":design_progress.project_id},{"$set":{"sts":2}})
+        elif design_progress.type == 7:
+            await use_collection('project').update_one({"_id": design_progress.project_id}, {"$set": {"sts": 4}})
         get_record = GetRecord(
             project_id=design_progress.project_id,
             type=design_progress.type,