aiteamutils 0.2.14__tar.gz → 0.2.16__tar.gz

{aiteamutils-0.2.14 → aiteamutils-0.2.16}/PKG-INFO

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

{aiteamutils-0.2.14 → aiteamutils-0.2.16}/aiteamutils/__init__.py

@@ -1,5 +1,10 @@
 from .base_model import Base
-from .database import DatabaseService
+from .database import (
+    DatabaseService,
+    get_db,
+    get_database_service,
+    get_database_session
+)
 from .exceptions import (
     CustomException,
     ErrorCode,
@@ -30,6 +35,9 @@ __all__ = [
     
     # Database
     "DatabaseService",
+    "get_db",
+    "get_database_service",
+    "get_database_session",
     
     # Exceptions
     "CustomException",

aiteamutils-0.2.16/aiteamutils/config.py

@@ -0,0 +1,59 @@
+"""설정 모듈."""
+from typing import Union
+from .database import init_database_service
+
+class Settings:
+    """기본 설정 클래스"""
+    def __init__(self, jwt_secret: str, jwt_algorithm: str = "HS256"):
+        self.JWT_SECRET = jwt_secret
+        self.JWT_ALGORITHM = jwt_algorithm
+
+_settings: Union[Settings, 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
+    _settings = Settings(jwt_secret, jwt_algorithm)
+    
+    if db_url:
+        init_database_service(
+            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:
+    """현재 설정을 반환하는 함수
+    
+    Returns:
+        Settings: 설정 객체
+        
+    Raises:
+        RuntimeError: 설정이 초기화되지 않은 경우
+    """
+    if _settings is None:
+        raise RuntimeError("Settings not initialized. Call init_settings first.")
+    return _settings 

{aiteamutils-0.2.14 → aiteamutils-0.2.16}/aiteamutils/database.py

@@ -17,7 +17,7 @@ from .enums import ActivityType
 T = TypeVar("T", bound=BaseColumn)
 
 # 전역 데이터베이스 서비스 인스턴스
-_database_service: 'DatabaseService' | None = None
+_database_service: Union['DatabaseService', None] = None
 
 def get_database_service() -> 'DatabaseService':
     """DatabaseService 인스턴스를 반환하는 함수
@@ -880,4 +880,36 @@ class DatabaseService:
                 detail=str(e),
                 source_function="DatabaseService.refresh",
                 original_error=e
-            )
+            )
+
+def init_database_service(
+    db_url: str,
+    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 (str): 데이터베이스 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): 커넥션 재활용 시간
+        
+    Returns:
+        DatabaseService: 초기화된 데이터베이스 서비스 인스턴스
+    """
+    global _database_service
+    _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
+    )
+    return _database_service

{aiteamutils-0.2.14 → aiteamutils-0.2.16}/aiteamutils/dependencies.py

@@ -5,63 +5,7 @@ from jose import JWTError, jwt
 
 from .database import DatabaseService, get_database_service
 from .exceptions import CustomException, ErrorCode
-
-class Settings:
-    """기본 설정 클래스"""
-    def __init__(self, jwt_secret: str, jwt_algorithm: str = "HS256"):
-        self.JWT_SECRET = jwt_secret
-        self.JWT_ALGORITHM = jwt_algorithm
-
-_settings: Settings | 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
-    _settings = Settings(jwt_secret, jwt_algorithm)
-    
-    if db_url:
-        from .database import _database_service
-        _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:
-    """현재 설정을 반환하는 함수
-    
-    Returns:
-        Settings: 설정 객체
-        
-    Raises:
-        RuntimeError: 설정이 초기화되지 않은 경우
-    """
-    if _settings is None:
-        raise RuntimeError("Settings not initialized. Call init_settings first.")
-    return _settings
+from .config import get_settings
 
 class ServiceRegistry:
     """서비스 레지스트리를 관리하는 클래스"""

aiteamutils-0.2.16/aiteamutils/version.py

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

aiteamutils-0.2.14/aiteamutils/config.py

@@ -1,26 +0,0 @@
-"""설정 모듈."""
-from typing import Optional
-from pydantic_settings import BaseSettings
-
-class Settings(BaseSettings):
-    """애플리케이션 설정."""
-    
-    # 데이터베이스 설정
-    DB_ECHO: bool = False
-    DB_POOL_SIZE: int = 5
-    DB_MAX_OVERFLOW: int = 10
-    DB_POOL_TIMEOUT: int = 30
-    DB_POOL_RECYCLE: int = 1800
-    
-    # JWT 설정
-    JWT_SECRET: str = "your-secret-key"
-    JWT_ALGORITHM: str = "HS256"
-    ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
-    TOKEN_ISSUER: str = "ai-team-platform"
-    TOKEN_AUDIENCE: str = "ai-team-users"
-    
-    class Config:
-        env_file = ".env"
-        case_sensitive = True
-
-settings = Settings() 

aiteamutils-0.2.14/aiteamutils/version.py

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