From a28d4c22d10ca728eb62938e11b36f079b0bf08f Mon Sep 17 00:00:00 2001 From: berkay Date: Sun, 10 Nov 2024 13:39:34 +0300 Subject: [PATCH] alchemy functions updated --- .../sql_models/building/decision_book.py | 20 +++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/databases/sql_models/building/decision_book.py b/databases/sql_models/building/decision_book.py index 496767e..3329937 100644 --- a/databases/sql_models/building/decision_book.py +++ b/databases/sql_models/building/decision_book.py @@ -110,16 +110,24 @@ class BuildDecisionBook(CrudCollection): year=system_arrow.now().date().year, month=related_date.month, day=1 ) if system_arrow.now().date() <= date_processed: - book = cls.filter_active( + book = cls.filter_one( cls.expiry_ends <= date_processed, cls.decision_type == "RBM", cls.build_id == related_build.id, - ) + ).data if not book: - return - return book.data[0] - else: - return + cls.raise_http_exception( + status_code="HTTP_404_NOT_FOUND", + error_case="NOTFOUND", + message=f"Decision Book is not found for {related_build.build_name}-RBM", + data=dict( + build_id=str(related_build.uu_id), + build_name=related_build.build_name, + decision_type="RBM", + ) + ) + return book + return @classmethod def select_action(cls, duty_id, token=None):