aiteamutils 0.2.8__tar.gz → 0.2.10__tar.gz

{aiteamutils-0.2.8 → aiteamutils-0.2.10}/PKG-INFO

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

{aiteamutils-0.2.8 → aiteamutils-0.2.10}/aiteamutils/database.py

@@ -18,32 +18,57 @@ from .config import settings
 T = TypeVar("T", bound=BaseColumn)
 
 class DatabaseService:
-    def __init__(self, db_url: str):
+    def __init__(
+        self,
+        db_url: str = None,
+        session: AsyncSession = None,
+        db_echo: bool = False,
+        db_pool_size: int = 5,
+        db_max_overflow: int = 10,
+        db_pool_timeout: int = 30,
+        db_pool_recycle: int = 1800
+    ):
         """DatabaseService 초기화.
         
         Args:
-            db_url: 데이터베이스 URL
+            db_url (str, optional): 데이터베이스 URL
+            session (AsyncSession, optional): 기존 세션
+            db_echo (bool, optional): SQL 로깅 여부
+            db_pool_size (int, optional): DB 커넥션 풀 크기
+            db_max_overflow (int, optional): 최대 초과 커넥션 수
+            db_pool_timeout (int, optional): 커넥션 풀 타임아웃
+            db_pool_recycle (int, optional): 커넥션 재활용 시간
         """
-        self.engine = create_async_engine(
-            db_url,
-            echo=settings.DB_ECHO,
-            pool_size=settings.DB_POOL_SIZE,
-            max_overflow=settings.DB_MAX_OVERFLOW,
-            pool_timeout=settings.DB_POOL_TIMEOUT,
-            pool_recycle=settings.DB_POOL_RECYCLE,
-            pool_pre_ping=True,
-            poolclass=QueuePool,
-        )
-        
-        self.async_session = sessionmaker(
-            bind=self.engine,
-            class_=AsyncSession,
-            expire_on_commit=False
-        )
-        
+        if db_url:
+            self.engine = create_async_engine(
+                db_url,
+                echo=db_echo,
+                pool_size=db_pool_size,
+                max_overflow=db_max_overflow,
+                pool_timeout=db_pool_timeout,
+                pool_recycle=db_pool_recycle,
+                pool_pre_ping=True,
+                poolclass=QueuePool,
+            )
+            self.async_session = sessionmaker(
+                bind=self.engine,
+                class_=AsyncSession,
+                expire_on_commit=False
+            )
+            self.db = None
+        elif session:
+            self.engine = session.bind
+            self.async_session = None
+            self.db = session
+        else:
+            raise ValueError("Either db_url or session must be provided")
+
     @asynccontextmanager
-    async def get_db(self) -> AsyncGenerator[AsyncSession, None]:
-        """데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터."""
+    async def get_session(self) -> AsyncGenerator[AsyncSession, None]:
+        """데이터베이스 세션을 생성하고 반환하는 비동기 컨텍스트 매니저."""
+        if self.async_session is None:
+            raise RuntimeError("Session factory not initialized")
+        
         async with self.async_session() as session:
             try:
                 yield session

{aiteamutils-0.2.8 → aiteamutils-0.2.10}/aiteamutils/dependencies.py

@@ -14,16 +14,42 @@ class Settings:
         self.JWT_ALGORITHM = jwt_algorithm
 
 _settings: Settings | None = None
-
-def init_settings(jwt_secret: str, jwt_algorithm: str = "HS256"):
+_database_service: DatabaseService | None = None
+
+def init_settings(
+    jwt_secret: str,
+    jwt_algorithm: str = "HS256",
+    db_url: str = None,
+    db_echo: bool = False,
+    db_pool_size: int = 5,
+    db_max_overflow: int = 10,
+    db_pool_timeout: int = 30,
+    db_pool_recycle: int = 1800
+):
     """설정 초기화 함수
     
     Args:
         jwt_secret (str): JWT 시크릿 키
         jwt_algorithm (str, optional): JWT 알고리즘. Defaults to "HS256".
+        db_url (str, optional): 데이터베이스 URL
+        db_echo (bool, optional): SQL 로깅 여부
+        db_pool_size (int, optional): DB 커넥션 풀 크기
+        db_max_overflow (int, optional): 최대 초과 커넥션 수
+        db_pool_timeout (int, optional): 커넥션 풀 타임아웃
+        db_pool_recycle (int, optional): 커넥션 재활용 시간
     """
-    global _settings
+    global _settings, _database_service
     _settings = Settings(jwt_secret, jwt_algorithm)
+    
+    if db_url:
+        _database_service = DatabaseService(
+            db_url=db_url,
+            db_echo=db_echo,
+            db_pool_size=db_pool_size,
+            db_max_overflow=db_max_overflow,
+            db_pool_timeout=db_pool_timeout,
+            db_pool_recycle=db_pool_recycle
+        )
 
 def get_settings() -> Settings:
     """현재 설정을 반환하는 함수
@@ -38,6 +64,39 @@ def get_settings() -> Settings:
         raise RuntimeError("Settings not initialized. Call init_settings first.")
     return _settings
 
+def get_database_service() -> DatabaseService:
+    """DatabaseService 인스턴스를 반환하는 함수
+    
+    Returns:
+        DatabaseService: DatabaseService 인스턴스
+        
+    Raises:
+        RuntimeError: DatabaseService가 초기화되지 않은 경우
+    """
+    if _database_service is None:
+        raise RuntimeError("DatabaseService not initialized. Call init_settings with db_url first.")
+    return _database_service
+
+async def get_db() -> AsyncGenerator[AsyncSession, None]:
+    """데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터."""
+    db_service = get_database_service()
+    async with db_service.get_session() as session:
+        try:
+            yield session
+        finally:
+            await session.close()
+
+def get_database_session(db: AsyncSession = Depends(get_db)) -> DatabaseService:
+    """DatabaseService 의존성
+
+    Args:
+        db (AsyncSession): 데이터베이스 세션
+
+    Returns:
+        DatabaseService: DatabaseService 인스턴스
+    """
+    return DatabaseService(session=db)
+
 class ServiceRegistry:
     """서비스 레지스트리를 관리하는 클래스"""
     def __init__(self):
@@ -81,17 +140,6 @@ class ServiceRegistry:
 # ServiceRegistry 초기화
 service_registry = ServiceRegistry()
 
-async def get_database_service(db: AsyncSession) -> DatabaseService:
-    """DatabaseService 의존성
-
-    Args:
-        db (AsyncSession): 데이터베이스 세션
-
-    Returns:
-        DatabaseService: DatabaseService 인스턴스
-    """
-    return DatabaseService(db)
-
 def get_service(name: str):
     """등록된 서비스를 가져오는 의존성 함수
 

aiteamutils-0.2.10/aiteamutils/version.py

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

aiteamutils-0.2.8/aiteamutils/version.py

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