mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-06-07 07:26:49 +00:00
1893 lines
75 KiB
Python
1893 lines
75 KiB
Python
import re
|
||
import traceback
|
||
from datetime import datetime, timedelta
|
||
from multiprocessing.dummy import Pool as ThreadPool
|
||
from multiprocessing.pool import ThreadPool
|
||
from typing import Any, List, Dict, Tuple, Optional
|
||
from urllib.parse import urljoin
|
||
|
||
import pytz
|
||
from app import schemas
|
||
from app.core.config import settings
|
||
from app.core.event import eventmanager, Event
|
||
from app.db.site_oper import SiteOper
|
||
from app.helper.browser import PlaywrightHelper
|
||
from app.helper.cloudflare import under_challenge
|
||
from app.helper.module import ModuleHelper
|
||
from app.helper.sites import SitesHelper
|
||
from app.log import logger
|
||
from app.plugins import _PluginBase
|
||
from app.schemas.types import EventType, NotificationType
|
||
from app.utils.http import RequestUtils
|
||
from app.utils.site import SiteUtils
|
||
from app.utils.string import StringUtils
|
||
from app.utils.timer import TimerUtils
|
||
from apscheduler.schedulers.background import BackgroundScheduler
|
||
from apscheduler.triggers.cron import CronTrigger
|
||
from ruamel.yaml import CommentedMap
|
||
|
||
|
||
class AutoSignIn(_PluginBase):
|
||
# 插件名称
|
||
plugin_name = "站点自动签到"
|
||
# 插件描述
|
||
plugin_desc = "自动模拟登录、签到站点。"
|
||
# 插件图标
|
||
plugin_icon = "signin.png"
|
||
# 插件版本
|
||
plugin_version = "2.9.0"
|
||
# 插件作者
|
||
plugin_author = "thsrite"
|
||
# 作者主页
|
||
author_url = "https://github.com/thsrite"
|
||
# 插件配置项ID前缀
|
||
plugin_config_prefix = "autosignin_"
|
||
# 加载顺序
|
||
plugin_order = 0
|
||
# 可使用的用户级别
|
||
auth_level = 2
|
||
|
||
# 定时器
|
||
_scheduler: Optional[BackgroundScheduler] = None
|
||
# 加载的模块
|
||
_site_schema: list = []
|
||
|
||
# 配置属性
|
||
_enabled: bool = False
|
||
_cron: str = ""
|
||
_onlyonce: bool = False
|
||
_notify: bool = False
|
||
_queue_cnt: int = 5
|
||
_sign_sites: list = []
|
||
_login_sites: list = []
|
||
_retry_keyword = None
|
||
_clean: bool = False
|
||
_start_time: int = None
|
||
_end_time: int = None
|
||
_auto_cf: int = 0
|
||
|
||
def init_plugin(self, config: dict = None):
|
||
|
||
# 停止现有任务
|
||
self.stop_service()
|
||
|
||
# 配置
|
||
if config:
|
||
self._enabled = config.get("enabled")
|
||
self._cron = config.get("cron")
|
||
self._onlyonce = config.get("onlyonce")
|
||
self._notify = config.get("notify")
|
||
self._queue_cnt = config.get("queue_cnt") or 5
|
||
self._sign_sites = config.get("sign_sites") or []
|
||
self._login_sites = config.get("login_sites") or []
|
||
self._retry_keyword = config.get("retry_keyword")
|
||
self._auto_cf = config.get("auto_cf")
|
||
self._clean = config.get("clean")
|
||
|
||
# 过滤掉已删除的站点
|
||
all_sites = [site.id for site in SiteOper().list_order_by_pri()] + [site.get("id") for site in
|
||
self.__custom_sites()]
|
||
self._sign_sites = [site_id for site_id in all_sites if site_id in self._sign_sites]
|
||
self._login_sites = [site_id for site_id in all_sites if site_id in self._login_sites]
|
||
# 保存配置
|
||
self.__update_config()
|
||
|
||
# 加载模块
|
||
if self._enabled or self._onlyonce:
|
||
|
||
self._site_schema = ModuleHelper.load('app.plugins.autosignin.sites',
|
||
filter_func=lambda _, obj: hasattr(obj, 'match'))
|
||
|
||
# 立即运行一次
|
||
if self._onlyonce:
|
||
# 定时服务
|
||
self._scheduler = BackgroundScheduler(timezone=settings.TZ)
|
||
logger.info("站点自动签到服务启动，立即运行一次")
|
||
self._scheduler.add_job(func=self.sign_in, trigger='date',
|
||
run_date=datetime.now(tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3),
|
||
name="站点自动签到")
|
||
|
||
# 关闭一次性开关
|
||
self._onlyonce = False
|
||
# 保存配置
|
||
self.__update_config()
|
||
|
||
# 启动任务
|
||
if self._scheduler.get_jobs():
|
||
self._scheduler.print_jobs()
|
||
self._scheduler.start()
|
||
|
||
def get_state(self) -> bool:
|
||
return self._enabled
|
||
|
||
def __update_config(self):
|
||
# 保存配置
|
||
self.update_config(
|
||
{
|
||
"enabled": self._enabled,
|
||
"notify": self._notify,
|
||
"cron": self._cron,
|
||
"onlyonce": self._onlyonce,
|
||
"queue_cnt": self._queue_cnt,
|
||
"sign_sites": self._sign_sites,
|
||
"login_sites": self._login_sites,
|
||
"retry_keyword": self._retry_keyword,
|
||
"auto_cf": self._auto_cf,
|
||
"clean": self._clean,
|
||
}
|
||
)
|
||
|
||
@staticmethod
|
||
def get_command() -> List[Dict[str, Any]]:
|
||
"""
|
||
定义远程控制命令
|
||
:return: 命令关键字、事件、描述、附带数据
|
||
"""
|
||
return [{
|
||
"cmd": "/site_signin",
|
||
"event": EventType.PluginAction,
|
||
"desc": "站点签到",
|
||
"category": "站点",
|
||
"data": {
|
||
"action": "site_signin"
|
||
}
|
||
}]
|
||
|
||
def get_api(self) -> List[Dict[str, Any]]:
|
||
"""
|
||
获取插件API
|
||
[{
|
||
"path": "/xx",
|
||
"endpoint": self.xxx,
|
||
"methods": ["GET", "POST"],
|
||
"summary": "API说明"
|
||
}]
|
||
"""
|
||
return [{
|
||
"path": "/signin_by_domain",
|
||
"endpoint": self.signin_by_domain,
|
||
"methods": ["GET"],
|
||
"summary": "站点签到",
|
||
"description": "使用站点域名签到站点",
|
||
}]
|
||
|
||
def get_service(self) -> List[Dict[str, Any]]:
|
||
"""
|
||
注册插件公共服务
|
||
[{
|
||
"id": "服务ID",
|
||
"name": "服务名称",
|
||
"trigger": "触发器：cron/interval/date/CronTrigger.from_crontab()",
|
||
"func": self.xxx,
|
||
"kwargs": {} # 定时器参数
|
||
}]
|
||
"""
|
||
if self._enabled and self._cron:
|
||
try:
|
||
if str(self._cron).strip().count(" ") == 4:
|
||
return [{
|
||
"id": "AutoSignIn",
|
||
"name": "站点自动签到服务",
|
||
"trigger": CronTrigger.from_crontab(self._cron),
|
||
"func": self.sign_in,
|
||
"kwargs": {}
|
||
}]
|
||
else:
|
||
# 2.3/9-23
|
||
crons = str(self._cron).strip().split("/")
|
||
if len(crons) == 2:
|
||
# 2.3
|
||
cron = crons[0]
|
||
# 9-23
|
||
times = crons[1].split("-")
|
||
if len(times) == 2:
|
||
# 9
|
||
self._start_time = int(times[0])
|
||
# 23
|
||
self._end_time = int(times[1])
|
||
if self._start_time and self._end_time:
|
||
return [{
|
||
"id": "AutoSignIn",
|
||
"name": "站点自动签到服务",
|
||
"trigger": "interval",
|
||
"func": self.sign_in,
|
||
"kwargs": {
|
||
"hours": float(str(cron).strip()),
|
||
}
|
||
}]
|
||
else:
|
||
logger.error("站点自动签到服务启动失败，周期格式错误")
|
||
else:
|
||
# 默认0-24 按照周期运行
|
||
return [{
|
||
"id": "AutoSignIn",
|
||
"name": "站点自动签到服务",
|
||
"trigger": "interval",
|
||
"func": self.sign_in,
|
||
"kwargs": {
|
||
"hours": float(str(self._cron).strip()),
|
||
}
|
||
}]
|
||
except Exception as err:
|
||
logger.error(f"定时任务配置错误：{str(err)}")
|
||
elif self._enabled:
|
||
# 随机时间
|
||
triggers = TimerUtils.random_scheduler(num_executions=2,
|
||
begin_hour=9,
|
||
end_hour=23,
|
||
max_interval=6 * 60,
|
||
min_interval=2 * 60)
|
||
ret_jobs = []
|
||
for trigger in triggers:
|
||
ret_jobs.append({
|
||
"id": f"AutoSignIn|{trigger.hour}:{trigger.minute}",
|
||
"name": "站点自动签到服务",
|
||
"trigger": "cron",
|
||
"func": self.sign_in,
|
||
"kwargs": {
|
||
"hour": trigger.hour,
|
||
"minute": trigger.minute
|
||
}
|
||
})
|
||
return ret_jobs
|
||
return []
|
||
|
||
def get_form(self) -> Tuple[List[dict], Dict[str, Any]]:
|
||
"""
|
||
拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构
|
||
"""
|
||
# 站点的可选项（内置站点 + 自定义站点）
|
||
customSites = self.__custom_sites()
|
||
|
||
site_options = ([{"title": site.name, "value": site.id}
|
||
for site in SiteOper().list_order_by_pri()]
|
||
+ [{"title": site.get("name"), "value": site.get("id")}
|
||
for site in customSites])
|
||
return [
|
||
{
|
||
'component': 'VForm',
|
||
'content': [
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'enabled',
|
||
'label': '启用插件',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'notify',
|
||
'label': '发送通知',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'onlyonce',
|
||
'label': '立即运行一次',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'clean',
|
||
'label': '清理本日缓存',
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VCronField',
|
||
'props': {
|
||
'model': 'cron',
|
||
'label': '执行周期',
|
||
'placeholder': '5位cron表达式，留空自动'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'queue_cnt',
|
||
'label': '队列数量'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'retry_keyword',
|
||
'label': '重试关键词',
|
||
'placeholder': '支持正则表达式，命中才重签'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'auto_cf',
|
||
'label': '自动优选',
|
||
'placeholder': '命中重试关键词次数（0-关闭）'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'content': [
|
||
{
|
||
'component': 'VSelect',
|
||
'props': {
|
||
'chips': True,
|
||
'multiple': True,
|
||
'model': 'sign_sites',
|
||
'label': '签到站点',
|
||
'items': site_options
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'content': [
|
||
{
|
||
'component': 'VSelect',
|
||
'props': {
|
||
'chips': True,
|
||
'multiple': True,
|
||
'model': 'login_sites',
|
||
'label': '登录站点',
|
||
'items': site_options
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '执行周期支持：'
|
||
'1、5位cron表达式；'
|
||
'2、配置间隔（小时），如2.3/9-23（9-23点之间每隔2.3小时执行一次）；'
|
||
'3、周期不填默认9-23点随机执行2次。'
|
||
'每天首次全量执行，其余执行命中重试关键词的站点。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '自动优选：0-关闭，命中重试关键词次数大于该数量时自动执行Cloudflare IP优选（需要开启且则正确配置Cloudflare IP优选插件和自定义Hosts插件）'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'warning',
|
||
'variant': 'tonal',
|
||
'text': '不是所有的站点都会把程序自动登录/签到定义为用户活跃（比如馒头），提示签到/登录成功仍然存在掉号风险！请结合站点公告说明自行把握。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
], {
|
||
"enabled": False,
|
||
"notify": True,
|
||
"cron": "",
|
||
"auto_cf": 0,
|
||
"onlyonce": False,
|
||
"clean": False,
|
||
"queue_cnt": 5,
|
||
"sign_sites": [],
|
||
"login_sites": [],
|
||
"retry_keyword": "错误|失败"
|
||
}
|
||
|
||
def __custom_sites(self) -> List[Any]:
|
||
custom_sites = []
|
||
custom_sites_config = self.get_config("CustomSites")
|
||
if custom_sites_config and custom_sites_config.get("enabled"):
|
||
custom_sites = custom_sites_config.get("sites")
|
||
return custom_sites
|
||
|
||
def get_page(self) -> List[dict]:
|
||
"""
|
||
拼装插件详情页面，需要返回页面配置，同时附带数据
|
||
"""
|
||
# 获取最近14天的日期数组
|
||
date_list = [(datetime.now() - timedelta(days=i)).date() for i in range(14)]
|
||
|
||
# 获取所有数据，包括签到和登录历史
|
||
all_data = {
|
||
"signin": [], # 签到数据
|
||
"login": [] # 登录数据
|
||
}
|
||
sites_info = {} # 记录站点信息
|
||
|
||
# 获取站点信息
|
||
site_indexers = SitesHelper().get_indexers()
|
||
for site in site_indexers:
|
||
if not site.get("public"):
|
||
sites_info[site.get("id")] = site.get("name")
|
||
|
||
# 自定义站点
|
||
custom_sites = self.__custom_sites()
|
||
for site in custom_sites:
|
||
sites_info[site.get("id")] = site.get("name")
|
||
|
||
# 获取常规日期格式数据
|
||
for day in date_list:
|
||
day_str = f"{day.month}月{day.day}日"
|
||
day_formatted = day.strftime('%Y-%m-%d')
|
||
|
||
# 获取"月日"格式数据
|
||
day_data = self.get_data(day_str)
|
||
if day_data:
|
||
# 添加日期信息到每条记录
|
||
if isinstance(day_data, list):
|
||
for record in day_data:
|
||
if isinstance(record, dict):
|
||
record["date"] = day_str
|
||
record["day_obj"] = day
|
||
# 区分签到和登录数据
|
||
if "登录" in record.get("status", ""):
|
||
all_data["login"].append(record)
|
||
else:
|
||
all_data["signin"].append(record)
|
||
|
||
# 获取"签到-yyyy-mm-dd"和"登录-yyyy-mm-dd"格式数据
|
||
signin_history = self.get_data(key="签到-" + day_formatted)
|
||
if signin_history and isinstance(signin_history, dict):
|
||
# 获取完成签到的站点ID列表
|
||
done_sites = signin_history.get("do", [])
|
||
retry_sites = signin_history.get("retry", [])
|
||
|
||
# 为所有已完成签到的站点创建记录
|
||
for site_id in done_sites:
|
||
site_name = self._get_site_display_name(site_id=site_id, sites_info=sites_info)
|
||
|
||
# 跳过需要重试的站点
|
||
if site_id in retry_sites:
|
||
status_text = "需要重试"
|
||
else:
|
||
status_text = "已签到"
|
||
all_data["signin"].append({
|
||
"site": site_name,
|
||
"status": status_text,
|
||
"date": day_str,
|
||
"day_obj": day,
|
||
"site_id": site_id
|
||
})
|
||
|
||
# 获取登录历史数据
|
||
login_history = self.get_data(key="登录-" + day_formatted)
|
||
if login_history and isinstance(login_history, dict):
|
||
# 获取完成登录的站点ID列表
|
||
done_sites = login_history.get("do", [])
|
||
retry_sites = login_history.get("retry", [])
|
||
|
||
# 为所有已完成登录的站点创建记录
|
||
for site_id in done_sites:
|
||
site_name = self._get_site_display_name(site_id=site_id, sites_info=sites_info)
|
||
|
||
# 跳过需要重试的站点
|
||
if site_id in retry_sites:
|
||
status_text = "登录需要重试"
|
||
else:
|
||
status_text = "登录成功"
|
||
all_data["login"].append({
|
||
"site": site_name,
|
||
"status": status_text,
|
||
"date": day_str,
|
||
"day_obj": day,
|
||
"site_id": site_id
|
||
})
|
||
|
||
# 如果没有数据且没有配置站点，显示提示信息
|
||
if not all_data["signin"] and not all_data["login"] and not self._sign_sites and not self._login_sites:
|
||
return [{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'text': '暂无签到数据',
|
||
'variant': 'tonal',
|
||
'class': 'mt-4',
|
||
'prepend-icon': 'mdi-information'
|
||
}
|
||
}]
|
||
|
||
# 按站点分组并去重数据
|
||
signin_site_data = {}
|
||
login_site_data = {}
|
||
|
||
# 处理签到数据 - 每个站点每天只保留一条最新记录
|
||
site_day_records = {} # 用于去重: {site}_{date} -> record
|
||
for data in all_data["signin"]:
|
||
site_name = data.get("site", "未知站点")
|
||
date_str = data.get("date", "")
|
||
site_day_key = f"{site_name}_{date_str}"
|
||
|
||
# 存储或更新记录（如有多条取最新）
|
||
site_day_records[site_day_key] = data
|
||
|
||
# 整理去重后的数据
|
||
for key, record in site_day_records.items():
|
||
site_name = record.get("site", "未知站点")
|
||
if site_name not in signin_site_data:
|
||
signin_site_data[site_name] = []
|
||
signin_site_data[site_name].append(record)
|
||
|
||
# 处理登录数据 - 同样去重
|
||
site_day_records = {} # 重置去重字典
|
||
for data in all_data["login"]:
|
||
site_name = data.get("site", "未知站点")
|
||
date_str = data.get("date", "")
|
||
site_day_key = f"{site_name}_{date_str}"
|
||
|
||
# 存储或更新记录
|
||
site_day_records[site_day_key] = data
|
||
|
||
# 整理去重后的数据
|
||
for key, record in site_day_records.items():
|
||
site_name = record.get("site", "未知站点")
|
||
if site_name not in login_site_data:
|
||
login_site_data[site_name] = []
|
||
login_site_data[site_name].append(record)
|
||
|
||
# 补齐已配置但暂无历史记录的站点，详情页能直接看出未记录项。
|
||
for site_id in self._sign_sites:
|
||
site_name = self._get_site_display_name(site_id=site_id, sites_info=sites_info)
|
||
signin_site_data.setdefault(site_name, [])
|
||
for site_id in self._login_sites:
|
||
site_name = self._get_site_display_name(site_id=site_id, sites_info=sites_info)
|
||
login_site_data.setdefault(site_name, [])
|
||
|
||
display_dates = date_list[:7]
|
||
today_label = self._date_label(day=date_list[0])
|
||
signin_stats = self._calculate_day_stats(site_data=signin_site_data, date_label=today_label)
|
||
login_stats = self._calculate_day_stats(site_data=login_site_data, date_label=today_label)
|
||
|
||
# 添加紧凑状态矩阵样式
|
||
return [
|
||
{
|
||
'component': 'style',
|
||
'text': """
|
||
.autosignin-page {
|
||
display: flex;
|
||
flex-direction: column;
|
||
gap: 12px;
|
||
}
|
||
.autosignin-summary {
|
||
display: grid;
|
||
grid-template-columns: repeat(4, minmax(0, 1fr));
|
||
gap: 8px;
|
||
}
|
||
.autosignin-stat {
|
||
min-width: 0;
|
||
padding: 10px 12px;
|
||
}
|
||
.autosignin-stat__head {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 6px;
|
||
min-width: 0;
|
||
color: rgba(var(--v-theme-on-surface), var(--v-medium-emphasis-opacity));
|
||
font-size: .75rem;
|
||
font-weight: 600;
|
||
line-height: 1.25;
|
||
overflow: hidden;
|
||
text-overflow: ellipsis;
|
||
white-space: nowrap;
|
||
}
|
||
.autosignin-stat__head .v-icon {
|
||
color: rgb(var(--app-card-accent-rgb));
|
||
}
|
||
.autosignin-stat__value {
|
||
margin-top: 8px;
|
||
font-size: 1.25rem;
|
||
font-weight: 700;
|
||
line-height: 1;
|
||
letter-spacing: 0;
|
||
}
|
||
.autosignin-stat__meta {
|
||
margin-top: 4px;
|
||
color: rgba(var(--v-theme-on-surface), .56);
|
||
font-size: .72rem;
|
||
line-height: 1.25;
|
||
white-space: nowrap;
|
||
overflow: hidden;
|
||
text-overflow: ellipsis;
|
||
}
|
||
.autosignin-section {
|
||
min-width: 0;
|
||
}
|
||
.autosignin-section-head {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 8px;
|
||
min-height: 30px;
|
||
margin-bottom: 6px;
|
||
}
|
||
.autosignin-section-title {
|
||
font-size: .95rem;
|
||
font-weight: 700;
|
||
letter-spacing: 0;
|
||
}
|
||
.autosignin-table-wrap {
|
||
overflow-x: auto;
|
||
border: 1px solid rgba(var(--v-theme-on-surface), .08);
|
||
border-radius: 8px;
|
||
}
|
||
.autosignin-table {
|
||
min-width: 620px;
|
||
}
|
||
html[data-theme="transparent"] .autosignin-table-wrap,
|
||
.v-theme--transparent .autosignin-table-wrap {
|
||
backdrop-filter: blur(var(--transparent-blur, 10px));
|
||
background-color: rgba(var(--v-theme-surface), 0) !important;
|
||
}
|
||
html[data-theme="transparent"] .autosignin-table,
|
||
html[data-theme="transparent"] .autosignin-table .v-table__wrapper,
|
||
html[data-theme="transparent"] .autosignin-table table,
|
||
html[data-theme="transparent"] .autosignin-table tbody tr,
|
||
.v-theme--transparent .autosignin-table,
|
||
.v-theme--transparent .autosignin-table .v-table__wrapper,
|
||
.v-theme--transparent .autosignin-table table,
|
||
.v-theme--transparent .autosignin-table tbody tr {
|
||
background-color: transparent !important;
|
||
}
|
||
.autosignin-table th {
|
||
height: 34px !important;
|
||
padding: 0 8px !important;
|
||
color: rgba(var(--v-theme-on-surface), .62);
|
||
font-size: .75rem;
|
||
font-weight: 600 !important;
|
||
white-space: nowrap;
|
||
}
|
||
.autosignin-table td {
|
||
height: 38px !important;
|
||
padding: 0 8px !important;
|
||
vertical-align: middle;
|
||
}
|
||
.autosignin-table tbody tr:last-child td {
|
||
border-bottom: 0 !important;
|
||
}
|
||
.autosignin-site-name {
|
||
max-width: 160px;
|
||
overflow: hidden;
|
||
text-overflow: ellipsis;
|
||
white-space: nowrap;
|
||
font-weight: 600;
|
||
line-height: 1.2;
|
||
}
|
||
.autosignin-site-meta {
|
||
margin-top: 2px;
|
||
color: rgba(var(--v-theme-on-surface), .52);
|
||
font-size: .68rem;
|
||
line-height: 1.1;
|
||
}
|
||
.autosignin-status-cell {
|
||
min-width: 92px;
|
||
}
|
||
.autosignin-dot-cell {
|
||
width: 40px;
|
||
text-align: center;
|
||
}
|
||
.autosignin-dot {
|
||
width: 22px;
|
||
height: 22px;
|
||
display: inline-flex;
|
||
align-items: center;
|
||
justify-content: center;
|
||
border-radius: 999px;
|
||
border: 1px solid transparent;
|
||
}
|
||
.autosignin-dot--success {
|
||
color: rgb(var(--v-theme-success));
|
||
background: rgba(var(--v-theme-success), .10);
|
||
}
|
||
.autosignin-dot--warning {
|
||
color: rgb(var(--v-theme-warning));
|
||
background: rgba(var(--v-theme-warning), .14);
|
||
}
|
||
.autosignin-dot--error {
|
||
color: rgb(var(--v-theme-error));
|
||
background: rgba(var(--v-theme-error), .12);
|
||
}
|
||
.autosignin-dot--none {
|
||
color: rgba(var(--v-theme-on-surface), .42);
|
||
background: rgba(var(--v-theme-on-surface), .06);
|
||
border-color: rgba(var(--v-theme-on-surface), .08);
|
||
}
|
||
@media (max-width: 720px) {
|
||
.autosignin-page {
|
||
gap: 10px;
|
||
}
|
||
.autosignin-summary {
|
||
grid-template-columns: repeat(2, minmax(0, 1fr));
|
||
}
|
||
.autosignin-stat {
|
||
padding: 8px 10px;
|
||
}
|
||
.autosignin-stat__value {
|
||
font-size: 1.08rem;
|
||
}
|
||
.autosignin-table {
|
||
min-width: 560px;
|
||
}
|
||
.autosignin-table th,
|
||
.autosignin-table td {
|
||
padding-left: 6px !important;
|
||
padding-right: 6px !important;
|
||
}
|
||
.autosignin-site-name {
|
||
max-width: 126px;
|
||
}
|
||
.autosignin-dot-cell {
|
||
width: 34px;
|
||
}
|
||
}
|
||
"""
|
||
},
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-page'
|
||
},
|
||
'content': [
|
||
self._build_summary(signin_stats=signin_stats, login_stats=login_stats, days=len(date_list)),
|
||
self._build_status_section(
|
||
title="签到状态",
|
||
icon="mdi-calendar-check",
|
||
site_data=signin_site_data,
|
||
display_dates=display_dates,
|
||
empty_text="暂无签到数据"
|
||
),
|
||
self._build_status_section(
|
||
title="登录状态",
|
||
icon="mdi-login-variant",
|
||
site_data=login_site_data,
|
||
display_dates=display_dates,
|
||
empty_text="暂无登录数据"
|
||
)
|
||
]
|
||
}
|
||
]
|
||
|
||
@staticmethod
|
||
def _get_site_display_name(site_id, sites_info: dict) -> str:
|
||
"""
|
||
根据站点ID获取详情页中展示的站点名称。
|
||
"""
|
||
site_id_str = str(site_id)
|
||
return sites_info.get(site_id_str) or sites_info.get(site_id) or f"站点ID: {site_id}"
|
||
|
||
@staticmethod
|
||
def _status_meta(status_text: str) -> dict:
|
||
"""
|
||
将签到或登录状态文本转换为页面展示需要的颜色、图标和排序权重。
|
||
"""
|
||
status_text = str(status_text or "").strip()
|
||
if "Cookie已失效" in status_text or "失效" in status_text:
|
||
return {
|
||
"level": "error",
|
||
"color": "error",
|
||
"icon": "mdi-cookie-off",
|
||
"label": status_text or "Cookie失效",
|
||
"sort": 0
|
||
}
|
||
if "失败" in status_text or "错误" in status_text:
|
||
return {
|
||
"level": "error",
|
||
"color": "error",
|
||
"icon": "mdi-alert-circle",
|
||
"label": status_text or "失败",
|
||
"sort": 0
|
||
}
|
||
if "重试" in status_text:
|
||
return {
|
||
"level": "warning",
|
||
"color": "warning",
|
||
"icon": "mdi-refresh",
|
||
"label": status_text or "需要重试",
|
||
"sort": 1
|
||
}
|
||
if "成功" in status_text or "已签到" in status_text:
|
||
return {
|
||
"level": "success",
|
||
"color": "success",
|
||
"icon": "mdi-check-circle",
|
||
"label": status_text or "成功",
|
||
"sort": 3
|
||
}
|
||
if status_text:
|
||
return {
|
||
"level": "none",
|
||
"color": "grey",
|
||
"icon": "mdi-help-circle-outline",
|
||
"label": status_text,
|
||
"sort": 2
|
||
}
|
||
return {
|
||
"level": "none",
|
||
"color": "grey",
|
||
"icon": "mdi-minus-circle-outline",
|
||
"label": "未记录",
|
||
"sort": 2
|
||
}
|
||
|
||
@staticmethod
|
||
def _latest_record(records: list, date_label: str = None) -> dict:
|
||
"""
|
||
获取指定日期或整个记录列表中的最新一条记录。
|
||
"""
|
||
filtered_records = records
|
||
if date_label:
|
||
filtered_records = [record for record in records if record.get("date") == date_label]
|
||
if not filtered_records:
|
||
return {}
|
||
try:
|
||
return sorted(
|
||
filtered_records,
|
||
key=lambda item: item.get("day_obj", datetime.min.date()),
|
||
reverse=True
|
||
)[0]
|
||
except Exception as e:
|
||
logger.debug(f"获取最新记录失败: {str(e)}")
|
||
return filtered_records[0]
|
||
|
||
@staticmethod
|
||
def _date_label(day) -> str:
|
||
"""
|
||
将日期对象格式化为历史记录使用的月日标签。
|
||
"""
|
||
return f"{day.month}月{day.day}日"
|
||
|
||
@classmethod
|
||
def _calculate_day_stats(cls, site_data: dict, date_label: str) -> dict:
|
||
"""
|
||
统计指定日期下各站点的成功、异常和未记录数量。
|
||
"""
|
||
stats = {
|
||
"total": len(site_data),
|
||
"recorded": 0,
|
||
"success": 0,
|
||
"warning": 0,
|
||
"error": 0,
|
||
"none": 0
|
||
}
|
||
for records in site_data.values():
|
||
record = cls._latest_record(records=records, date_label=date_label)
|
||
if not record:
|
||
stats["none"] += 1
|
||
continue
|
||
stats["recorded"] += 1
|
||
level = cls._status_meta(record.get("status", "")).get("level")
|
||
if level in stats:
|
||
stats[level] += 1
|
||
return stats
|
||
|
||
@staticmethod
|
||
def _build_stat_item(label: str, value: str, meta: str, color: str, icon: str) -> dict:
|
||
"""
|
||
构建顶部紧凑统计块。
|
||
"""
|
||
return {
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-stat app-card-shell app-card-colorful',
|
||
'style': f'--app-card-accent-rgb: var(--v-theme-{color});'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-stat__head'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VIcon',
|
||
'props': {
|
||
'icon': icon,
|
||
'size': 'x-small',
|
||
'color': color
|
||
}
|
||
},
|
||
{
|
||
'component': 'span',
|
||
'text': label
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': f'autosignin-stat__value text-{color}'
|
||
},
|
||
'text': value
|
||
},
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-stat__meta'
|
||
},
|
||
'text': meta
|
||
}
|
||
]
|
||
}
|
||
|
||
@classmethod
|
||
def _build_summary(cls, signin_stats: dict, login_stats: dict, days: int) -> dict:
|
||
"""
|
||
构建详情页顶部的签到和登录概要。
|
||
"""
|
||
signin_total = signin_stats.get("total") or 0
|
||
login_total = login_stats.get("total") or 0
|
||
signin_problem_count = (signin_stats.get("error") or 0) + (signin_stats.get("warning") or 0)
|
||
signin_missing_count = signin_stats.get("none") or 0
|
||
signin_color = "info"
|
||
if signin_total:
|
||
signin_color = "success" if not signin_problem_count and not signin_missing_count else "warning"
|
||
if signin_total and signin_stats.get("error"):
|
||
signin_color = "error"
|
||
|
||
return {
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-summary'
|
||
},
|
||
'content': [
|
||
cls._build_stat_item(
|
||
label="今日签到",
|
||
value=f"{signin_stats.get('success') or 0}/{signin_total}",
|
||
meta=f"异常 {signin_problem_count} · 未记录 {signin_missing_count}",
|
||
color=signin_color,
|
||
icon="mdi-calendar-check"
|
||
),
|
||
cls._build_stat_item(
|
||
label="异常重试",
|
||
value=str(signin_problem_count),
|
||
meta=f"失败 {signin_stats.get('error') or 0} · 重试 {signin_stats.get('warning') or 0}",
|
||
color="error" if signin_stats.get("error") else "warning",
|
||
icon="mdi-alert-circle-outline"
|
||
),
|
||
cls._build_stat_item(
|
||
label="今日登录",
|
||
value=f"{login_stats.get('success') or 0}/{login_total}",
|
||
meta=f"异常 {(login_stats.get('error') or 0) + (login_stats.get('warning') or 0)} · 未记录 {login_stats.get('none') or 0}",
|
||
color="success" if login_total and not login_stats.get("error") and not login_stats.get("warning") else "info",
|
||
icon="mdi-login-variant"
|
||
),
|
||
cls._build_stat_item(
|
||
label="历史范围",
|
||
value=f"{days}天",
|
||
meta="矩阵显示最近7天",
|
||
color="info",
|
||
icon="mdi-history"
|
||
)
|
||
]
|
||
}
|
||
|
||
@classmethod
|
||
def _build_status_section(cls, title: str, icon: str, site_data: dict, display_dates: list, empty_text: str) -> dict:
|
||
"""
|
||
构建签到或登录状态区块，使用按站点排列的紧凑矩阵展示最近状态。
|
||
"""
|
||
return {
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-section'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-section-head'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VIcon',
|
||
'props': {
|
||
'icon': icon,
|
||
'size': 'small',
|
||
'color': 'primary'
|
||
}
|
||
},
|
||
{
|
||
'component': 'span',
|
||
'props': {
|
||
'class': 'autosignin-section-title'
|
||
},
|
||
'text': title
|
||
},
|
||
{
|
||
'component': 'VSpacer'
|
||
},
|
||
{
|
||
'component': 'VChip',
|
||
'props': {
|
||
'size': 'x-small',
|
||
'variant': 'tonal',
|
||
'color': 'primary'
|
||
},
|
||
'text': f"{len(site_data)} 个站点"
|
||
}
|
||
]
|
||
},
|
||
cls._build_status_table(
|
||
site_data=site_data,
|
||
display_dates=display_dates,
|
||
empty_text=empty_text
|
||
)
|
||
]
|
||
}
|
||
|
||
@classmethod
|
||
def _build_status_table(cls, site_data: dict, display_dates: list, empty_text: str) -> dict:
|
||
"""
|
||
构建按站点和日期交叉展示的状态表格。
|
||
"""
|
||
if not site_data:
|
||
return {
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'text': empty_text,
|
||
'variant': 'tonal',
|
||
'density': 'compact',
|
||
'prepend-icon': 'mdi-information'
|
||
}
|
||
}
|
||
|
||
table_headers = [
|
||
{
|
||
'component': 'th',
|
||
'props': {
|
||
'class': 'text-start'
|
||
},
|
||
'text': '站点'
|
||
},
|
||
{
|
||
'component': 'th',
|
||
'props': {
|
||
'class': 'text-start'
|
||
},
|
||
'text': '今日'
|
||
}
|
||
]
|
||
for day in display_dates:
|
||
table_headers.append({
|
||
'component': 'th',
|
||
'props': {
|
||
'class': 'text-center'
|
||
},
|
||
'text': f"{day.month}/{day.day}"
|
||
})
|
||
|
||
sorted_sites = sorted(
|
||
site_data.items(),
|
||
key=lambda item: cls._site_sort_key(site_name=item[0], records=item[1], display_dates=display_dates)
|
||
)
|
||
table_rows = []
|
||
for site_name, records in sorted_sites:
|
||
table_rows.append(cls._build_status_row(site_name=site_name, records=records, display_dates=display_dates))
|
||
|
||
return {
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-table-wrap'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTable',
|
||
'props': {
|
||
'hover': True,
|
||
'density': 'compact',
|
||
'class': 'autosignin-table'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'thead',
|
||
'content': [
|
||
{
|
||
'component': 'tr',
|
||
'content': table_headers
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'tbody',
|
||
'content': table_rows
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
|
||
@classmethod
|
||
def _site_sort_key(cls, site_name: str, records: list, display_dates: list) -> tuple:
|
||
"""
|
||
生成站点行排序键，让今日异常和未记录站点优先展示。
|
||
"""
|
||
today_label = cls._date_label(day=display_dates[0]) if display_dates else ""
|
||
today_record = cls._latest_record(records=records, date_label=today_label)
|
||
latest_record = cls._latest_record(records=records)
|
||
status_meta = cls._status_meta(today_record.get("status", "") if today_record else "")
|
||
latest_day = latest_record.get("day_obj", datetime.min.date()) if latest_record else datetime.min.date()
|
||
return status_meta.get("sort", 2), -latest_day.toordinal(), site_name
|
||
|
||
@classmethod
|
||
def _build_status_row(cls, site_name: str, records: list, display_dates: list) -> dict:
|
||
"""
|
||
构建单个站点在状态矩阵中的一行。
|
||
"""
|
||
today_label = cls._date_label(day=display_dates[0]) if display_dates else ""
|
||
today_record = cls._latest_record(records=records, date_label=today_label)
|
||
today_status = today_record.get("status", "") if today_record else ""
|
||
today_meta = cls._status_meta(today_status)
|
||
row_cells = [
|
||
{
|
||
'component': 'td',
|
||
'content': [
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-site-name',
|
||
'title': site_name
|
||
},
|
||
'text': site_name
|
||
},
|
||
{
|
||
'component': 'div',
|
||
'props': {
|
||
'class': 'autosignin-site-meta'
|
||
},
|
||
'text': f"{len(records)} 条记录" if records else "暂无记录"
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'td',
|
||
'props': {
|
||
'class': 'autosignin-status-cell'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VChip',
|
||
'props': {
|
||
'size': 'x-small',
|
||
'variant': 'tonal',
|
||
'color': today_meta.get("color"),
|
||
'prepend-icon': today_meta.get("icon")
|
||
},
|
||
'text': today_meta.get("label")
|
||
}
|
||
]
|
||
}
|
||
]
|
||
for day in display_dates:
|
||
date_label = cls._date_label(day=day)
|
||
record = cls._latest_record(records=records, date_label=date_label)
|
||
row_cells.append({
|
||
'component': 'td',
|
||
'props': {
|
||
'class': 'autosignin-dot-cell'
|
||
},
|
||
'content': [
|
||
cls._build_status_dot(record=record, date_label=date_label)
|
||
]
|
||
})
|
||
return {
|
||
'component': 'tr',
|
||
'content': row_cells
|
||
}
|
||
|
||
@classmethod
|
||
def _build_status_dot(cls, record: dict, date_label: str) -> dict:
|
||
"""
|
||
构建矩阵中单日状态的图标点。
|
||
"""
|
||
status_text = record.get("status", "") if record else ""
|
||
status_meta = cls._status_meta(status_text)
|
||
return {
|
||
'component': 'span',
|
||
'props': {
|
||
'class': f"autosignin-dot autosignin-dot--{status_meta.get('level')}",
|
||
'title': f"{date_label} {status_meta.get('label')}"
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VIcon',
|
||
'props': {
|
||
'icon': status_meta.get("icon"),
|
||
'size': 'x-small'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
|
||
@eventmanager.register(EventType.PluginAction)
|
||
def sign_in(self, event: Event = None):
|
||
"""
|
||
自动签到|模拟登录
|
||
"""
|
||
if event:
|
||
event_data = event.event_data
|
||
if not event_data or event_data.get("action") != "site_signin":
|
||
return
|
||
# 日期
|
||
today = datetime.today()
|
||
if self._start_time and self._end_time:
|
||
if int(datetime.today().hour) < self._start_time or int(datetime.today().hour) > self._end_time:
|
||
logger.error(
|
||
f"当前时间 {int(datetime.today().hour)} 不在 {self._start_time}-{self._end_time} 范围内，暂不执行任务")
|
||
return
|
||
if event:
|
||
logger.info("收到命令，开始站点签到 ...")
|
||
self.post_message(channel=event.event_data.get("channel"),
|
||
title="开始站点签到 ...",
|
||
userid=event.event_data.get("user"))
|
||
|
||
if self._sign_sites:
|
||
self.__do(today=today, type_str="签到", do_sites=self._sign_sites, event=event)
|
||
if self._login_sites:
|
||
self.__do(today=today, type_str="登录", do_sites=self._login_sites, event=event)
|
||
|
||
def __do(self, today: datetime, type_str: str, do_sites: list, event: Event = None):
|
||
"""
|
||
签到逻辑
|
||
"""
|
||
yesterday = today - timedelta(days=1)
|
||
yesterday_str = yesterday.strftime('%Y-%m-%d')
|
||
# 删除昨天历史
|
||
self.del_data(key=type_str + "-" + yesterday_str)
|
||
self.del_data(key=f"{yesterday.month}月{yesterday.day}日")
|
||
|
||
# 查看今天有没有签到|登录历史
|
||
today = today.strftime('%Y-%m-%d')
|
||
today_history = self.get_data(key=type_str + "-" + today)
|
||
|
||
# 查询所有站点
|
||
all_sites = [site for site in SitesHelper().get_indexers() if not site.get("public")] + self.__custom_sites()
|
||
# 过滤掉没有选中的站点
|
||
if do_sites:
|
||
do_sites = [site for site in all_sites if site.get("id") in do_sites]
|
||
else:
|
||
do_sites = all_sites
|
||
|
||
# 今日没数据
|
||
if not today_history or self._clean:
|
||
logger.info(f"今日 {today} 未{type_str}，开始{type_str}已选站点")
|
||
if self._clean:
|
||
# 关闭开关
|
||
self._clean = False
|
||
else:
|
||
# 需要重试站点
|
||
retry_sites = today_history.get("retry") or []
|
||
# 今天已签到|登录站点
|
||
already_sites = today_history.get("do") or []
|
||
|
||
# 今日未签|登录站点
|
||
no_sites = [site for site in do_sites if
|
||
site.get("id") not in already_sites or site.get("id") in retry_sites]
|
||
|
||
if not no_sites:
|
||
logger.info(f"今日 {today} 已{type_str}，无重新{type_str}站点，本次任务结束")
|
||
return
|
||
|
||
# 任务站点 = 需要重试+今日未do
|
||
do_sites = no_sites
|
||
logger.info(f"今日 {today} 已{type_str}，开始重试命中关键词站点")
|
||
|
||
if not do_sites:
|
||
logger.info(f"没有需要{type_str}的站点")
|
||
return
|
||
|
||
# 执行签到
|
||
logger.info(f"开始执行{type_str}任务 ...")
|
||
if type_str == "签到":
|
||
with ThreadPool(min(len(do_sites), int(self._queue_cnt))) as p:
|
||
status = p.map(self.signin_site, do_sites)
|
||
else:
|
||
with ThreadPool(min(len(do_sites), int(self._queue_cnt))) as p:
|
||
status = p.map(self.login_site, do_sites)
|
||
|
||
if status:
|
||
logger.info(f"站点{type_str}任务完成！")
|
||
# 获取今天的日期
|
||
key = f"{datetime.now().month}月{datetime.now().day}日"
|
||
today_data = self.get_data(key)
|
||
if today_data:
|
||
if not isinstance(today_data, list):
|
||
today_data = [today_data]
|
||
for s in status:
|
||
today_data.append({
|
||
"site": s[0],
|
||
"status": s[1]
|
||
})
|
||
else:
|
||
today_data = [{
|
||
"site": s[0],
|
||
"status": s[1]
|
||
} for s in status]
|
||
# 保存数据
|
||
self.save_data(key, today_data)
|
||
|
||
# 命中重试词的站点id
|
||
retry_sites = []
|
||
# 命中重试词的站点签到msg
|
||
retry_msg = []
|
||
# 登录成功
|
||
login_success_msg = []
|
||
# 签到成功
|
||
sign_success_msg = []
|
||
# 已签到
|
||
already_sign_msg = []
|
||
# 仿真签到成功
|
||
fz_sign_msg = []
|
||
# 失败｜错误
|
||
failed_msg = []
|
||
|
||
sites = {site.get('name'): site.get("id") for site in SitesHelper().get_indexers() if
|
||
not site.get("public")}
|
||
for s in status:
|
||
site_name = s[0]
|
||
site_id = None
|
||
if site_name:
|
||
site_id = sites.get(site_name)
|
||
|
||
if 'Cookie已失效' in str(s) and site_id:
|
||
# 触发自动登录插件登录
|
||
logger.info(f"触发站点 {site_name} 自动登录更新Cookie和Ua")
|
||
self.eventmanager.send_event(EventType.PluginAction,
|
||
{
|
||
"site_id": site_id,
|
||
"action": "site_refresh"
|
||
})
|
||
# 记录本次命中重试关键词的站点
|
||
if self._retry_keyword:
|
||
if site_id:
|
||
match = re.search(self._retry_keyword, s[1])
|
||
if match:
|
||
logger.debug(f"站点 {site_name} 命中重试关键词 {self._retry_keyword}")
|
||
retry_sites.append(site_id)
|
||
# 命中的站点
|
||
retry_msg.append(s)
|
||
continue
|
||
|
||
if "登录成功" in str(s):
|
||
login_success_msg.append(s)
|
||
elif "仿真签到成功" in str(s):
|
||
fz_sign_msg.append(s)
|
||
continue
|
||
elif "签到成功" in str(s):
|
||
sign_success_msg.append(s)
|
||
elif '已签到' in str(s):
|
||
already_sign_msg.append(s)
|
||
else:
|
||
failed_msg.append(s)
|
||
|
||
if not self._retry_keyword:
|
||
# 没设置重试关键词则重试已选站点
|
||
retry_sites = self._sign_sites if type_str == "签到" else self._login_sites
|
||
logger.debug(f"下次{type_str}重试站点 {retry_sites}")
|
||
|
||
# 存入历史
|
||
self.save_data(key=type_str + "-" + today,
|
||
value={
|
||
"do": self._sign_sites if type_str == "签到" else self._login_sites,
|
||
"retry": retry_sites
|
||
})
|
||
|
||
# 自动Cloudflare IP优选
|
||
if self._auto_cf and int(self._auto_cf) > 0 and retry_msg and len(retry_msg) >= int(self._auto_cf):
|
||
self.eventmanager.send_event(EventType.PluginAction, {
|
||
"action": "cloudflare_speedtest"
|
||
})
|
||
|
||
# 发送通知
|
||
if self._notify:
|
||
# 签到详细信息 登录成功、签到成功、已签到、仿真签到成功、失败--命中重试
|
||
signin_message = login_success_msg + sign_success_msg + already_sign_msg + fz_sign_msg + failed_msg
|
||
if len(retry_msg) > 0:
|
||
signin_message += retry_msg
|
||
|
||
signin_message = "\n".join([f'【{s[0]}】{s[1]}' for s in signin_message if s])
|
||
self.post_message(title=f"【站点自动{type_str}】",
|
||
mtype=NotificationType.SiteMessage,
|
||
text=f"全部{type_str}数量: {len(self._sign_sites if type_str == '签到' else self._login_sites)} \n"
|
||
f"本次{type_str}数量: {len(do_sites)} \n"
|
||
f"下次{type_str}数量: {len(retry_sites) if self._retry_keyword else 0} \n"
|
||
f"{signin_message}"
|
||
)
|
||
if event:
|
||
self.post_message(channel=event.event_data.get("channel"),
|
||
title=f"站点{type_str}完成！", userid=event.event_data.get("user"))
|
||
else:
|
||
logger.error(f"站点{type_str}任务失败！")
|
||
if event:
|
||
self.post_message(channel=event.event_data.get("channel"),
|
||
title=f"站点{type_str}任务失败！", userid=event.event_data.get("user"))
|
||
# 保存配置
|
||
self.__update_config()
|
||
|
||
def __build_class(self, url) -> Any:
|
||
for site_schema in self._site_schema:
|
||
try:
|
||
if site_schema.match(url):
|
||
return site_schema
|
||
except Exception as e:
|
||
logger.error("站点模块加载失败：%s" % str(e))
|
||
return None
|
||
|
||
def signin_by_domain(self, url: str, apikey: str) -> schemas.Response:
|
||
"""
|
||
签到一个站点，可由API调用
|
||
"""
|
||
# 校验
|
||
if apikey != settings.API_TOKEN:
|
||
return schemas.Response(success=False, message="API密钥错误")
|
||
domain = StringUtils.get_url_domain(url)
|
||
site_info = SitesHelper().get_indexer(domain)
|
||
if not site_info:
|
||
return schemas.Response(
|
||
success=True,
|
||
message=f"站点【{url}】不存在"
|
||
)
|
||
else:
|
||
site_name, message = self.signin_site(site_info)
|
||
return schemas.Response(
|
||
success=True,
|
||
message=f"站点【{site_name}】{message or '签到成功'}"
|
||
)
|
||
|
||
def signin_site(self, site_info: CommentedMap) -> Tuple[str, str]:
|
||
"""
|
||
签到一个站点
|
||
"""
|
||
site_module = self.__build_class(site_info.get("url"))
|
||
# 开始记时
|
||
start_time = datetime.now()
|
||
if site_module and hasattr(site_module, "signin"):
|
||
try:
|
||
state, message = site_module().signin(site_info)
|
||
except Exception as e:
|
||
traceback.print_exc()
|
||
state, message = False, f"签到失败：{str(e)}"
|
||
else:
|
||
state, message = self.__signin_base(site_info)
|
||
# 统计
|
||
seconds = (datetime.now() - start_time).seconds
|
||
domain = StringUtils.get_url_domain(site_info.get('url'))
|
||
if state:
|
||
SiteOper().success(domain=domain, seconds=seconds)
|
||
else:
|
||
SiteOper().fail(domain)
|
||
return site_info.get("name"), message
|
||
|
||
@staticmethod
|
||
def __signin_base(site_info: CommentedMap) -> Tuple[bool, str]:
|
||
"""
|
||
通用签到处理
|
||
:param site_info: 站点信息
|
||
:return: 签到结果信息
|
||
"""
|
||
if not site_info:
|
||
return False, ""
|
||
site = site_info.get("name")
|
||
site_url = site_info.get("url")
|
||
site_cookie = site_info.get("cookie")
|
||
ua = site_info.get("ua")
|
||
render = site_info.get("render")
|
||
proxies = settings.PROXY if site_info.get("proxy") else None
|
||
proxy_server = settings.PROXY_SERVER if site_info.get("proxy") else None
|
||
timeout = site_info.get("timeout") or 60
|
||
if not site_url or not site_cookie:
|
||
logger.warn(f"未配置 {site} 的站点地址或Cookie，无法签到")
|
||
return False, ""
|
||
# 模拟登录
|
||
try:
|
||
# 访问链接
|
||
checkin_url = site_url
|
||
if site_url.find("attendance.php") == -1:
|
||
# 拼登签到地址
|
||
checkin_url = urljoin(site_url, "attendance.php")
|
||
logger.info(f"开始站点签到：{site}，地址：{checkin_url}...")
|
||
if render:
|
||
page_source = PlaywrightHelper().get_page_source(url=checkin_url,
|
||
cookies=site_cookie,
|
||
ua=ua,
|
||
proxies=proxy_server,
|
||
timeout=timeout)
|
||
if not SiteUtils.is_logged_in(page_source):
|
||
if under_challenge(page_source):
|
||
return False, f"无法通过Cloudflare！"
|
||
return False, f"仿真登录失败，Cookie已失效！"
|
||
else:
|
||
# 判断是否已签到
|
||
if re.search(r'已签|签到已得', page_source, re.IGNORECASE) \
|
||
or SiteUtils.is_checkin(page_source):
|
||
return True, f"签到成功"
|
||
return True, "仿真签到成功"
|
||
else:
|
||
res = RequestUtils(cookies=site_cookie,
|
||
ua=ua,
|
||
proxies=proxies,
|
||
timeout=timeout
|
||
).get_res(url=checkin_url)
|
||
if not res and site_url != checkin_url:
|
||
logger.info(f"开始站点模拟登录：{site}，地址：{site_url}...")
|
||
res = RequestUtils(cookies=site_cookie,
|
||
ua=ua,
|
||
proxies=proxies,
|
||
timeout=timeout
|
||
).get_res(url=site_url)
|
||
# 判断登录状态
|
||
if res and res.status_code in [200, 500, 403]:
|
||
if not SiteUtils.is_logged_in(res.text):
|
||
if under_challenge(res.text):
|
||
msg = "站点被Cloudflare防护，请打开站点浏览器仿真"
|
||
elif res.status_code == 200:
|
||
msg = "Cookie已失效"
|
||
else:
|
||
msg = f"状态码：{res.status_code}"
|
||
logger.warn(f"{site} 签到失败，{msg}")
|
||
return False, f"签到失败，{msg}！"
|
||
else:
|
||
logger.info(f"{site} 签到成功")
|
||
return True, f"签到成功"
|
||
elif res is not None:
|
||
logger.warn(f"{site} 签到失败，状态码：{res.status_code}")
|
||
return False, f"签到失败，状态码：{res.status_code}！"
|
||
else:
|
||
logger.warn(f"{site} 签到失败，无法打开网站")
|
||
return False, f"签到失败，无法打开网站！"
|
||
except Exception as e:
|
||
logger.warn("%s 签到失败：%s" % (site, str(e)))
|
||
traceback.print_exc()
|
||
return False, f"签到失败：{str(e)}！"
|
||
|
||
def login_site(self, site_info: CommentedMap) -> Tuple[str, str]:
|
||
"""
|
||
模拟登录一个站点
|
||
"""
|
||
site_module = self.__build_class(site_info.get("url"))
|
||
# 开始记时
|
||
start_time = datetime.now()
|
||
if site_module and hasattr(site_module, "login"):
|
||
try:
|
||
state, message = site_module().login(site_info)
|
||
except Exception as e:
|
||
traceback.print_exc()
|
||
state, message = False, f"模拟登录失败：{str(e)}"
|
||
else:
|
||
state, message = self.__login_base(site_info)
|
||
# 统计
|
||
seconds = (datetime.now() - start_time).seconds
|
||
domain = StringUtils.get_url_domain(site_info.get('url'))
|
||
if state:
|
||
SiteOper().success(domain=domain, seconds=seconds)
|
||
else:
|
||
SiteOper().fail(domain)
|
||
return site_info.get("name"), message
|
||
|
||
@staticmethod
|
||
def __login_base(site_info: CommentedMap) -> Tuple[bool, str]:
|
||
"""
|
||
模拟登录通用处理
|
||
:param site_info: 站点信息
|
||
:return: 签到结果信息
|
||
"""
|
||
if not site_info:
|
||
return False, ""
|
||
site = site_info.get("name")
|
||
site_url = site_info.get("url")
|
||
site_cookie = site_info.get("cookie")
|
||
ua = site_info.get("ua")
|
||
render = site_info.get("render")
|
||
proxies = settings.PROXY if site_info.get("proxy") else None
|
||
proxy_server = settings.PROXY_SERVER if site_info.get("proxy") else None
|
||
timeout = site_info.get("timeout") or 60
|
||
if not site_url or not site_cookie:
|
||
logger.warn(f"未配置 {site} 的站点地址或Cookie，无法签到")
|
||
return False, ""
|
||
# 模拟登录
|
||
try:
|
||
# 访问链接
|
||
site_url = str(site_url).replace("attendance.php", "")
|
||
logger.info(f"开始站点模拟登录：{site}，地址：{site_url}...")
|
||
if render:
|
||
page_source = PlaywrightHelper().get_page_source(url=site_url,
|
||
cookies=site_cookie,
|
||
ua=ua,
|
||
proxies=proxy_server,
|
||
timeout=timeout)
|
||
if not SiteUtils.is_logged_in(page_source):
|
||
if under_challenge(page_source):
|
||
return False, f"无法通过Cloudflare！"
|
||
return False, f"仿真登录失败，Cookie已失效！"
|
||
else:
|
||
return True, "模拟登录成功"
|
||
else:
|
||
res = RequestUtils(cookies=site_cookie,
|
||
ua=ua,
|
||
proxies=proxies,
|
||
timeout=timeout
|
||
).get_res(url=site_url)
|
||
# 判断登录状态
|
||
if res and res.status_code in [200, 500, 403]:
|
||
if not SiteUtils.is_logged_in(res.text):
|
||
if under_challenge(res.text):
|
||
msg = "站点被Cloudflare防护，请打开站点浏览器仿真"
|
||
elif res.status_code == 200:
|
||
msg = "Cookie已失效"
|
||
else:
|
||
msg = f"状态码：{res.status_code}"
|
||
logger.warn(f"{site} 模拟登录失败，{msg}")
|
||
return False, f"模拟登录失败，{msg}！"
|
||
else:
|
||
logger.info(f"{site} 模拟登录成功")
|
||
return True, f"模拟登录成功"
|
||
elif res is not None:
|
||
logger.warn(f"{site} 模拟登录失败，状态码：{res.status_code}")
|
||
return False, f"模拟登录失败，状态码：{res.status_code}！"
|
||
else:
|
||
logger.warn(f"{site} 模拟登录失败，无法打开网站")
|
||
return False, f"模拟登录失败，无法打开网站！"
|
||
except Exception as e:
|
||
logger.warn("%s 模拟登录失败：%s" % (site, str(e)))
|
||
traceback.print_exc()
|
||
return False, f"模拟登录失败：{str(e)}！"
|
||
|
||
def stop_service(self):
|
||
"""
|
||
退出插件
|
||
"""
|
||
try:
|
||
if self._scheduler:
|
||
self._scheduler.remove_all_jobs()
|
||
if self._scheduler.running:
|
||
self._scheduler.shutdown()
|
||
self._scheduler = None
|
||
except Exception as e:
|
||
logger.error("退出插件失败：%s" % str(e))
|
||
|
||
@eventmanager.register(EventType.SiteDeleted)
|
||
def site_deleted(self, event):
|
||
"""
|
||
删除对应站点选中
|
||
"""
|
||
site_id = event.event_data.get("site_id")
|
||
config = self.get_config()
|
||
if config:
|
||
self._sign_sites = self.__remove_site_id(config.get("sign_sites") or [], site_id)
|
||
self._login_sites = self.__remove_site_id(config.get("login_sites") or [], site_id)
|
||
# 保存配置
|
||
self.__update_config()
|
||
|
||
def __remove_site_id(self, do_sites, site_id):
|
||
if do_sites:
|
||
if isinstance(do_sites, str):
|
||
do_sites = [do_sites]
|
||
|
||
# 删除对应站点
|
||
if site_id:
|
||
do_sites = [site for site in do_sites if int(site) != int(site_id)]
|
||
else:
|
||
# 清空
|
||
do_sites = []
|
||
|
||
# 若无站点，则停止
|
||
if len(do_sites) == 0:
|
||
self._enabled = False
|
||
|
||
return do_sites
|
||
|
||
|
||
def record_to_row(record):
|
||
"""辅助函数：将记录转换为表格行"""
|
||
status = record.get("status", "")
|
||
|
||
# 确定状态图标和颜色
|
||
icon = "mdi-check-circle"
|
||
color = "success"
|
||
|
||
if "失败" in status or "错误" in status:
|
||
icon = "mdi-alert-circle"
|
||
color = "error"
|
||
elif "Cookie已失效" in status:
|
||
icon = "mdi-cookie-off"
|
||
color = "error"
|
||
elif "已签到" in status:
|
||
icon = "mdi-check"
|
||
color = "grey"
|
||
elif "成功" in status:
|
||
icon = "mdi-check-circle"
|
||
color = "success"
|
||
|
||
return {
|
||
'component': 'tr',
|
||
'props': {
|
||
'class': 'text-sm'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'td',
|
||
'props': {
|
||
'class': 'text-start'
|
||
},
|
||
'text': record.get("date", "")
|
||
},
|
||
{
|
||
'component': 'td',
|
||
'props': {
|
||
'class': 'text-start'
|
||
},
|
||
'text': status
|
||
},
|
||
{
|
||
'component': 'td',
|
||
'props': {
|
||
'class': 'text-center'
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VIcon',
|
||
'props': {
|
||
'color': color,
|
||
'size': 'small'
|
||
},
|
||
'text': icon
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|