From ced12fb0143d87f03bae6d146d2196bcb153a7c6 Mon Sep 17 00:00:00 2001
From: shahzeydyfah <shahzeydyfah>
Date: Tue, 21 Jan 2025 00:04:22 +0330
Subject: [PATCH] feat(financial_adjust):add update project_price

---
 app/repositories/financial_adjust.py | 38 ++++++++++++++++++++++++++++
 app/repositories/financial_status.py |  1 -
 2 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/app/repositories/financial_adjust.py b/app/repositories/financial_adjust.py
index d5f9e8d..04c7fd7 100644
--- a/app/repositories/financial_adjust.py
+++ b/app/repositories/financial_adjust.py
@@ -3,10 +3,48 @@ import uuid
 from app.config import TaskEnv
 from app.models.financial_adjust import AdjustFinancialModel
 from core.repositories.base import BaseRepository
+from core.service.mongo_db import use_collection
 
 
 class FinancialAdjustRepository(BaseRepository):
     def __init__(self):
         super().__init__(collection=TaskEnv.CLN_FINANCIAL_ADJUST, model=AdjustFinancialModel)
 
+    async def create(self, coding: AdjustFinancialModel):
+        result = await super().create(coding)
+        price = await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).find_one({'project_id': coding.project_id})
+        if price and coding.amount:
+            await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).update_one(
+                {'project_id': coding.project_id},
+                {"$inc": {'amount_adjust': coding.amount}}
+            )
+        return result
 
+    async def update(self, _id: uuid.UUID, _update: AdjustFinancialModel):
+        _old_dic = await use_collection(TaskEnv.CLN_FINANCIAL_ADJUST).find_one({'_id':_id})
+        change = 0
+        result = await super().update(_id, _update)
+        price = await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).find_one({'project_id': _update.project_id})
+        if price:
+            _old = AdjustFinancialModel(**_old_dic)
+            if 'amount' not in _old_dic and _update.amount:
+                change = _update.amount
+            if _old.amount != _update.amount:
+                change = _update.amount - _old.amount
+            if change != 0 :
+                await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).update_one(
+                    {'project_id': _update.project_id},
+                    {"$inc": {'amount_adjust': change}}
+                )
+        return result
+
+    async def delete(self, _id: uuid.UUID):
+        _old_dic = await use_collection(TaskEnv.CLN_FINANCIAL_ADJUST).find_one({'_id': _id})
+        await super().delete(_id)
+        price = await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).find_one({'project_id': _old_dic['project_id']})
+        if price and 'amount_adjust' in price:
+            await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).update_one(
+                {'project_id': _old_dic['project_id']},
+                {"$inc": {'amount_adjust': -(_old_dic['amount'])}}
+            )
+        return True
diff --git a/app/repositories/financial_status.py b/app/repositories/financial_status.py
index 76aa80f..d19e231 100644
--- a/app/repositories/financial_status.py
+++ b/app/repositories/financial_status.py
@@ -26,7 +26,6 @@ class FinancialStatusRepository(BaseRepository):
         result = await super().update(_id, _update)
         price = await use_collection(TaskEnv.CLN_PROJECT_PRICE_REPORT).find_one({'project_id': _update.project_id})
         if price:
-            change = 0
             _old = StatusFinancialModel(**_old_dic)
             if 'amount' not in _old_dic and _update.amount:
                 change = _update.amount