aiteamutils 0.2.21__py3-none-any.whl → 0.2.23__py3-none-any.whl

aiteamutils/dependencies.py

@@ -4,7 +4,7 @@ from fastapi.security import OAuth2PasswordBearer
 from jose import JWTError, jwt
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from .database import DatabaseService, get_database_service
+from .database import DatabaseService, get_database_service, get_db
 from .exceptions import CustomException, ErrorCode
 from .config import get_settings
 

aiteamutils/security.py

@@ -268,24 +268,24 @@ async def create_jwt_token(
     """JWT 토큰을 생성하고 로그를 기록합니다.
     
     Args:
-        user_data: 사용자 데이터 (username, ulid 등 필수)
+        user_data: 사용자 데이터 딕셔너리 (username, ulid, name, role_ulid, status, organization 정보 등)
         token_type: 토큰 타입 ("access" 또는 "refresh")
         db_service: 데이터베이스 서비스
-        log_model: 로그를 저장할 모델 클래스
+        log_model: 로그 모델
         request: FastAPI 요청 객체
         
     Returns:
         str: 생성된 JWT 토큰
         
     Raises:
-        TokenCreationError: 토큰 생성 실패 시
-        SecurityError: 기타 보안 관련 오류 발생 시
+        CustomException: 토큰 생성 실패 시
     """
     try:
         settings = get_settings()
+        
         # 필수 필드 검증
         required_fields = {"username", "ulid"}
-        missing_fields = required_fields - user_data.keys()
+        missing_fields = required_fields - set(user_data.keys())
         if missing_fields:
             raise TokenCreationError(
                 detail=f"Missing required fields: {', '.join(missing_fields)}",
@@ -297,12 +297,28 @@ async def create_jwt_token(
         if token_type == "access":
             expires_at = datetime.now(UTC) + timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES)
             token_data = {
+                # 등록 클레임
                 "iss": settings.TOKEN_ISSUER,
                 "sub": user_data["username"],
                 "aud": settings.TOKEN_AUDIENCE,
                 "exp": expires_at,
+                
+                # 공개 클레임
+                "username": user_data["username"],
+                "name": user_data.get("name"),
+                
+                # 비공개 클레임
+                "user_ulid": user_data["ulid"],
+                "role_ulid": user_data.get("role_ulid"),
+                "status": user_data.get("status"),
+                "last_login": datetime.now(UTC).isoformat(),
                 "token_type": token_type,
-                **user_data
+                
+                # 조직 관련 클레임
+                "organization_ulid": user_data.get("organization_ulid"),
+                "organization_id": user_data.get("organization_id"),
+                "organization_name": user_data.get("organization_name"),
+                "company_name": user_data.get("company_name")
             }
         else:  # refresh token
             expires_at = datetime.now(UTC) + timedelta(days=14)

aiteamutils/version.py

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

{aiteamutils-0.2.21.dist-info → aiteamutils-0.2.23.dist-info}/METADATA

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

{aiteamutils-0.2.21.dist-info → aiteamutils-0.2.23.dist-info}/RECORD

@@ -5,12 +5,12 @@ aiteamutils/base_service.py,sha256=E4dHGE0DvhmRyFplh46SwKJOSF_nUL7OAsCkf_ZJF_8,2
 aiteamutils/cache.py,sha256=tr0Yn8VPYA9QHiKCUzciVlQ2J1RAwNo2K9lGMH4rY3s,1334
 aiteamutils/config.py,sha256=7pJHBml8RRXC9FpoRBx0emofFkvtxsyk9_Rcur2F9hI,2724
 aiteamutils/database.py,sha256=raYKRZjEfefM3SnoPYozPu7OUGm62XWOUqZzN4_cayw,33479
-aiteamutils/dependencies.py,sha256=gy1G4DFb-UooFDFZOSIThEDDCUtsJn9odnm5dKV1kQM,4000
+aiteamutils/dependencies.py,sha256=e5P-DQi5OMMh7FADaAUqU0kLVBcReE7za55EDNZN950,4008
 aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
 aiteamutils/exceptions.py,sha256=YV-ISya4wQlHk4twvGo16I5r8h22-tXpn9wa-b3WwDM,15231
-aiteamutils/security.py,sha256=2k5j8iV4jkp0IGRmOtEbVyoeuoDHbguzaYju0Wm-mmI,12715
+aiteamutils/security.py,sha256=4J9Q18RvlHH3zD8gW8ymFio7k5JuIJxvS2blU19AipA,13458
 aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
-aiteamutils/version.py,sha256=Cm_lURyI4GuJ_Ec6ukumU-cRzbGIFhPP_6hKl8HcuS8,44
-aiteamutils-0.2.21.dist-info/METADATA,sha256=AyaS1T8470tkEw0iyTdXbrCGkz4w4dICfBPOwblO9Cg,1718
-aiteamutils-0.2.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.21.dist-info/RECORD,,
+aiteamutils/version.py,sha256=FoxokvvxnU_rqoyqNTry6Th-i69tCu9uLXqkhxLbDwU,44
+aiteamutils-0.2.23.dist-info/METADATA,sha256=S-LkVpOYUZgX1BssQ9c144uavb-3FjiKsR4RJNYBIS0,1718
+aiteamutils-0.2.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.23.dist-info/RECORD,,