"""
模型包 - 统一导出所有数据库模型和会话管理
所有模型现在都在 app.models.database.models 中定义
"""

# 导出数据库实例和所有模型
from app.models.database import (
    db,
    # 用户相关
    User, UserSession, Permission, Role,
    # 题目相关
    ChallengeRecord,
    # 部署相关
    DeploymentRecord,
    # 系统配置
    SystemConfig,
    # Augment配置
    AugmentSessionToken,
    # 管理类
    ChallengeManager
)

# 导出会话管理
from app.models.session import (
    SessionManager,
    SessionData,
    update_session_state,
    get_session_state,
    clear_session_state
)

__all__ = [
    'db',
    # 用户相关
    'User', 'UserSession', 'Permission', 'Role',
    # 题目相关
    'ChallengeRecord',
    # 部署相关
    'DeploymentRecord',
    # 系统配置
    'SystemConfig',
    # Augment配置
    'AugmentSessionToken',
    # 管理类
    'ChallengeManager',
    # 会话管理
    'SessionManager', 'SessionData',
    'update_session_state', 'get_session_state', 'clear_session_state'
]