aiteamutils 0.2.0__tar.gz

aiteamutils-0.2.0/.gitignore

@@ -0,0 +1,27 @@
+# Dependencies
+node_modules/
+__pycache__/
+*.pyc
+.env
+
+# Build
+dist/
+build/
+.svelte-kit/
+
+# IDE
+.vscode/
+.idea/
+
+# Client specific
+clients/
+!clients/.gitkeep
+!frontend/src/clients
+
+# Logs
+*.log
+npm-debug.log*
+
+# System
+.DS_Store
+Thumbs.db 

aiteamutils-0.2.0/PKG-INFO

@@ -0,0 +1,72 @@
+Metadata-Version: 2.4
+Name: aiteamutils
+Version: 0.2.0
+Summary: AI Team Utilities
+Project-URL: Homepage, https://github.com/yourusername/aiteamutils
+Project-URL: Issues, https://github.com/yourusername/aiteamutils/issues
+Author: AI Team
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Requires-Dist: fastapi
+Requires-Dist: python-jose
+Requires-Dist: sqlalchemy
+Description-Content-Type: text/markdown
+
+# AI Team Core Utils
+
+AI Team Platform의 공통 유틸리티 패키지입니다.
+
+## 설치 방법
+
+```bash
+pip install ai-team-core-utils
+```
+
+## 사용 예시
+
+```python
+from ai_team_core_utils.database import DatabaseManager
+from ai_team_core_utils.base_model import Base
+
+# DB 매니저 초기화
+db = DatabaseManager("postgresql+asyncpg://user:pass@localhost/db")
+
+# DB 세션 사용
+async with db.get_session() as session:
+    # DB 작업 수행
+    pass
+
+# 예외 처리
+from ai_team_core_utils.exceptions import CustomException, ErrorCode
+
+try:
+    # 작업 수행
+    pass
+except CustomException as e:
+    # 에러 처리
+    print(e.to_dict())
+```
+
+## 주요 기능
+
+- 데이터베이스 유틸리티
+  - 세션 관리
+  - 트랜잭션 관리
+  - 기본 CRUD 작업
+
+- 인증/인가 유틸리티
+  - JWT 토큰 관리
+  - 비밀번호 해싱
+  - Rate Limiting
+
+- 예외 처리
+  - 표준화된 에러 코드
+  - 에러 체인 추적
+  - 로깅 통합
+
+- 공통 모델
+  - 기본 모델 클래스
+  - 타입 검증
+  - 유효성 검사 

aiteamutils-0.2.0/README.md

@@ -0,0 +1,56 @@
+# AI Team Core Utils
+
+AI Team Platform의 공통 유틸리티 패키지입니다.
+
+## 설치 방법
+
+```bash
+pip install ai-team-core-utils
+```
+
+## 사용 예시
+
+```python
+from ai_team_core_utils.database import DatabaseManager
+from ai_team_core_utils.base_model import Base
+
+# DB 매니저 초기화
+db = DatabaseManager("postgresql+asyncpg://user:pass@localhost/db")
+
+# DB 세션 사용
+async with db.get_session() as session:
+    # DB 작업 수행
+    pass
+
+# 예외 처리
+from ai_team_core_utils.exceptions import CustomException, ErrorCode
+
+try:
+    # 작업 수행
+    pass
+except CustomException as e:
+    # 에러 처리
+    print(e.to_dict())
+```
+
+## 주요 기능
+
+- 데이터베이스 유틸리티
+  - 세션 관리
+  - 트랜잭션 관리
+  - 기본 CRUD 작업
+
+- 인증/인가 유틸리티
+  - JWT 토큰 관리
+  - 비밀번호 해싱
+  - Rate Limiting
+
+- 예외 처리
+  - 표준화된 에러 코드
+  - 에러 체인 추적
+  - 로깅 통합
+
+- 공통 모델
+  - 기본 모델 클래스
+  - 타입 검증
+  - 유효성 검사 

aiteamutils-0.2.0/aiteamutils/__init__.py

@@ -0,0 +1,60 @@
+from .base_model import Base
+from .database import DatabaseManager
+from .exceptions import (
+    CustomException,
+    ErrorCode,
+    custom_exception_handler,
+    request_validation_exception_handler,
+    sqlalchemy_exception_handler,
+    generic_exception_handler
+)
+from .security import (
+    verify_password,
+    hash_password,
+    create_jwt_token,
+    verify_jwt_token,
+    rate_limit,
+    RateLimitExceeded
+)
+from .base_service import BaseService
+from .base_repository import BaseRepository
+from .validators import validate_with
+from .enums import ActivityType
+from .cache import CacheManager
+
+__version__ = "0.1.0"
+
+__all__ = [
+    # Base Models
+    "Base",
+    "BaseService",
+    "BaseRepository",
+    
+    # Database
+    "DatabaseManager",
+    
+    # Exceptions
+    "CustomException",
+    "ErrorCode",
+    "custom_exception_handler",
+    "request_validation_exception_handler",
+    "sqlalchemy_exception_handler",
+    "generic_exception_handler",
+    
+    # Security
+    "verify_password",
+    "hash_password",
+    "create_jwt_token",
+    "verify_jwt_token",
+    "rate_limit",
+    "RateLimitExceeded",
+    
+    # Validators
+    "validate_with",
+    
+    # Enums
+    "ActivityType",
+    
+    # Cache
+    "CacheManager"
+] 

aiteamutils-0.2.0/aiteamutils/base_model.py

@@ -0,0 +1,81 @@
+from datetime import datetime, timezone
+from typing import Any, Dict, TypeVar, Generic, Optional
+from ulid import ULID
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+from sqlalchemy import Column, String, PrimaryKeyConstraint, UniqueConstraint
+from sqlalchemy.dialects.postgresql import TIMESTAMP
+from pydantic import BaseModel, ConfigDict
+from pydantic import Field
+
+class Base(DeclarativeBase):
+    """SQLAlchemy 기본 모델"""
+    pass
+
+class BaseColumn(Base):
+    """공통 설정 및 메서드를 제공하는 BaseColumn"""
+    __abstract__ = True
+
+    ulid: Mapped[str] = mapped_column(
+        String,
+        primary_key=True,
+        unique=True,
+        default=lambda: str(ULID()),
+        doc="ULID",
+        nullable=False
+    )
+    created_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
+    updated_at: Mapped[datetime] = mapped_column(
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow
+    )
+    is_deleted: Mapped[bool] = mapped_column(
+        default=False,
+        index=True
+    )
+
+    def to_dict(self) -> Dict[str, Any]:
+        """모델을 딕셔너리로 변환합니다.
+        
+        Returns:
+            Dict[str, Any]: 모델의 속성을 포함하는 딕셔너리
+        """
+        result = {}
+        
+        # 테이블 컬럼 처리
+        for column in self.__table__.columns:
+            value = getattr(self, column.name)
+            if isinstance(value, datetime):
+                value = value.isoformat()
+            result[column.name] = value
+        
+        # Relationship 처리 (이미 로드된 관계만 처리)
+        for relationship in self.__mapper__.relationships:
+            if relationship.key == "organizations":  # 순환 참조 방지
+                continue
+            try:
+                value = getattr(self, relationship.key)
+                if value is not None:
+                    if isinstance(value, list):
+                        result[relationship.key] = [item.to_dict() for item in value]
+                    else:
+                        result[relationship.key] = value.to_dict()
+                else:
+                    result[relationship.key] = None
+            except Exception:
+                result[relationship.key] = None
+                
+        return result
+
+class BaseSchema(BaseModel):
+    """공통 설정 및 메서드를 제공하는 BaseSchema"""
+    model_config = ConfigDict(
+        str_strip_whitespace=True,
+        extra="allow",
+        from_attributes=True,
+        populate_by_name=True,
+        use_enum_values=True
+    )
+
+    def to_dict(self) -> Dict[str, Any]:
+        """모델을 딕셔너리로 변환"""
+        return self.model_dump()