session maker updated

BankServices/WriterService/app.py

@@ -18,7 +18,7 @@ def collect_parsed_data_from_mongo_database(mongo_provider) -> list:
 def write_parsed_data_to_account_records(
     file: str, data_dict: dict, collection_name: str, mongo_provider
 ):
-    db_session = AccountRecords.new_session()
+    with AccountRecords.new_session() as db_session:
         data_dict["bank_balance"] = data_dict.pop("balance")
         data_dict["import_file_name"] = collection_name
         data_dict = BankReceive(**data_dict).model_dump()

Services/PostgresService/controllers/core_controllers.py

@@ -24,11 +24,9 @@ class BaseAlchemyModel:
     __abstract__ = True
 
     @classmethod
-    def new_session(cls) -> Session:
+    def new_session(cls):
         """Get database session."""
-
-        with get_db() as session:
-            return session
+        return get_db()
 
     @classmethod
     def flush(cls: Type[T], db: Session) -> T: