codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #503 from thsrite/main

Browse Source 
feat 签到插件支持分别配置签到、登录站点
This commit is contained in:
jxxghp 2023-09-08 12:26:15 +08:00 committed by GitHub
commit d1f9647a63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 95 additions and 83 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

178
app/plugins/autosignin/__init__.py
View File

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