refactor: core dependencies parse integer ids

Author: fiqrikm18

src/core/database/postgres/session.py

@@ -1,5 +1,4 @@
 from collections.abc import AsyncGenerator
-from uuid import UUID
 
 from fastapi import Depends
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
@@ -33,6 +32,6 @@ async def get_db() -> AsyncGenerator[AsyncSession]:
 
 def get_unit_of_work(
     db: AsyncSession = Depends(get_db),
-    tenant_id: UUID | None = Depends(get_optional_tenant_id),
+    tenant_id: int | None = Depends(get_optional_tenant_id),
 ) -> UnitOfWork:
     return SQLAlchemyUnitOfWork(db, tenant_id)

src/core/database/unit_of_work.py

@@ -1,6 +1,5 @@
 from types import TracebackType
 from typing import TYPE_CHECKING, Self
-from uuid import UUID
 
 from sqlalchemy.ext.asyncio import AsyncSession
 
@@ -12,7 +11,7 @@
 
 
 class SQLAlchemyUnitOfWork(UnitOfWork):
-    def __init__(self, session: AsyncSession, tenant_id: UUID | None = None):
+    def __init__(self, session: AsyncSession, tenant_id: int | None = None):
         self._session = session
         self._tenant_id = tenant_id
         self._committed = False

src/core/dependency/auth.py

@@ -1,5 +1,3 @@
-from uuid import UUID
-
 from fastapi import Depends, HTTPException, Request, status
 from fastapi.security import OAuth2PasswordBearer
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -21,7 +19,7 @@ async def get_current_user(
     request: Request,
     token: str = Depends(oauth2_scheme),
     db: AsyncSession = Depends(get_db),
-    tenant_id: UUID | None = Depends(get_optional_tenant_id),
+    tenant_id: int | None = Depends(get_optional_tenant_id),
 ) -> dict:
     """
     1. 'token' is extracted by oauth2_scheme (for Swagger docs).
@@ -37,7 +35,7 @@ async def get_current_user(
         )
 
     repo = SQLAlchemyUserRepository(db, tenant_id)
-    user = await repo.get_by_id(UUID(user_id))
+    user = await repo.get_by_id(int(user_id))
 
     if not user:
         raise HTTPException(

src/core/dependency/tenant.py

@@ -1,11 +1,9 @@
-from uuid import UUID
-
 from fastapi import HTTPException, Request
 from starlette.status import HTTP_400_BAD_REQUEST
 
 
-def get_current_tenant_id(request: Request) -> UUID:
-    tenant_id: UUID | None = getattr(request.state, "tenant_id", None)
+def get_current_tenant_id(request: Request) -> int:
+    tenant_id: int | None = getattr(request.state, "tenant_id", None)
     if tenant_id is None:
         raise HTTPException(
             status_code=HTTP_400_BAD_REQUEST,
@@ -14,5 +12,5 @@ def get_current_tenant_id(request: Request) -> UUID:
     return tenant_id
 
 
-def get_optional_tenant_id(request: Request) -> UUID | None:
+def get_optional_tenant_id(request: Request) -> int | None:
     return getattr(request.state, "tenant_id", None)

src/core/middleware/tenant.py

@@ -1,16 +1,14 @@
-from uuid import UUID
-
 from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
 from starlette.requests import Request
 from starlette.responses import JSONResponse, Response
 
 from src.core.config.setting import get_settings
 
 settings = get_settings()
-_default_tenant_id: UUID | None = None
+_default_tenant_id: int | None = None
 
 
-def set_default_tenant_id(tenant_id: UUID) -> None:
+def set_default_tenant_id(tenant_id: int) -> None:
     global _default_tenant_id
     _default_tenant_id = tenant_id
 
@@ -71,7 +69,7 @@ async def dispatch(
                 )
                 tid = payload.get("tenant_id")
                 if tid:
-                    request.state.tenant_id = UUID(tid) if isinstance(tid, str) else tid
+                    request.state.tenant_id = int(tid) if isinstance(tid, str) else tid
                     return await call_next(request)
             except Exception:
                 pass

src/core/security/infrastructure/repositories/login_attempt_repository.py

@@ -2,15 +2,13 @@
 
 from sqlalchemy import delete, func, select
 
-from uuid import UUID
-
 from src.core.security.infrastructure.models.login_attempt_model import (
     LoginAttemptModel,
 )
 
 
 class SQLAlchemyLoginAttemptRepository:
-    def __init__(self, db, tenant_id: UUID | None = None):
+    def __init__(self, db, tenant_id: int | None = None):
         self._db = db
         self._tenant_id = tenant_id