# -*- coding: utf-8 -*- import json from urllib.parse import urljoin from typing import Optional, Tuple from app.log import logger from app.core.config import settings from app.utils.http import RequestUtils from app.utils.string import StringUtils from app.modules.indexer.parser import SiteParserBase, SiteSchema class RousiSiteUserInfo(SiteParserBase): """ Rousi.pro 站点解析器 使用 API v1 接口，通过 Passkey (Bearer Token) 进行认证 """ schema = SiteSchema.RousiPro request_mode = "apikey" def _parse_site_page(self, html_text: str): """ 配置 API 请求地址和请求头 使用 API v1 的 /profile 接口获取用户信息 """ self._base_url = f"https://{StringUtils.get_url_domain(self._site_url)}" self._user_basic_page = "api/v1/profile?include_fields[user]=seeding_leeching_data" self._user_basic_params = {} self._user_basic_headers = { "Content-Type": "application/json", "Accept": "application/json", "Authorization": f"Bearer {self.apikey}" } # Rousi.pro API v1 在单个接口返回所有信息，无需额外页面 self._user_traffic_page = None self._user_detail_page = None self._torrent_seeding_page = None self._user_mail_unread_page = None self._sys_mail_unread_page = None def _parse_logged_in(self, html_text): """ 判断是否登录成功 API 认证模式下，通过 HTTP 状态码判断，此处始终返回 True """ return True def _parse_user_base_info(self, html_text: str): """ 解析用户基本信息 通过 API v1 接口获取用户完整信息，包括上传下载量、做种数据等 API 响应示例： { "code": 0, "message": "success", "data": { "id": 1, "username": "example", "level_text": "Lv.5", "registered_at": "2024-01-01T00:00:00Z", "uploaded": 1073741824, "downloaded": 536870912, "ratio": 2.0, "karma": 1000.5, "seeding_leeching_data": { "seeding_count": 10, "seeding_size": 10737418240, "leeching_count": 2, "leeching_size": 2147483648 } } } """ if not html_text: return try: data = json.loads(html_text) except json.JSONDecodeError: logger.error(f"{self._site_name} JSON 解析失败") return if not data or data.get("code") != 0: self.err_msg = data.get("message", "未知错误") logger.warn(f"{self._site_name} API 错误: {self.err_msg}") return user_info = data.get("data") if not user_info: return # 基本信息 self.userid = user_info.get("id") self.username = user_info.get("username") self.user_level = user_info.get("level_text") or user_info.get("role_text") # 注册时间：统一格式为 YYYY-MM-DD HH:MM:SS join_at = StringUtils.unify_datetime_str(user_info.get("registered_at")) if join_at: # 确保格式为 YYYY-MM-DD HH:MM:SS (19位) if len(join_at) >= 19: self.join_at = join_at[:19] else: self.join_at = join_at # 流量信息 self.upload = int(user_info.get("uploaded") or 0) self.download = int(user_info.get("downloaded") or 0) self.ratio = round(float(user_info.get("ratio") or 0), 2) # 魔力值（站点称为 karma） self.bonus = float(user_info.get("karma") or 0) # 做种/下载中数据 sl_data = user_info.get("seeding_leeching_data", {}) self.seeding = int(sl_data.get("seeding_count") or 0) self.seeding_size = int(sl_data.get("seeding_size") or 0) self.leeching = int(sl_data.get("leeching_count") or 0) self.leeching_size = int(sl_data.get("leeching_size") or 0) def _parse_user_traffic_info(self, html_text: str): """ 解析用户流量信息 Rousi.pro API v1 在 _parse_user_base_info 中已完成所有解析，此方法无需实现 """ pass def _parse_user_detail_info(self, html_text: str): """ 解析用户详细信息 Rousi.pro API v1 在 _parse_user_base_info 中已完成所有解析，此方法无需实现 """ pass def _parse_user_torrent_seeding_info(self, html_text: str, multi_page: Optional[bool] = False) -> Optional[str]: """ 解析用户做种信息 Rousi.pro API v1 在 _parse_user_base_info 中已通过 seeding_leeching_data 获取做种数据 :param html_text: 页面内容 :param multi_page: 是否多页数据 :return: 下页地址（无下页返回 None） """ return None def _parse_message_unread_links(self, html_text: str, msg_links: list) -> Optional[str]: """ 解析未读消息链接 Rousi.pro API v1 暂未提供消息相关接口 :param html_text: 页面内容 :param msg_links: 消息链接列表 :return: 下页地址（无下页返回 None） """ return None def _parse_message_content(self, html_text) -> Tuple[Optional[str], Optional[str], Optional[str]]: """ 解析消息内容 Rousi.pro API v1 暂未提供消息相关接口 :param html_text: 页面内容 :return: (标题, 日期, 内容) """ return None, None, None def _pase_unread_msgs(self): """ 解析所有未读消息标题和内容 Rousi.pro API v1 暂未提供消息相关接口，暂时以网页接口实现 :return: """ if not self.token: logger.warn(f"{self._site_name} 站点未配置 Authorization 请求头，跳过消息解析") return headers = { "User-Agent": self._ua, "Accept": "application/json, text/plain, */*", "Authorization": self.token if self.token.startswith("Bearer ") else f"Bearer {self.token}" } def __get_message_list(page: int): params = { "page": page, "page_size": 100, "unread_only": "true" } res = RequestUtils( headers=headers, timeout=60, proxies=settings.PROXY if self._proxy else None ).get_res( url=urljoin(self._base_url, "api/messages"), params=params ) if not res or res.status_code != 200 or res.json().get("code", -1) != 0: logger.warn(f"{self._site_name} 站点解析消息失败，状态码: {res.status_code if res else '无响应'}") return { "messages": [], "total_pages": 0 } return res.json().get("data") # 分页获取所有未读消息 page = 0 res = __get_message_list(page) page += 1 messages = res.get("messages", []) total_pages = res.get("total_pages", 0) while page < total_pages: res = __get_message_list(page) messages.extend(res.get("messages", [])) page += 1 self.message_unread = len(messages) for messsage in messages: head = messsage.get("title") date = StringUtils.unify_datetime_str(messsage.get("created_at")) content = messsage.get("content") logger.debug(f"{self._site_name} 标题 {head} 时间 {date} 内容 {content}") self.message_unread_contents.append((head, date, content)) # 更新消息为已读 RequestUtils( headers=headers, timeout=60, proxies=settings.PROXY if self._proxy else None ).post_res( url=urljoin(self._base_url, "api/messages/read-all") )