aiteamutils 0.2.0__py3-none-any.whl

aiteamutils/cache.py

@@ -0,0 +1,48 @@
+from typing import Any, Optional
+from redis.asyncio import Redis
+from app.config import settings
+
+class Cache:
+    _instance = None
+    _redis = None
+
+    @classmethod
+    async def get_instance(cls):
+        """캐시 인스턴스를 가져옵니다.
+        
+        Returns:
+            캐시 인스턴스
+        """
+        if not cls._instance:
+            cls._instance = cls()
+            cls._redis = Redis.from_url(settings.REDIS_URL, encoding="utf-8", decode_responses=True)
+        return cls._instance
+
+    async def get(self, key: str) -> Optional[str]:
+        """키에 해당하는 값을 가져옵니다.
+        
+        Args:
+            key: 캐시 키
+            
+        Returns:
+            캐시 값, 없으면 None
+        """
+        return await self._redis.get(key)
+
+    async def set(self, key: str, value: Any, expire: int = 3600):
+        """키에 값을 설정하고 만료 시간을 설정합니다.
+        
+        Args:
+            key: 캐시 키
+            value: 캐시 값
+            expire: 만료 시간 (기본값: 1시간)
+        """
+        await self._redis.set(key, value, ex=expire)
+
+    async def delete(self, key: str):
+        """키에 해당하는 값을 삭제합니다.
+        
+        Args:
+            key: 캐시 키
+        """
+        await self._redis.delete(key) 

aiteamutils/config.py

@@ -0,0 +1,26 @@
+"""설정 모듈."""
+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()