diff --git a/app/plugins/autosignin/__init__.py b/app/plugins/autosignin/__init__.py index be8ca22f7..e66c20c66 100644 --- a/app/plugins/autosignin/__init__.py +++ b/app/plugins/autosignin/__init__.py @@ -31,13 +31,13 @@ class AutoSignIn(_PluginBase): # 插件名称 plugin_name = "站点自动签到" # 插件描述 - plugin_desc = "自动模拟登录站点或签到。" + plugin_desc = "自动模拟登录站点、签到。" # 插件图标 plugin_icon = "signin.png" # 主题色 plugin_color = "#4179F4" # 插件版本 - plugin_version = "1.0" + plugin_version = "1.1" # 插件作者 plugin_author = "thsrite" # 作者主页 @@ -61,16 +61,15 @@ class AutoSignIn(_PluginBase): # 配置属性 _enabled: bool = False _cron: str = "" - _sign_type: 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 - _action: str = "" def init_plugin(self, config: dict = None): self.sites = SitesHelper() @@ -87,10 +86,9 @@ def init_plugin(self, config: dict = None): self._notify = config.get("notify") self._queue_cnt = config.get("queue_cnt") or 5 self._sign_sites = config.get("sign_sites") + self._login_sites = config.get("login_sites") self._retry_keyword = config.get("retry_keyword") self._clean = config.get("clean") - self._sign_type = config.get("sign_type") or "sign" - self._action = "签到" if str(self._sign_type) == "sign" else "模拟登陆" # 加载模块 if self._enabled or self._onlyonce: @@ -103,10 +101,10 @@ def init_plugin(self, config: dict = None): # 立即运行一次 if self._onlyonce: - logger.info(f"站点自动{self._action}服务启动，立即运行一次") + 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=f"站点自动{self._action}") + name="站点自动签到") # 关闭一次性开关 self._onlyonce = False @@ -120,8 +118,8 @@ def init_plugin(self, config: dict = None): if str(self._cron).strip().count(" ") == 4: self._scheduler.add_job(func=self.sign_in, trigger=CronTrigger.from_crontab(self._cron), - name=f"站点自动{self._action}") - logger.info(f"站点自动{self._action}服务启动，执行周期 {self._cron}") + name="站点自动签到") + logger.info(f"站点自动签到服务启动，执行周期 {self._cron}") else: # 2.3/9-23 crons = str(self._cron).strip().split("/") @@ -139,11 +137,11 @@ def init_plugin(self, config: dict = None): self._scheduler.add_job(func=self.sign_in, trigger="interval", hours=float(str(cron).strip()), - name=f"站点自动{self._action}") + name="站点自动签到") logger.info( - f"站点自动{self._action}服务启动，执行周期 {self._start_time}点-{self._end_time}点 每{cron}小时执行一次") + f"站点自动签到服务启动，执行周期 {self._start_time}点-{self._end_time}点 每{cron}小时执行一次") else: - logger.error(f"站点自动{self._action}服务启动失败，周期格式错误") + logger.error("站点自动签到服务启动失败，周期格式错误") # 推送实时消息 self.systemmessage.put(f"执行周期配置错误") self._cron = "" @@ -156,9 +154,9 @@ def init_plugin(self, config: dict = None): self._scheduler.add_job(func=self.sign_in, trigger="interval", hours=float(str(self._cron).strip()), - name=f"站点自动{self._action}") + name="站点自动签到") logger.info( - f"站点自动{self._action}服务启动，执行周期 {self._start_time}点-{self._end_time}点 每{self._cron}小时执行一次") + f"站点自动签到服务启动，执行周期 {self._start_time}点-{self._end_time}点 每{self._cron}小时执行一次") except Exception as err: logger.error(f"定时任务配置错误：{err}") # 推送实时消息 @@ -176,7 +174,7 @@ def init_plugin(self, config: dict = None): for trigger in triggers: self._scheduler.add_job(self.sign_in, "cron", hour=trigger.hour, minute=trigger.minute, - name=f"站点自动{self._action}") + name="站点自动签到") # 启动任务 if self._scheduler.get_jobs(): @@ -196,9 +194,9 @@ def __update_config(self): "onlyonce": self._onlyonce, "queue_cnt": self._queue_cnt, "sign_sites": self._sign_sites, + "login_sites": self._login_sites, "retry_keyword": self._retry_keyword, "clean": self._clean, - "sign_type": self._sign_type, } ) @@ -307,7 +305,7 @@ def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: 'component': 'VSwitch', 'props': { 'model': 'clean', - 'label': '清理本日已签到', + 'label': '清理本日缓存', } } ] @@ -321,27 +319,7 @@ def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: 'component': 'VCol', 'props': { 'cols': 12, - 'md': 6 - }, - 'content': [ - { - 'component': 'VSelect', - 'props': { - 'model': 'sign_type', - 'label': '签到方式', - 'items': [ - {'title': '签到', 'value': 'sign'}, - {'title': '登录', 'value': 'login'} - ] - } - } - ] - }, - { - 'component': 'VCol', - 'props': { - 'cols': 12, - 'md': 6 + 'md': 4 }, 'content': [ { @@ -358,7 +336,7 @@ def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: 'component': 'VCol', 'props': { 'cols': 12, - 'md': 6 + 'md': 4 }, 'content': [ { @@ -374,7 +352,7 @@ def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: 'component': 'VCol', 'props': { 'cols': 12, - 'md': 6 + 'md': 4 }, 'content': [ { @@ -409,6 +387,26 @@ def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: } ] }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'content': [ + { + 'component': 'VSelect', + 'props': { + 'chips': True, + 'multiple': True, + 'model': 'login_sites', + 'label': '登录站点', + 'items': site_options + } + } + ] + } + ] + }, { 'component': 'VRow', 'content': [ @@ -437,12 +435,12 @@ def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: ], { "enabled": False, "notify": True, - "sign_type": "sign", "cron": "", "onlyonce": False, "clean": False, "queue_cnt": 5, "sign_sites": [], + "login_sites": [], "retry_keyword": "错误|失败" } @@ -556,70 +554,84 @@ def sign_in(self, event: Event = None): 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} 范围内，暂不{self._action}") + f"当前时间 {int(datetime.today().hour)} 不在 {self._start_time}-{self._end_time} 范围内，暂不执行任务") return if event: - logger.info(f"收到命令，开始站点{self._action} ...") + logger.info("收到命令，开始站点签到 ...") self.post_message(channel=event.event_data.get("channel"), - title=f"开始站点{self._action} ...", + title="开始站点签到 ...", userid=event.event_data.get("user")) + if self._sign_sites: + self.__do(today=today, type="签到", do_sites=self._sign_sites, event=event) + if self._login_sites: + self.__do(today=today, type="登录", do_sites=self._login_sites, event=event) + + def __do(self, today: datetime, type: str, do_sites: list, event: Event = None): + """ + 签到逻辑 + """ yesterday = today - timedelta(days=1) yesterday_str = yesterday.strftime('%Y-%m-%d') # 删除昨天历史 - self.del_data(key=yesterday_str) + self.del_data(key=type + "-" + yesterday_str) - # 查看今天有没有签到历史 + # 查看今天有没有签到|登录历史 today = today.strftime('%Y-%m-%d') - today_history = self.get_data(key=today) + today_history = self.get_data(key=type + "-" + today) - # 查询签到站点 - sign_sites = [site for site in self.sites.get_indexers() if not site.get("public")] + # 查询所有站点 + all_sites = [site for site in self.sites.get_indexers() if not site.get("public")] # 过滤掉没有选中的站点 - if self._sign_sites: - sign_sites = [site for site in sign_sites if site.get("id") in self._sign_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} 未{self._action}，开始{self._action}已选站点") + logger.info(f"今日 {today} 未{type}，开始{type}已选站点") # 过滤删除的站点 - self._sign_sites = [site.get("id") for site in sign_sites if site] + if type == "签到": + self._sign_sites = [site.get("id") for site in do_sites if site] + if type == "登录": + self._login_sites = [site.get("id") for site in do_sites if site] if self._clean: # 关闭开关 self._clean = False else: - # 今天已签到需要重签站点 + # 需要重试站点 retry_sites = today_history.get("retry") - # 今天已签到站点 - already_sign_sites = today_history.get("sign") + # 今天已签到|登录站点 + already_sites = today_history.get("sign") - # 今日未签站点 - no_sign_sites = [site for site in sign_sites if - site.get("id") not in already_sign_sites or site.get("id") in retry_sites] + # 今日未签|登录站点 + 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_sign_sites: - logger.info(f"今日 {today} 已{self._action}，无重新{self._action}站点，本次任务结束") + if not no_sites: + logger.info(f"今日 {today} 已{type}，无重新{type}站点，本次任务结束") return - # 签到站点 = 需要重签+今日未签 - sign_sites = no_sign_sites - logger.info(f"今日 {today} 已{self._action}，开始重试命中关键词站点") + # 任务站点 = 需要重试+今日未do + do_sites = no_sites + logger.info(f"今日 {today} 已{type}，开始重试命中关键词站点") - if not sign_sites: - logger.info(f"没有需要{self._action}的站点") + if not do_sites: + logger.info(f"没有需要{type}的站点") return # 执行签到 - logger.info(f"开始执行{self._action}任务 ...") - if str(self._sign_type) == "sign": - with ThreadPool(min(len(sign_sites), int(self._queue_cnt))) as p: - status = p.map(self.signin_site, sign_sites) + logger.info(f"开始执行{type}任务 ...") + if type == "签到": + 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(sign_sites), int(self._queue_cnt))) as p: - status = p.map(self.login_site, sign_sites) + 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"站点{self._action}任务完成！") + logger.info(f"站点{type}任务完成！") # 获取今天的日期 key = f"{datetime.now().month}月{datetime.now().day}日" # 保存数据 @@ -675,10 +687,10 @@ def sign_in(self, event: Event = None): if not self._retry_keyword: # 没设置重试关键词则重试已选站点 retry_sites = self._sign_sites - logger.debug(f"下次{self._action}重试站点 {retry_sites}") + logger.debug(f"下次{type}重试站点 {retry_sites}") # 存入历史 - self.save_data(key=today, + self.save_data(key=type + "-" + today, value={ "sign": self._sign_sites, "retry": retry_sites @@ -692,21 +704,21 @@ def sign_in(self, event: Event = None): 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"站点自动{self._action}", + self.post_message(title=f"站点自动{type}", mtype=NotificationType.SiteMessage, - text=f"全部{self._action}数量: {len(list(self._sign_sites))} \n" - f"本次{self._action}数量: {len(sign_sites)} \n" - f"下次{self._action}数量: {len(retry_sites) if self._retry_keyword else 0} \n" + text=f"全部{type}数量: {len(list(self._sign_sites))} \n" + f"本次{type}数量: {len(do_sites)} \n" + f"下次{type}数量: {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"站点{self._action}完成！", userid=event.event_data.get("user")) + title=f"站点{type}完成！", userid=event.event_data.get("user")) else: - logger.error(f"站点{self._action}任务失败！") + logger.error(f"站点{type}任务失败！") if event: self.post_message(channel=event.event_data.get("channel"), - title=f"站点{self._action}任务失败！", userid=event.event_data.get("user")) + title=f"站点{type}任务失败！", userid=event.event_data.get("user")) # 保存配置 self.__update_config()