⚡新增 ISIM 电费查缴系统

provider/aufe/isim/depends.py

@@ -0,0 +1,116 @@
+from fastapi import Depends, HTTPException
+from database.creator import get_db_session
+from provider.loveac.authme import fetch_user_by_token
+from provider.aufe.isim import ISIMClient
+from provider.aufe.client import AUFEConnection
+from database.user import User
+from sqlalchemy.ext.asyncio import AsyncSession
+from typing import Dict
+from database.isim import ISIMRoomBinding
+from sqlalchemy import select
+
+
+# 全局ISIM客户端池
+_isim_clients: Dict[str, ISIMClient] = {}
+
+def get_cached_isim_client(user_id: str) -> ISIMClient:
+    """
+    获取缓存的ISIM客户端
+    
+    Args:
+        user_id: 用户ID
+        
+    Returns:
+        ISIMClient: 缓存的ISIM客户端实例，如果未找到则返回None
+    """
+    return _isim_clients.get(user_id)
+
+def cache_isim_client(user_id: str, client: ISIMClient) -> None:
+    """
+    缓存ISIM客户端
+    
+    Args:
+        user_id: 用户ID
+        client: ISIM客户端实例
+    """
+    _isim_clients[user_id] = client
+
+
+async def get_isim_client(
+    user: User = Depends(fetch_user_by_token),
+    session: AsyncSession = Depends(get_db_session),
+) -> ISIMClient:
+    from loguru import logger
+    """
+    获取ISIM客户端实例
+    
+    Args:
+        user: 用户对象（通过认证令牌获取）
+        
+    Returns:
+        ISIMClient: ISIM客户端实例
+        
+    Raises:
+        HTTPException: 认证失败时抛出
+    """
+    if not user:
+        raise HTTPException(status_code=400, detail="无效的令牌或用户不存在")
+    
+    # 首先检查是否已有缓存的ISIM客户端
+    cached_client = get_cached_isim_client(user.userid)
+    if cached_client:
+        # 检查缓存的客户端是否仍然有效
+        try:
+            if cached_client.is_session_valid():
+                from loguru import logger
+                logger.info(f"复用缓存的ISIM客户端: user_id={user.userid}")
+                return cached_client
+        except Exception as e:
+            from loguru import logger
+            logger.warning(f"缓存的ISIM客户端无效，将重新创建: {str(e)}")
+    
+    # 创建或获取VPN连接
+    aufe = AUFEConnection.create_or_get_connection("vpn.aufe.edu.cn", user.userid)
+    
+    # 检查VPN登录状态
+    if not aufe.login_status():
+        userid = user.userid
+        easyconnect_password = user.easyconnect_password
+        if not await aufe.login(userid, easyconnect_password):
+            raise HTTPException(
+                status_code=400,
+                detail="VPN登录失败，请检查用户名和密码",
+            )
+    
+    # 检查UAAP登录状态
+    if not aufe.uaap_login_status():
+        userid = user.userid
+        password = user.password
+        if not await aufe.uaap_login(userid, password):
+            raise HTTPException(
+                status_code=400,
+                detail="大学登录失败，请检查用户名和密码",
+            )
+    
+    # 创建新的ISIM客户端
+    isim_client = ISIMClient(aufe)
+
+
+    result_query = await session.execute(
+            select(ISIMRoomBinding).where(ISIMRoomBinding.userid == user.userid)
+        )
+    binding_record = result_query.scalars().first()
+    if binding_record:
+        logger.info(f"找到用户({user.userid})绑定记录，进行启动再绑定")
+        await isim_client.bind_room(
+                building_code=binding_record.building_code,
+                floor_code=binding_record.floor_code,
+                room_code=binding_record.room_code,
+            )
+    
+    # 缓存客户端
+    cache_isim_client(user.userid, isim_client)
+
+    logger.info(f"创建并缓存新的ISIM客户端: user_id={user.userid}")
+    
+    return isim_client