From 72b562d25a4a5118fa049ca2fab4ed7c00f5c48e Mon Sep 17 00:00:00 2001
From: shahzeydyfah <shahzeydyfah>
Date: Mon, 25 Mar 2024 20:44:47 +0330
Subject: [PATCH] fix:design progress

---
 app/repositories/design_progress.py | 10 +++++-----
 app/repositories/register_status.py |  7 +++----
 app/urls/design_progress.py         |  4 ++--
 app/urls/register_status.py         |  4 +---
 4 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/app/repositories/design_progress.py b/app/repositories/design_progress.py
index e9783af..207a34a 100644
--- a/app/repositories/design_progress.py
+++ b/app/repositories/design_progress.py
@@ -13,9 +13,8 @@ from core.utils.crudify import mongo_soft_deleter
 
 COLLECTION = "design_progress"
 
-
-class DesignProgressRepository(BaseRepository[CreateDesignProgress]):
-    def __int__(self):
+class DesignProgressRepository(BaseRepository):
+    def __init__(self):
         super().__init__(collection=COLLECTION, model=CreateDesignProgress)
 
     async def create(self, _inp: CreateDesignProgress):
@@ -99,6 +98,7 @@ class DesignProgressRepository(BaseRepository[CreateDesignProgress]):
 
         return result
 
-class DesignProgressRecordRepository(BaseRepository[RecordModel]):
-    def __int__(self):
+class DesignProgressRecordRepository(BaseRepository):
+    def __init__(self):
         super().__init__(collection='design_progress_record', model=RecordModel)
+
diff --git a/app/repositories/register_status.py b/app/repositories/register_status.py
index 338fb24..fbf91cb 100644
--- a/app/repositories/register_status.py
+++ b/app/repositories/register_status.py
@@ -5,10 +5,9 @@ from app.models.register_status import RegisterStatusModel
 from core.repositories.base import BaseRepository
 from core.service.mongo_db import use_collection
 
-
-class RegisterStatusRepository(BaseRepository[RegisterStatusModel]):
-    def __int__(self):
-        super().__init__(collection=TaskEnv.CLN_AUDIT_STATUS,model=RegisterStatusModel)
+class RegisterStatusRepository(BaseRepository):
+    def __init__(self):
+        super().__init__(collection=TaskEnv.CLN_AUDIT_STATUS, model=RegisterStatusModel)
 
     async def get_last(self,project_id:uuid.UUID):
         last = await use_collection(TaskEnv.CLN_AUDIT_STATUS).find({"project_id":project_id}).sort('$natural', -1).to_list(1)
diff --git a/app/urls/design_progress.py b/app/urls/design_progress.py
index b106df8..909cb21 100644
--- a/app/urls/design_progress.py
+++ b/app/urls/design_progress.py
@@ -15,8 +15,8 @@ _tag_base = [router_base_url]
 router = generate_router(router_base_url, auth_dependencies=False, version='v2')
 
 
-repo = DesignProgressRepository(collection='design_progress')
-repo_record = DesignProgressRecordRepository(collection='design_progress_record')
+repo = DesignProgressRepository()
+repo_record = DesignProgressRecordRepository()
 
 @router.post('/create', response_model=DesignProgressModel, tags=_tag_base)
 @shrink(repo.create)
diff --git a/app/urls/register_status.py b/app/urls/register_status.py
index 7449fd1..9f8452e 100644
--- a/app/urls/register_status.py
+++ b/app/urls/register_status.py
@@ -13,9 +13,7 @@ ThisModel = RegisterStatusModel
 router_base_url = 'audit-status'
 _tag_base = [router_base_url]
 router = generate_router(router_base_url, auth_dependencies=False, version='v2')
-
-
-repo = RegisterStatusRepository(collection=TaskEnv.CLN_AUDIT_STATUS)
+repo = RegisterStatusRepository()
 
 @router.post('/create', response_model=ThisModel, tags=_tag_base)
 @shrink(repo.create)