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

Merge pull request #315 from thsrite/main

Browse Source 
fix 站点签到插件支持仅模拟登陆
This commit is contained in:
jxxghp 2023-08-29 14:04:57 +08:00 committed by GitHub
commit 72fbbffa02
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 315 additions and 211 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,7 +31,7 @@ class AutoSignIn(_PluginBase):
# 插件名称 # 插件名称
plugin_name = "站点自动签到" plugin_name = "站点自动签到"
# 插件描述 # 插件描述
plugin_desc = "自动模拟登录站点并签到。" plugin_desc = "自动模拟登录站点或签到。"
# 插件图标 # 插件图标
plugin_icon = "signin.png" plugin_icon = "signin.png"
# 主题色 # 主题色
@ -61,6 +61,7 @@ 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
@ -69,6 +70,7 @@ class AutoSignIn(_PluginBase):
_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,6 +89,8 @@ class AutoSignIn(_PluginBase):
self._sign_sites = config.get("sign_sites") self._sign_sites = config.get("sign_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:
@ -102,8 +106,8 @@ class AutoSignIn(_PluginBase):
if self._cron.strip().count(" ") == 4: if 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="站点自动签到") name=f"站点自动{self._action}")
logger.info(f"站点自动签到服务启动，执行周期 {self._cron}") logger.info(f"站点自动{self._action}服务启动，执行周期 {self._cron}")
else: else:
# 2.3/9-23 # 2.3/9-23
crons = self._cron.strip().split("/") crons = self._cron.strip().split("/")
@ -121,10 +125,10 @@ 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(self._cron.strip()), hours=float(self._cron.strip()),
name="站点自动签到") name=f"站点自动{self._action}")
logger.info(f"站点自动签到服务启动，执行周期 {self._cron}") logger.info(f"站点自动{self._action}服务启动，执行周期 {self._cron}")
else: else:
logger.error("站点自动签到服务启动失败，周期格式错误") logger.error(f"站点自动{self._action}服务启动失败，周期格式错误")
# 推送实时消息 # 推送实时消息
self.systemmessage.put(f"执行周期配置错误") self.systemmessage.put(f"执行周期配置错误")
except Exception as err: except Exception as err:
@ -141,13 +145,13 @@ 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="站点自动签到") name=f"站点自动{self._action}")
if self._onlyonce: if self._onlyonce:
logger.info(f"站点自动签到服务启动，立即运行一次") logger.info(f"站点自动{self._action}服务启动，立即运行一次")
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="站点自动签到") name=f"站点自动{self._action}")
# 关闭一次性开关 # 关闭一次性开关
self._onlyonce = False self._onlyonce = False
@ -174,6 +178,7 @@ class AutoSignIn(_PluginBase):
"sign_sites": self._sign_sites, "sign_sites": self._sign_sites,
"retry_keyword": self._retry_keyword, "retry_keyword": self._retry_keyword,
"clean": self._clean, "clean": self._clean,
"sign_type": self._sign_type,
} }
) )
@ -216,188 +221,209 @@ class AutoSignIn(_PluginBase):
site_options = [{"title": site.get("name"), "value": site.get("id")} site_options = [{"title": site.get("name"), "value": site.get("id")}
for site in self.sites.get_indexers()] for site in self.sites.get_indexers()]
return [ return [
{ {
'component': 'VForm', 'component': 'VForm',
'content': [ 'content': [
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 3 'md': 3
}, },
'content': [ 'content': [
{ {
'component': 'VSwitch', 'component': 'VSwitch',
'props': { 'props': {
'model': 'enabled', 'model': 'enabled',
'label': '启用插件', 'label': '启用插件',
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 3 'md': 3
}, },
'content': [ 'content': [
{ {
'component': 'VSwitch', 'component': 'VSwitch',
'props': { 'props': {
'model': 'notify', 'model': 'notify',
'label': '发送通知', 'label': '发送通知',
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 3 'md': 3
}, },
'content': [ 'content': [
{ {
'component': 'VSwitch', 'component': 'VSwitch',
'props': { 'props': {
'model': 'onlyonce', 'model': 'onlyonce',
'label': '立即运行一次', 'label': '立即运行一次',
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 3 'md': 3
}, },
'content': [ 'content': [
{ {
'component': 'VSwitch', 'component': 'VSwitch',
'props': { 'props': {
'model': 'clean', 'model': 'clean',
'label': '清理本日已签到', 'label': '清理本日已签到',
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 4 'md': 3
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VSelect',
'props': { 'props': {
'model': 'cron', 'model': 'sign_type',
'label': '执行周期', 'label': '签到方式',
'placeholder': '5位cron表达式，留空自动' 'items': [
} {'title': '签到', 'value': 'sign'},
} {'title': '登录', 'value': 'login'}
] ]
}, }
{ }
'component': 'VCol', ]
'props': { },
'cols': 12, {
'md': 4 'component': 'VCol',
}, 'props': {
'content': [ 'cols': 12,
{ 'md': 3
'component': 'VTextField', },
'props': { 'content': [
'model': 'queue_cnt', {
'label': '队列数量' 'component': 'VTextField',
} 'props': {
} 'model': 'cron',
] 'label': '执行周期',
}, 'placeholder': '5位cron表达式，留空自动'
{ }
'component': 'VCol', }
'props': { ]
'cols': 12, },
'md': 4 {
}, 'component': 'VCol',
'content': [ 'props': {
{ 'cols': 12,
'component': 'VTextField', 'md': 3
'props': { },
'model': 'retry_keyword', 'content': [
'label': '重试关键词', {
'placeholder': '重新签到关键词，支持正则表达式；每天首次全签，后续如果设置了重试词则只签到命中重试词的站点，否则全签。' 'component': 'VTextField',
} 'props': {
} 'model': 'queue_cnt',
] 'label': '队列数量'
} }
] }
}, ]
{ },
'component': 'VRow', {
'content': [ 'component': 'VCol',
{ 'props': {
'component': 'VCol', 'cols': 12,
'content': [ 'md': 3
{ },
'component': 'VSelect', 'content': [
'props': { {
'chips': True, 'component': 'VTextField',
'multiple': True, 'props': {
'model': 'sign_sites', 'model': 'retry_keyword',
'label': '签到站点', 'label': '重试关键词',
'items': site_options 'placeholder': '重新签到关键词，支持正则表达式；每天首次全签，后续如果设置了重试词则只签到命中重试词的站点，否则全签。'
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'content': [
'cols': 12, {
}, 'component': 'VSelect',
'content': [ 'props': {
{ 'chips': True,
'component': 'VAlert', 'multiple': True,
'props': { 'model': 'sign_sites',
'text': '签到周期支持：' 'label': '签到站点',
'1、5位cron表达式；' 'items': site_options
'2、配置间隔（小时），如2.3/9-23（9-23点之间每隔2.3小时执行一次）；' }
'3、周期不填默认9-23点随机执行2次。' }
'每天首次签到全量签到，其余执行命中重试关键词的站点。' ]
} }
} ]
] },
} {
] 'component': 'VRow',
} 'content': [
] {
} 'component': 'VCol',
], { 'props': {
"enabled": False, 'cols': 12,
"notify": True, },
"cron": "", 'content': [
"onlyonce": False, {
"clean": False, 'component': 'VAlert',
"queue_cnt": 5, 'props': {
"sign_sites": [], 'text': '执行周期支持：'
"retry_keyword": "错误|失败" '1、5位cron表达式；'
} '2、配置间隔（小时），如2.3/9-23（9-23点之间每隔2.3小时执行一次）；'
'3、周期不填默认9-23点随机执行2次。'
'每天首次全量执行，其余执行命中重试关键词的站点。'
}
}
]
}
]
}
]
}
], {
"enabled": False,
"notify": True,
"sign_type": "sign",
"cron": "",
"onlyonce": False,
"clean": False,
"queue_cnt": 5,
"sign_sites": [],
"retry_keyword": "错误|失败"
}
def get_page(self) -> List[dict]: def get_page(self) -> List[dict]:
""" """
@ -502,19 +528,19 @@ class AutoSignIn(_PluginBase):
@eventmanager.register(EventType.SiteSignin) @eventmanager.register(EventType.SiteSignin)
def sign_in(self, event: Event = None): def sign_in(self, event: Event = None):
""" """
自动签到 自动签到|模拟登陆
""" """
# 日期 # 日期
today = datetime.today() today = datetime.today()
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} 范围内，暂不签到") f"当前时间 {int(datetime.today().hour)} 不在 {self._start_time}-{self._end_time} 范围内，暂不{self._action}")
return return
if event: if event:
logger.info("收到命令，开始站点签到 ...") logger.info(f"收到命令，开始站点{self._action} ...")
self.post_message(channel=event.event_data.get("channel"), self.post_message(channel=event.event_data.get("channel"),
title="开始站点签到 ...", title=f"开始站点{self._action} ...",
userid=event.event_data.get("user")) userid=event.event_data.get("user"))
yesterday = today - timedelta(days=1) yesterday = today - timedelta(days=1)
@ -534,7 +560,7 @@ class AutoSignIn(_PluginBase):
# 今日没数据 # 今日没数据
if not today_history or self._clean: if not today_history or self._clean:
logger.info(f"今日 {today} 未签到，开始签到已选站点") logger.info(f"今日 {today} 未{self._action}，开始{self._action}已选站点")
# 过滤删除的站点 # 过滤删除的站点
self._sign_sites = [site.get("id") for site in sign_sites if site] self._sign_sites = [site.get("id") for site in sign_sites if site]
if self._clean: if self._clean:
@ -551,24 +577,28 @@ class AutoSignIn(_PluginBase):
site.get("id") not in already_sign_sites or site.get("id") in retry_sites] site.get("id") not in already_sign_sites or site.get("id") in retry_sites]
if not no_sign_sites: if not no_sign_sites:
logger.info(f"今日 {today} 已签到，无重新签到站点，本次任务结束") logger.info(f"今日 {today} 已{self._action}，无重新{self._action}站点，本次任务结束")
return return
# 签到站点 = 需要重签+今日未签 # 签到站点 = 需要重签+今日未签
sign_sites = no_sign_sites sign_sites = no_sign_sites
logger.info(f"今日 {today} 已签到，开始重签重试站点、特殊站点、未签站点") logger.info(f"今日 {today} 已{self._action}，开始重试命中关键词站点")
if not sign_sites: if not sign_sites:
logger.info("没有需要签到的站点") logger.info(f"没有需要{self._action}的站点")
return return
# 执行签到 # 执行签到
logger.info("开始执行签到任务 ...") logger.info(f"开始执行{self._action}任务 ...")
with ThreadPool(min(len(sign_sites), int(self._queue_cnt))) as p: if str(self._sign_type) == "sign":
status = p.map(self.signin_site, sign_sites) with ThreadPool(min(len(sign_sites), int(self._queue_cnt))) as p:
status = p.map(self.signin_site, sign_sites)
else:
with ThreadPool(min(len(sign_sites), int(self._queue_cnt))) as p:
status = p.map(self.login_site, sign_sites)
if status: if status:
logger.info("站点签到任务完成！") logger.info(f"站点{self._action}任务完成！")
# 获取今天的日期 # 获取今天的日期
key = f"{datetime.now().month}月{datetime.now().day}日" key = f"{datetime.now().month}月{datetime.now().day}日"
# 保存数据 # 保存数据
@ -624,7 +654,7 @@ 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"下次签到重试站点 {retry_sites}") logger.debug(f"下次{self._action}重试站点 {retry_sites}")
# 存入历史 # 存入历史
self.save_data(key=today, self.save_data(key=today,
@ -640,21 +670,22 @@ class AutoSignIn(_PluginBase):
if len(retry_msg) > 0: if len(retry_msg) > 0:
signin_message += retry_msg signin_message += retry_msg
self.post_message(title="站点自动签到", signin_message = "\n".join([f'【{s[0]}】{s[1]}' for s in signin_message if s])
self.post_message(title=f"站点自动{self._action}",
mtype=NotificationType.SiteMessage, mtype=NotificationType.SiteMessage,
text=f"全部签到数量: {len(list(self._sign_sites))} \n" text=f"全部{self._action}数量: {len(list(self._sign_sites))} \n"
f"本次签到数量: {len(sign_sites)} \n" f"本次{self._action}数量: {len(sign_sites)} \n"
f"下次签到数量: {len(retry_sites) if self._retry_keyword else 0} \n" f"下次{self._action}数量: {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="站点签到完成！", userid=event.event_data.get("user")) title=f"站点{self._action}完成！", userid=event.event_data.get("user"))
else: else:
logger.error("站点签到任务失败！") logger.error(f"站点{self._action}任务失败！")
if event: if event:
self.post_message(channel=event.event_data.get("channel"), self.post_message(channel=event.event_data.get("channel"),
title="站点签到任务失败！", userid=event.event_data.get("user")) title=f"站点{self._action}任务失败！", userid=event.event_data.get("user"))
# 保存配置 # 保存配置
self.__update_config() self.__update_config()
@ -736,6 +767,8 @@ class AutoSignIn(_PluginBase):
if under_challenge(page_source): if under_challenge(page_source):
return f"无法通过Cloudflare！" return f"无法通过Cloudflare！"
return f"仿真登录失败，Cookie已失效！" return f"仿真登录失败，Cookie已失效！"
else:
return "仿真签到成功"
else: else:
res = RequestUtils(cookies=site_cookie, res = RequestUtils(cookies=site_cookie,
ua=ua, ua=ua,
@ -772,6 +805,77 @@ class AutoSignIn(_PluginBase):
traceback.print_exc() traceback.print_exc()
return f"签到失败：{str(e)}！" return f"签到失败：{str(e)}！"
def login_site(self, site_info: CommentedMap) -> Tuple[str, str]:
"""
模拟登陆一个站点
"""
return site_info.get("name"), self.__login_base(site_info)
@staticmethod
def __login_base(site_info: CommentedMap) -> str:
"""
模拟登陆通用处理
:param site_info: 站点信息
:return: 签到结果信息
"""
if not site_info:
return ""
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
if not site_url or not site_cookie:
logger.warn(f"未配置 {site} 的站点地址或Cookie，无法签到")
return ""
# 模拟登录
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)
if not SiteUtils.is_logged_in(page_source):
if under_challenge(page_source):
return f"无法通过Cloudflare！"
return f"仿真登录失败，Cookie已失效！"
else:
return "模拟登陆成功"
else:
res = RequestUtils(cookies=site_cookie,
ua=ua,
proxies=proxies
).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 f"模拟登陆失败，{msg}！"
else:
logger.info(f"{site} 模拟登陆成功")
return f"模拟登陆成功"
elif res is not None:
logger.warn(f"{site} 模拟登陆失败，状态码：{res.status_code}")
return f"模拟登陆失败，状态码：{res.status_code}！"
else:
logger.warn(f"{site} 模拟登陆失败，无法打开网站")
return f"模拟登陆失败，无法打开网站！"
except Exception as e:
logger.warn("%s 模拟登陆失败：%s" % (site, str(e)))
traceback.print_exc()
return f"模拟登陆失败：{str(e)}！"
def stop_service(self): def stop_service(self):
""" """
退出插件 退出插件