288 lines
10 KiB
Python
288 lines
10 KiB
Python
from typing import Optional
|
||
from urllib.parse import unquote
|
||
from loguru import logger
|
||
from provider.aufe.aac.model import (
|
||
LoveACScoreInfo,
|
||
LoveACScoreInfoResponse,
|
||
LoveACScoreListResponse,
|
||
SimpleResponse,
|
||
ErrorLoveACScoreInfo,
|
||
ErrorLoveACScoreInfoResponse,
|
||
ErrorLoveACScoreListResponse,
|
||
ErrorLoveACScoreCategory,
|
||
)
|
||
from provider.aufe.client import (
|
||
AUFEConnection,
|
||
AUFEConfig,
|
||
activity_tracker,
|
||
retry_async,
|
||
AUFEConnectionError,
|
||
AUFELoginError,
|
||
AUFEParseError,
|
||
RetryConfig
|
||
)
|
||
|
||
|
||
class AACConfig:
|
||
"""AAC 模块配置常量"""
|
||
BASE_URL = "http://api-dekt-ac-acxk-net.vpn2.aufe.edu.cn:8118"
|
||
WEB_URL = "http://dekt-ac-acxk-net.vpn2.aufe.edu.cn:8118"
|
||
LOGIN_SERVICE_URL = "http://uaap-aufe-edu-cn.vpn2.aufe.edu.cn:8118/cas/login?service=http%3a%2f%2fapi.dekt.ac.acxk.net%2fUser%2fIndex%2fCoreLoginCallback%3fisCASGateway%3dtrue"
|
||
|
||
|
||
@retry_async()
|
||
async def get_system_token(vpn_connection: AUFEConnection) -> Optional[str]:
|
||
"""
|
||
获取系统令牌 (sys_token)
|
||
|
||
Args:
|
||
vpn_connection: VPN连接实例
|
||
|
||
Returns:
|
||
Optional[str]: 系统令牌，失败时返回None
|
||
|
||
Raises:
|
||
AUFEConnectionError: 连接失败
|
||
AUFEParseError: 令牌解析失败
|
||
"""
|
||
try:
|
||
next_location = AACConfig.LOGIN_SERVICE_URL
|
||
max_redirects = 10 # 防止无限重定向
|
||
redirect_count = 0
|
||
|
||
while redirect_count < max_redirects:
|
||
response = await vpn_connection.requester().get(
|
||
next_location, follow_redirects=False
|
||
)
|
||
|
||
# 如果是重定向，继续跟踪
|
||
if response.status_code in (301, 302, 303, 307, 308):
|
||
next_location = response.headers.get("Location")
|
||
if not next_location:
|
||
raise AUFEParseError("重定向响应中缺少Location头")
|
||
|
||
logger.debug(f"重定向到: {next_location}")
|
||
redirect_count += 1
|
||
|
||
if "register?ticket=" in next_location:
|
||
logger.info(f"重定向到爱安财注册页面: {next_location}")
|
||
try:
|
||
sys_token = next_location.split("ticket=")[-1]
|
||
# URL编码转为正常字符串
|
||
sys_token = unquote(sys_token)
|
||
if sys_token:
|
||
logger.info(f"获取到系统令牌: {sys_token[:10]}...")
|
||
return sys_token
|
||
else:
|
||
raise AUFEParseError("提取的系统令牌为空")
|
||
except Exception as e:
|
||
raise AUFEParseError(f"解析系统令牌失败: {str(e)}") from e
|
||
else:
|
||
break
|
||
|
||
if redirect_count >= max_redirects:
|
||
raise AUFEConnectionError(f"重定向次数过多 ({max_redirects})")
|
||
|
||
raise AUFEParseError("未能从重定向中获取到系统令牌")
|
||
|
||
except (AUFEConnectionError, AUFEParseError):
|
||
raise
|
||
except Exception as e:
|
||
logger.error(f"获取系统令牌异常: {str(e)}")
|
||
raise AUFEConnectionError(f"获取系统令牌失败: {str(e)}") from e
|
||
|
||
|
||
class AACClient:
|
||
"""爱安财系统客户端"""
|
||
|
||
def __init__(
|
||
self,
|
||
vpn_connection: AUFEConnection,
|
||
ticket: Optional[str] = None,
|
||
retry_config: Optional[RetryConfig] = None
|
||
):
|
||
"""
|
||
初始化爱安财系统客户端
|
||
|
||
Args:
|
||
vpn_connection: VPN连接实例
|
||
ticket: 系统令牌
|
||
retry_config: 重试配置
|
||
"""
|
||
self.vpn_connection = vpn_connection
|
||
self.base_url = AACConfig.BASE_URL.rstrip("/")
|
||
self.web_url = AACConfig.WEB_URL.rstrip("/")
|
||
self.twfid = vpn_connection.get_twfid()
|
||
self.system_token: Optional[str] = ticket
|
||
self.retry_config = retry_config or RetryConfig()
|
||
|
||
logger.info(
|
||
f"爱安财系统客户端初始化: base_url={self.base_url}, web_url={self.web_url}"
|
||
)
|
||
|
||
def _get_default_headers(self) -> dict:
|
||
"""获取默认请求头"""
|
||
return {
|
||
**AUFEConfig.DEFAULT_HEADERS,
|
||
"ticket": self.system_token or "",
|
||
"sdp-app-session": self.twfid or "",
|
||
}
|
||
|
||
@activity_tracker
|
||
@retry_async()
|
||
async def validate_connection(self) -> bool:
|
||
"""
|
||
验证爱安财系统连接
|
||
|
||
Returns:
|
||
bool: 连接是否有效
|
||
|
||
Raises:
|
||
AUFEConnectionError: 连接失败
|
||
"""
|
||
try:
|
||
headers = AUFEConfig.DEFAULT_HEADERS.copy()
|
||
|
||
response = await self.vpn_connection.requester().get(
|
||
f"{self.web_url}/", headers=headers
|
||
)
|
||
is_valid = response.status_code == 200
|
||
|
||
logger.info(
|
||
f"爱安财系统连接验证结果: {'有效' if is_valid else '无效'} (HTTP状态码: {response.status_code})"
|
||
)
|
||
|
||
if not is_valid:
|
||
raise AUFEConnectionError(f"爱安财系统连接验证失败，状态码: {response.status_code}")
|
||
|
||
return is_valid
|
||
|
||
except AUFEConnectionError:
|
||
raise
|
||
except Exception as e:
|
||
logger.error(f"验证爱安财系统连接异常: {str(e)}")
|
||
raise AUFEConnectionError(f"验证连接失败: {str(e)}") from e
|
||
|
||
@activity_tracker
|
||
async def fetch_score_info(self) -> LoveACScoreInfo:
|
||
"""
|
||
获取爱安财总分信息，使用重试机制
|
||
|
||
Returns:
|
||
LoveACScoreInfo: 总分信息，失败时返回错误模型
|
||
"""
|
||
try:
|
||
logger.info("开始获取爱安财总分信息")
|
||
|
||
headers = self._get_default_headers()
|
||
|
||
# 使用新的重试机制
|
||
score_response = await self.vpn_connection.model_request(
|
||
model=LoveACScoreInfoResponse,
|
||
url=f"{self.base_url}/User/Center/DoGetScoreInfo?sf_request_type=ajax",
|
||
method="POST",
|
||
headers=headers,
|
||
data={}, # 空的POST请求体
|
||
follow_redirects=True,
|
||
)
|
||
|
||
if score_response and score_response.code == 0 and score_response.data:
|
||
logger.info(
|
||
f"爱安财总分信息获取成功: {score_response.data.total_score}分"
|
||
)
|
||
return score_response.data
|
||
else:
|
||
error_msg = score_response.msg if score_response else '未知错误'
|
||
logger.error(f"获取爱安财总分信息失败: {error_msg}")
|
||
# 返回错误模型
|
||
return ErrorLoveACScoreInfo(
|
||
TotalScore=-1.0,
|
||
IsTypeAdopt=False,
|
||
TypeAdoptResult=f"请求失败: {error_msg}",
|
||
)
|
||
except (AUFEConnectionError, AUFEParseError) as e:
|
||
logger.error(f"获取爱安财总分信息失败: {str(e)}")
|
||
return ErrorLoveACScoreInfo(
|
||
TotalScore=-1.0,
|
||
IsTypeAdopt=False,
|
||
TypeAdoptResult=f"请求失败: {str(e)}",
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"获取爱安财总分信息异常: {str(e)}")
|
||
# 返回错误模型
|
||
return ErrorLoveACScoreInfo(
|
||
TotalScore=-1.0,
|
||
IsTypeAdopt=False,
|
||
TypeAdoptResult="系统错误，请稍后重试",
|
||
)
|
||
|
||
@activity_tracker
|
||
async def fetch_score_list(
|
||
self, page_index: int = 1, page_size: int = 10
|
||
) -> LoveACScoreListResponse:
|
||
"""
|
||
获取爱安财分数列表，使用重试机制
|
||
|
||
Args:
|
||
page_index: 页码，默认为1
|
||
page_size: 每页大小，默认为10
|
||
|
||
Returns:
|
||
LoveACScoreListResponse: 分数列表响应，失败时返回错误模型
|
||
"""
|
||
def _create_error_response(error_msg: str) -> ErrorLoveACScoreListResponse:
|
||
"""创建错误响应模型"""
|
||
return ErrorLoveACScoreListResponse(
|
||
code=-1,
|
||
msg=error_msg,
|
||
data=[
|
||
ErrorLoveACScoreCategory(
|
||
ID="error",
|
||
ShowNum=-1,
|
||
TypeName="请求失败",
|
||
TotalScore=-1.0,
|
||
children=[],
|
||
)
|
||
],
|
||
)
|
||
|
||
try:
|
||
logger.info(
|
||
f"开始获取爱安财分数列表，页码: {page_index}, 每页大小: {page_size}"
|
||
)
|
||
|
||
headers = self._get_default_headers()
|
||
data = {"pageIndex": str(page_index), "pageSize": str(page_size)}
|
||
|
||
# 使用新的重试机制
|
||
score_list_response = await self.vpn_connection.model_request(
|
||
model=LoveACScoreListResponse,
|
||
url=f"{self.base_url}/User/Center/DoGetScoreList?sf_request_type=ajax",
|
||
method="POST",
|
||
headers=headers,
|
||
data=data,
|
||
follow_redirects=True,
|
||
)
|
||
|
||
if (
|
||
score_list_response
|
||
and score_list_response.code == 0
|
||
and score_list_response.data
|
||
):
|
||
logger.info(
|
||
f"爱安财分数列表获取成功，分类数量: {len(score_list_response.data)}"
|
||
)
|
||
return score_list_response
|
||
else:
|
||
error_msg = score_list_response.msg if score_list_response else '未知错误'
|
||
logger.error(f"获取爱安财分数列表失败: {error_msg}")
|
||
return _create_error_response(f"请求失败: {error_msg}")
|
||
|
||
except (AUFEConnectionError, AUFEParseError) as e:
|
||
logger.error(f"获取爱安财分数列表失败: {str(e)}")
|
||
return _create_error_response(f"请求失败: {str(e)}")
|
||
except Exception as e:
|
||
logger.error(f"获取爱安财分数列表异常: {str(e)}")
|
||
return _create_error_response("系统错误，已进行多次重试")
|
||
|