From 76ee60e90acecc86ab19fdbdf8b614d1915be429 Mon Sep 17 00:00:00 2001
From: shahzeydyfah <shahzeydyfah>
Date: Sat, 23 Mar 2024 14:03:39 +0330
Subject: [PATCH] fix:design progress

---
 app/repositories/design_progress.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/app/repositories/design_progress.py b/app/repositories/design_progress.py
index ff4ee69..2f5bc2c 100644
--- a/app/repositories/design_progress.py
+++ b/app/repositories/design_progress.py
@@ -67,11 +67,12 @@ class DesignProgressRepository(BaseRepository[CreateDesignProgress]):
     async def check_complete(self,get_record:GetRecord):
         result = False
         if get_record.type and get_record.type < 7:
-            progress = await use_collection(COLLECTION).find_one(
-                {"project_id": get_record.project_id, "type": int(get_record.type)+1})
+            progress = await use_collection(COLLECTION).find(
+                {"project_id": get_record.project_id, "type": int(get_record.type)+1}).to_list(None)
             if progress:
+                ids = [i["_id"] for i in progress]
                 records = await use_collection('design_progress_record').find_one(
-                    {"design_progress_id": progress["_id"]})
+                    {"design_progress_id": {"$in":ids}})
                 if records:
                     result=True
         return result