aiteamutils 0.2.19__tar.gz → 0.2.21__tar.gz

{aiteamutils-0.2.19 → aiteamutils-0.2.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.19
+Version: 0.2.21
 Summary: AI Team Utilities
 Project-URL: Homepage, https://github.com/yourusername/aiteamutils
 Project-URL: Issues, https://github.com/yourusername/aiteamutils/issues

{aiteamutils-0.2.19 → aiteamutils-0.2.21}/aiteamutils/database.py

@@ -85,7 +85,7 @@ class DatabaseService:
                 pool_pre_ping=True,
                 poolclass=QueuePool,
             )
-            self.async_session = sessionmaker(
+            self.session_factory = sessionmaker(
                 bind=self.engine,
                 class_=AsyncSession,
                 expire_on_commit=False
@@ -93,7 +93,11 @@ class DatabaseService:
             self.db = None
         elif session:
             self.engine = session.bind
-            self.async_session = None
+            self.session_factory = sessionmaker(
+                bind=self.engine,
+                class_=AsyncSession,
+                expire_on_commit=False
+            )
             self.db = session
         else:
             raise ValueError("Either db_url or session must be provided")
@@ -101,10 +105,10 @@ class DatabaseService:
     @asynccontextmanager
     async def get_session(self) -> AsyncGenerator[AsyncSession, None]:
         """데이터베이스 세션을 생성하고 반환하는 비동기 컨텍스트 매니저."""
-        if self.async_session is None:
+        if self.session_factory is None:
             raise RuntimeError("Session factory not initialized")
         
-        async with self.async_session() as session:
+        async with self.session_factory() as session:
             try:
                 yield session
             finally:
@@ -788,7 +792,10 @@ class DatabaseService:
             CustomException: 데이터베이스 작업 중 오류 발생 시
         """
         try:
-            async with self.db as session:
+            if self.db is not None:
+                return await self.db.execute(query)
+            
+            async with self.get_session() as session:
                 result = await session.execute(query)
                 return result
         except Exception as e:

{aiteamutils-0.2.19 → aiteamutils-0.2.21}/aiteamutils/dependencies.py

@@ -2,6 +2,7 @@ from typing import Type, Dict, Tuple, Any, Callable
 from fastapi import Depends, status
 from fastapi.security import OAuth2PasswordBearer
 from jose import JWTError, jwt
+from sqlalchemy.ext.asyncio import AsyncSession
 
 from .database import DatabaseService, get_database_service
 from .exceptions import CustomException, ErrorCode
@@ -59,7 +60,8 @@ def get_service(name: str):
     Returns:
         Callable: 서비스 인스턴스를 반환하는 의존성 함수
     """
-    def _get_service(db_service: DatabaseService = Depends(get_database_service)):
+    def _get_service(db: AsyncSession = Depends(get_db)):
+        db_service = DatabaseService(session=db)
         repository_class, service_class = service_registry.get(name)
         repository = repository_class(db_service)
         return service_class(repository, db_service)

aiteamutils-0.2.21/aiteamutils/version.py

@@ -0,0 +1,2 @@
+"""버전 정보"""
+__version__ = "0.2.21" 

aiteamutils-0.2.19/aiteamutils/version.py

@@ -1,2 +0,0 @@
-"""버전 정보"""
-__version__ = "0.2.19" 

aiteamutils-0.2.19/app/auth/service.py

@@ -1,4 +0,0 @@
-from aiteamutils.config import get_settings
-
-# settings를 사용하는 곳에서는 get_settings()를 호출하여 사용
-settings = get_settings()