From f19f779538b6bb4d6afa9d91ba440583fa70b00a Mon Sep 17 00:00:00 2001 From: thsrite Date: Sun, 6 Jul 2025 13:02:46 +0800 Subject: [PATCH] =?UTF-8?q?fix=20=E5=B0=9D=E8=AF=95=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?=E8=8D=AF=E4=B8=B8=E7=AD=BE=E5=88=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- package.json | 15 + plugins/invitessignin/__init__.py | 466 ++++++++++++++++++++++++++++++ 2 files changed, 481 insertions(+) create mode 100644 plugins/invitessignin/__init__.py diff --git a/package.json b/package.json index acd1e67..442434d 100644 --- a/package.json +++ b/package.json @@ -788,5 +788,20 @@ "v1.0.1": "支持Lucky最新版本获取ipv4。", "v1.0.0": "Lucky HomePage自定义API。" } + }, + "InvitesSignin": { + "name": "药丸签到", + "description": "药丸论坛签到。", + "labels": "站点", + "version": "1.5", + "icon": "invites.png", + "author": "thsrite", + "level": 2, + "v2": true, + "history": { + "v1.5": "尝试修复签到", + "v1.4.1": "更新签到域名前缀", + "v1.4": "自定义保留消息天数" + } } } diff --git a/plugins/invitessignin/__init__.py b/plugins/invitessignin/__init__.py new file mode 100644 index 0000000..3277c06 --- /dev/null +++ b/plugins/invitessignin/__init__.py @@ -0,0 +1,466 @@ +import json +import re +import time +from datetime import datetime, timedelta + +import pytz +from apscheduler.schedulers.background import BackgroundScheduler +from apscheduler.triggers.cron import CronTrigger + +from app.core.config import settings +from app.plugins import _PluginBase +from typing import Any, List, Dict, Tuple, Optional +from app.log import logger +from app.schemas import NotificationType +from app.utils.http import RequestUtils + + +class InvitesSignin(_PluginBase): + # 插件名称 + plugin_name = "药丸签到" + # 插件描述 + plugin_desc = "药丸论坛签到。" + # 插件图标 + plugin_icon = "invites.png" + # 插件版本 + plugin_version = "1.5" + # 插件作者 + plugin_author = "thsrite" + # 作者主页 + author_url = "https://github.com/thsrite" + # 插件配置项ID前缀 + plugin_config_prefix = "invitessignin_" + # 加载顺序 + plugin_order = 24 + # 可使用的用户级别 + auth_level = 2 + + # 私有属性 + _enabled = False + # 任务执行间隔 + _cron = None + _cookie = None + _onlyonce = False + _notify = False + _history_days = None + + # 定时器 + _scheduler: Optional[BackgroundScheduler] = None + + def init_plugin(self, config: dict = None): + # 停止现有任务 + self.stop_service() + + if config: + self._enabled = config.get("enabled") + self._cron = config.get("cron") + self._cookie = config.get("cookie") + self._notify = config.get("notify") + self._onlyonce = config.get("onlyonce") + self._history_days = config.get("history_days") or 30 + + if self._onlyonce: + # 定时服务 + self._scheduler = BackgroundScheduler(timezone=settings.TZ) + logger.info(f"药丸签到服务启动，立即运行一次") + self._scheduler.add_job(func=self.__signin, trigger='date', + run_date=datetime.now(tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3), + name="药丸签到") + # 关闭一次性开关 + self._onlyonce = False + self.update_config({ + "onlyonce": False, + "cron": self._cron, + "enabled": self._enabled, + "cookie": self._cookie, + "notify": self._notify, + "history_days": self._history_days, + }) + + # 启动任务 + if self._scheduler.get_jobs(): + self._scheduler.print_jobs() + self._scheduler.start() + + def __signin(self): + """ + 药丸签到 + """ + res = RequestUtils(cookies=self._cookie).get_res(url="https://invites.fun") + if not res or res.status_code != 200: + logger.error("请求药丸错误") + return + + # 获取csrfToken + pattern = r'"csrfToken":"(.*?)"' + csrfToken = re.findall(pattern, res.text) + if not csrfToken: + logger.error("请求csrfToken失败") + return + + csrfToken = csrfToken[0] + logger.info(f"获取csrfToken成功 {csrfToken}") + + # 获取userid + pattern = r'"userId":(\d+)' + match = re.search(pattern, res.text) + + if match: + userId = match.group(1) + logger.info(f"获取userid成功 {userId}") + else: + logger.error("未找到userId") + return + + headers = { + "X-Csrf-Token": csrfToken, + "X-Http-Method-Override": "PATCH", + "Cookie": self._cookie + } + + data = { + "data": { + "type": "users", + "attributes": { + "canCheckin": False, + "totalContinuousCheckIn": 2 + }, + "id": userId + } + } + + # 开始签到 + res = RequestUtils(headers=headers).post_res(url=f"https://invites.fun/api/users/{userId}", json=data) + + if not res or res.status_code != 200: + logger.error("药丸签到失败") + + # 发送通知 + if self._notify: + self.post_message( + mtype=NotificationType.SiteMessage, + title="【药丸签到任务完成】", + text="签到失败，请检查cookie是否失效") + return + + sign_dict = json.loads(res.text) + money = sign_dict['data']['attributes']['money'] + totalContinuousCheckIn = sign_dict['data']['attributes']['totalContinuousCheckIn'] + + # 发送通知 + if self._notify: + self.post_message( + mtype=NotificationType.SiteMessage, + title="【药丸签到任务完成】", + text=f"累计签到 {totalContinuousCheckIn} \n" + f"剩余药丸 {money}") + + # 读取历史记录 + history = self.get_data('history') or [] + + history.append({ + "date": datetime.today().strftime('%Y-%m-%d %H:%M:%S'), + "totalContinuousCheckIn": totalContinuousCheckIn, + "money": money + }) + + thirty_days_ago = time.time() - int(self._history_days) * 24 * 60 * 60 + history = [record for record in history if + datetime.strptime(record["date"], + '%Y-%m-%d %H:%M:%S').timestamp() >= thirty_days_ago] + # 保存签到历史 + self.save_data(key="history", value=history) + + def get_state(self) -> bool: + return self._enabled + + @staticmethod + def get_command() -> List[Dict[str, Any]]: + pass + + def get_api(self) -> List[Dict[str, Any]]: + pass + + 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: + return [{ + "id": "InvitesSignin", + "name": "药丸签到服务", + "trigger": CronTrigger.from_crontab(self._cron), + "func": self.__signin, + "kwargs": {} + }] + return [] + + def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: + """ + 拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构 + """ + return [ + { + 'component': 'VForm', + 'content': [ + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 4 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'enabled', + 'label': '启用插件', + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 4 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'notify', + 'label': '开启通知', + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 4 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'onlyonce', + 'label': '立即运行一次', + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 3 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'cron', + 'label': '签到周期' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 3 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'history_days', + 'label': '保留历史天数' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'cookie', + 'label': '药丸cookie' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VAlert', + 'props': { + 'type': 'info', + 'variant': 'tonal', + 'text': '整点定时签到失败？不妨换个时间试试。登录获取ck：https://invites.fun' + } + } + ] + } + ] + } + ] + } + ], { + "enabled": False, + "onlyonce": False, + "notify": False, + "cookie": "", + "history_days": 30, + "cron": "0 9 * * *" + } + + def get_page(self) -> List[dict]: + # 查询同步详情 + historys = self.get_data('history') + if not historys: + return [ + { + 'component': 'div', + 'text': '暂无数据', + 'props': { + 'class': 'text-center', + } + } + ] + + if not isinstance(historys, list): + historys = [historys] + + # 按照签到时间倒序 + historys = sorted(historys, key=lambda x: x.get("date") or 0, reverse=True) + + # 签到消息 + sign_msgs = [ + { + 'component': 'tr', + 'props': { + 'class': 'text-sm' + }, + 'content': [ + { + 'component': 'td', + 'props': { + 'class': 'whitespace-nowrap break-keep text-high-emphasis' + }, + 'text': history.get("date") + }, + { + 'component': 'td', + 'text': history.get("totalContinuousCheckIn") + }, + { + 'component': 'td', + 'text': history.get("money") + } + ] + } for history in historys + ] + + # 拼装页面 + return [ + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VTable', + 'props': { + 'hover': True + }, + 'content': [ + { + 'component': 'thead', + 'content': [ + { + 'component': 'th', + 'props': { + 'class': 'text-start ps-4' + }, + 'text': '时间' + }, + { + 'component': 'th', + 'props': { + 'class': 'text-start ps-4' + }, + 'text': '连续签到次数' + }, + { + 'component': 'th', + 'props': { + 'class': 'text-start ps-4' + }, + 'text': '剩余药丸' + }, + ] + }, + { + 'component': 'tbody', + 'content': sign_msgs + } + ] + } + ] + } + ] + } + ] + + pass + + 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))