67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
|
from fastapi import Depends, HTTPException
|
|||
|
|
from loguru import logger
|
|||
|
|
from provider.loveac.authme import fetch_user_by_token
|
|||
|
|
from provider.aufe.aac import AACClient, get_system_token
|
|||
|
|
from provider.aufe.client import AUFEConnection
|
|||
|
|
from database.user import User, AACTicket
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
from database.creator import get_db_session
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_aac_client(
|
|||
|
|
user: User = Depends(fetch_user_by_token),
|
|||
|
|
db: AsyncSession = Depends(get_db_session),
|
|||
|
|
) -> AACClient:
|
|||
|
|
"""
|
|||
|
|
获取AAC客户端
|
|||
|
|
:param user: 用户信息
|
|||
|
|
:return: AACClient
|
|||
|
|
:raises HTTPException: 如果用户无效或登录失败
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
if not user:
|
|||
|
|
raise HTTPException(status_code=400, detail="无效的令牌或用户不存在")
|
|||
|
|
aufe = AUFEConnection.create_or_get_connection("vpn.aufe.edu.cn", user.userid)
|
|||
|
|
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登录失败，请检查用户名和密码",
|
|||
|
|
)
|
|||
|
|
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="大学登录失败，请检查用户名和密码",
|
|||
|
|
)
|
|||
|
|
# 检查AAC Ticket是否存在
|
|||
|
|
async with db as session:
|
|||
|
|
result = await session.execute(
|
|||
|
|
select(AACTicket).where(AACTicket.userid == user.userid)
|
|||
|
|
)
|
|||
|
|
aac_ticket = result.scalars().first()
|
|||
|
|
if not aac_ticket:
|
|||
|
|
# 如果不存在，尝试获取新的AAC Ticket
|
|||
|
|
logger.info(f"用户 {user.userid} 的 AAC Ticket 不存在，正在获取新的 Ticket")
|
|||
|
|
aac_ticket = await get_system_token(aufe)
|
|||
|
|
if not aac_ticket:
|
|||
|
|
logger.error(f"用户 {user.userid} 获取 AAC Ticket 失败")
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=400,
|
|||
|
|
detail="获取AAC Ticket失败，请稍后再试",
|
|||
|
|
)
|
|||
|
|
# 保存到数据库
|
|||
|
|
async with db as session:
|
|||
|
|
session.add(AACTicket(userid=user.userid, aac_token=aac_ticket))
|
|||
|
|
await session.commit()
|
|||
|
|
logger.success(f"用户 {user.userid} 成功获取并保存新的 AAC Ticket")
|
|||
|
|
else:
|
|||
|
|
logger.info(f"用户 {user.userid} 使用现有的 AAC Ticket")
|
|||
|
|
aac_ticket = aac_ticket.aac_token
|
|||
|
|
return AACClient(aufe, aac_ticket)
|