Implement updates for better performance

Author: abrar2030

code/backend/config/database.py

@@ -177,7 +177,9 @@ async def get_async_read_session() -> AsyncGenerator[AsyncSession, None]:
     Dependency function to get async read-only database session.
     Uses read replica if available, otherwise falls back to primary.
     """
-    session_maker = AsyncReadSessionLocal if AsyncReadSessionLocal else AsyncSessionLocal
+    session_maker = (
+        AsyncReadSessionLocal if AsyncReadSessionLocal else AsyncSessionLocal
+    )
     async with session_maker() as session:
         try:
             yield session

code/backend/models/compliance.py

@@ -92,9 +92,7 @@ class AuditLog(BaseModel, TimestampMixin):
     event_description = Column(Text, nullable=True)
 
     # User and Session
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     session_id = Column(String(255), nullable=True, index=True)
 
     # Request Details
@@ -168,9 +166,7 @@ class ComplianceCheck(BaseModel, TimestampMixin, AuditMixin):
     check_description = Column(Text, nullable=True)
 
     # Subject of Check
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     transaction_id = Column(
         GUID(), ForeignKey("transactions.id"), nullable=True, index=True
     )
@@ -328,9 +324,7 @@ class SuspiciousActivityReport(BaseModel, TimestampMixin, AuditMixin):
     sar_number = Column(String(50), unique=True, nullable=True, index=True)
 
     # Subject Information
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     transaction_ids = Column(JSON, nullable=True)  # List of related transaction IDs
 
     # Suspicious Activity Details

code/backend/models/portfolio.py

@@ -74,9 +74,7 @@ class Portfolio(BaseModel, TimestampMixin, AuditMixin):
 
     __tablename__ = "portfolios"
 
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=False, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=False, index=True)
 
     # Portfolio Details
     name = Column(String(100), nullable=False)

code/backend/models/risk.py

@@ -73,9 +73,7 @@ class RiskAssessment(BaseModel, TimestampMixin, AuditMixin):
     __tablename__ = "risk_assessments"
 
     # Assessment Subject
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     portfolio_id = Column(
         GUID(), ForeignKey("portfolios.id"), nullable=True, index=True
     )
@@ -157,9 +155,7 @@ class RiskMetrics(BaseModel, TimestampMixin):
     __tablename__ = "risk_metrics"
 
     # Metric Subject
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     portfolio_id = Column(
         GUID(), ForeignKey("portfolios.id"), nullable=True, index=True
     )
@@ -304,9 +300,7 @@ class RiskAlert(BaseModel, TimestampMixin, AuditMixin):
     )
 
     # Alert Subject
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     portfolio_id = Column(
         GUID(), ForeignKey("portfolios.id"), nullable=True, index=True
     )
@@ -399,9 +393,7 @@ class RiskLimit(BaseModel, TimestampMixin, AuditMixin):
     __tablename__ = "risk_limits"
 
     # Limit Subject
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=True, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=True, index=True)
     portfolio_id = Column(
         GUID(), ForeignKey("portfolios.id"), nullable=True, index=True
     )

code/backend/models/transaction.py

@@ -75,9 +75,7 @@ class Transaction(BaseModel, TimestampMixin, AuditMixin):
     __tablename__ = "transactions"
 
     # User Association
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=False, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=False, index=True)
 
     # Blockchain Information
     tx_hash = Column(
@@ -226,9 +224,7 @@ class TransactionAlert(BaseModel, TimestampMixin, AuditMixin):
     transaction_id = Column(
         GUID(), ForeignKey("transactions.id"), nullable=False, index=True
     )
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=False, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=False, index=True)
 
     # Alert Details
     alert_type = Column(String(50), nullable=False, index=True)
@@ -273,9 +269,7 @@ class TransactionPattern(BaseModel, TimestampMixin):
 
     __tablename__ = "transaction_patterns"
 
-    user_id = Column(
-        GUID(), ForeignKey("users.id"), nullable=False, index=True
-    )
+    user_id = Column(GUID(), ForeignKey("users.id"), nullable=False, index=True)
 
     # Pattern Details
     pattern_type = Column(String(50), nullable=False, index=True)

code/backend/models/user.py

@@ -21,7 +21,14 @@
 )
 from sqlalchemy.orm import relationship
 
-from .base import GUID, AuditMixin, BaseModel, EncryptedMixin, SoftDeleteMixin, TimestampMixin
+from .base import (
+    GUID,
+    AuditMixin,
+    BaseModel,
+    EncryptedMixin,
+    SoftDeleteMixin,
+    TimestampMixin,
+)
 
 
 class UserStatus(enum.Enum):

code/backend/tests/conftest.py

@@ -33,6 +33,7 @@
 os.environ.setdefault("REDIS_URL", "redis://localhost:6379/0")
 os.environ.setdefault("FIELD_ENCRYPTION_ENABLED", "false")
 
+
 # ── 2. Cache mock ─────────────────────────────────────────────────────────────
 class _MockCacheManager:
     def __init__(self) -> None:
@@ -71,18 +72,19 @@ async def _noop():
 _db_module.init_database = _noop
 _db_module.close_database = _noop
 
-from app.main import app  # noqa: E402  (triggers app module import)
 import app.main as _app_main  # noqa: E402
+from app.main import app  # noqa: E402  (triggers app module import)
 
 _app_main.init_database = _noop
 _app_main.close_database = _noop
 
-# ── 4. Public test fixtures ───────────────────────────────────────────────────
-import pytest  # noqa: E402
 from typing import Any  # noqa: E402
 from unittest.mock import AsyncMock, MagicMock  # noqa: E402
 from uuid import uuid4  # noqa: E402
 
+# ── 4. Public test fixtures ───────────────────────────────────────────────────
+import pytest  # noqa: E402
+
 
 @pytest.fixture
 def mock_db() -> Any:
@@ -104,10 +106,10 @@ def mock_db() -> Any:
 @pytest.fixture
 def mock_user():
     """In-memory User object with sensible defaults."""
-    from datetime import datetime, timezone
-    from models.user import User, UserStatus
 
+    from models.user import User, UserStatus
     from passlib.context import CryptContext
+
     pwd_ctx = CryptContext(schemes=["bcrypt"], deprecated="auto")
 
     u = User.__new__(User)
@@ -134,8 +136,11 @@ def mock_user():
 def auth_token(mock_user) -> str:
     """Real JWT access-token for the mock_user (no DB required)."""
     from services.auth.jwt_service import JWTService
+
     jwt = JWTService()
-    return jwt.create_access_token(data={"sub": str(mock_user.id), "email": mock_user.email})
+    return jwt.create_access_token(
+        data={"sub": str(mock_user.id), "email": mock_user.email}
+    )
 
 
 @pytest.fixture
@@ -145,6 +150,7 @@ def auth_headers(auth_token) -> dict:
 
 # ── Misc data fixtures ────────────────────────────────────────────────────────
 
+
 @pytest.fixture
 def sample_transaction_data() -> dict:
     return {

code/backend/tests/integration/test_auth_endpoints.py

@@ -5,29 +5,34 @@
 No aiosqlite, no async DB — all database interactions are AsyncMock.
 """
 
-from unittest.mock import AsyncMock, MagicMock, patch
 from datetime import datetime, timezone
+from unittest.mock import AsyncMock, MagicMock, patch
 from uuid import uuid4
 
 import pytest
-from fastapi import status
-from fastapi.testclient import TestClient
-
-from app.main import app
 from app.api.dependencies import get_current_user
+from app.main import app
 from config.database import get_async_session
-
+from fastapi import status
+from fastapi.testclient import TestClient
 
 # ── helpers ───────────────────────────────────────────────────────────────────
 
+
 def _make_user(email: str = "test@example.com", password_plain: str = "testpassword"):
     """Build an in-memory User object."""
     from datetime import datetime, timezone
+
     from models.user import User, UserStatus
     from passlib.context import CryptContext
 
     pwd = CryptContext(schemes=["bcrypt"], deprecated="auto")
-    u = User(email=email, hashed_password=pwd.hash(password_plain), status=UserStatus.ACTIVE, email_verified=True)
+    u = User(
+        email=email,
+        hashed_password=pwd.hash(password_plain),
+        status=UserStatus.ACTIVE,
+        email_verified=True,
+    )
     u.id = uuid4()
     u.is_deleted = False
     u.mfa_enabled = False
@@ -61,9 +66,12 @@ def _mock_session():
 
 def _tokens(user):
     from services.auth.jwt_service import JWTService
+
     j = JWTService()
     return {
-        "access_token": j.create_access_token({"sub": str(user.id), "email": user.email}),
+        "access_token": j.create_access_token(
+            {"sub": str(user.id), "email": user.email}
+        ),
         "refresh_token": j.create_refresh_token({"sub": str(user.id)}),
         "token_type": "bearer",
         "expires_in": 1800,
@@ -162,7 +170,9 @@ def test_login_success(self, client):
         async def _mock_auth(*a, **kw):
             return user, tokens
 
-        with patch("app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth):
+        with patch(
+            "app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth
+        ):
             response = c.post(
                 "/api/v1/auth/login",
                 json={"email": user.email, "password": "testpassword"},
@@ -183,7 +193,9 @@ def test_login_invalid_credentials(self, client):
         async def _mock_auth(*a, **kw):
             raise HTTPException(status_code=401, detail="Invalid credentials")
 
-        with patch("app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth):
+        with patch(
+            "app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth
+        ):
             response = c.post(
                 "/api/v1/auth/login",
                 json={"email": "test@example.com", "password": "wrongpassword"},
@@ -200,7 +212,9 @@ def test_login_nonexistent_user(self, client):
         async def _mock_auth(*a, **kw):
             raise HTTPException(status_code=401, detail="Invalid credentials")
 
-        with patch("app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth):
+        with patch(
+            "app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth
+        ):
             response = c.post(
                 "/api/v1/auth/login",
                 json={"email": "nobody@example.com", "password": "testpassword"},
@@ -223,7 +237,9 @@ def test_get_current_user(self, authed_client):
     def test_get_current_user_invalid_token(self, client):
         """Invalid token → 401."""
         c, _ = client
-        response = c.get("/api/v1/auth/me", headers={"Authorization": "Bearer bad_token"})
+        response = c.get(
+            "/api/v1/auth/me", headers={"Authorization": "Bearer bad_token"}
+        )
         assert response.status_code == status.HTTP_401_UNAUTHORIZED
 
     def test_get_current_user_no_token(self, client):
@@ -291,7 +307,10 @@ def test_logout_success(self, authed_client):
         """Valid token → successful logout."""
         c, user, _ = authed_client
         from services.auth.jwt_service import JWTService
-        token = JWTService().create_access_token({"sub": str(user.id), "email": user.email})
+
+        token = JWTService().create_access_token(
+            {"sub": str(user.id), "email": user.email}
+        )
 
         async def _mock_logout(*a, **kw):
             pass
@@ -327,7 +346,9 @@ def test_oauth2_login_form(self, client):
         async def _mock_auth(*a, **kw):
             return user, tokens
 
-        with patch("app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth):
+        with patch(
+            "app.api.v1.endpoints.auth.auth_service.authenticate_user", _mock_auth
+        ):
             response = c.post(
                 "/api/v1/auth/login/form",
                 data={"username": user.email, "password": "testpassword"},
@@ -341,8 +362,11 @@ async def _mock_auth(*a, **kw):
 
 # ── tiny async helper ─────────────────────────────────────────────────────────
 
+
 def _async_return(value):
     """Return an async function that returns value."""
+
     async def _inner(*args, **kwargs):
         return value
+
     return _inner