feat(project):add sts

This commit is contained in:
shahzeydyfah
2024-03-14 14:52:02 +03:30
parent ebb63de1e8
commit d9a9f09bf3
2 changed files with 12 additions and 1 deletions
+8 -1
View File
@@ -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='کد شناسایی کاربر در این فیلد قرار می گیرد')
+4
View File
@@ -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,