fix import issue

Author: djeck1432

apps/data_handler/db/models/__init__.py

@@ -1,18 +1,15 @@
 """ This module contains all the models used in the database. """
 from .base import Base
 from .liquidable_debt import HealthRatioLevel, LiquidableDebt
-from .loan_states import (
-    InterestRate,
-    LoanState,
-    ZkLendCollateralDebt,
-)
+from .loan_states import InterestRate, LoanState, ZkLendCollateralDebt
 from .order_book import OrderBookModel
+from .vesu import VesuPosition
 from .zklend_events import (
     AccumulatorsSyncEventModel,
-    LiquidationEventModel,
-    WithdrawalEventModel,
     BorrowingEventModel,
-    RepaymentEventModel,
-    DepositEventModel,
     CollateralEnabledDisabledEventModel,
+    DepositEventModel,
+    LiquidationEventModel,
+    RepaymentEventModel,
+    WithdrawalEventModel,
 )

apps/data_handler/models/base.py renamed to apps/data_handler/db/models/vesu.py

@@ -1,8 +1,9 @@
-from sqlalchemy import Column, BigInteger, String
+from sqlalchemy import BigInteger, Column, String
 from sqlalchemy.ext.declarative import declarative_base
 
 Base = declarative_base()
 
+
 class VesuPosition(Base):
     __tablename__ = "vesu_positions"
 

apps/data_handler/handlers/loan_states/vesu/events.py

@@ -5,13 +5,14 @@
 """
 
 from decimal import Decimal
-from sqlalchemy import select
 
-from shared.starknet_client import StarknetClient
-from shared.db import session
-from apps.data_handler.models import VesuPosition
+from sqlalchemy import select
 from starknet_py.hash.selector import get_selector_from_name
 
+from apps.data_handler.db.crud import DBConnector
+from apps.data_handler.db.models import VesuPosition
+from apps.shared.starknet_client import StarknetClient
+
 
 class VesuLoanEntity:
     """
@@ -26,8 +27,8 @@ class VesuLoanEntity:
     def __init__(self):
         """Initialize Starknet client and storage."""
         self.client = StarknetClient()
-        self.session = session
-        # self.mock_db = {}
+        self.db_connector = DBConnector()
+        self.session = self.db_connector.Session()
         self._cache = {}
         self.last_processed_block = 654244  # First VESU event block