🐛 统一被动技能判断

This commit is contained in:
HibiKier
2024-08-03 00:34:19 +08:00
parent 7ab800be97
commit 131cd5ea9a
6 changed files with 14 additions and 23 deletions
@@ -26,6 +26,7 @@ from zhenxun.models.group_console import GroupConsole
from zhenxun.models.group_member_info import GroupInfoUser
from zhenxun.models.level_user import LevelUser
from zhenxun.models.plugin_info import PluginInfo
from zhenxun.models.task_info import TaskInfo
from zhenxun.services.log import logger
from zhenxun.utils.enum import PluginType, RequestHandleType
from zhenxun.utils.utils import FreqLimiter
@@ -240,7 +241,7 @@ async def _(bot: Bot, event: GroupIncreaseNoticeEvent | GroupMemberIncreaseEvent
img_file = path / f"{i}.png"
if img_file.exists():
msg_list.append(Image(img_file))
if not GroupConsole.is_block_task(group_id, "group_welcome"):
if not TaskInfo.is_block("group_welcome", group_id):
logger.info(f"发送群欢迎消息...", "入群检测", group_id=group_id)
if msg_list:
await MessageFactory(msg_list).send()
@@ -311,5 +312,5 @@ async def _(bot: Bot, event: GroupDecreaseNoticeEvent | GroupMemberDecreaseEvent
)
operator_name = operator["card"] if operator["card"] else operator["nickname"]
result = f"{user_name} 被 {operator_name} 送走了."
if not GroupConsole.is_block_task(str(event.group_id), "refund_group_remind"):
if not TaskInfo.is_block("refund_group_remind", str(event.group_id)):
await group_decrease_handle.send(f"{result}")
+2 -5
View File
@@ -40,11 +40,8 @@ async def _():
)
async def check(group_id: str, channel_id: str | None) -> bool:
task = await TaskInfo.get_or_none(module="morning_goodnight")
if not task or not task.status:
return False
return await GroupConsole.is_block_task(group_id, "morning_goodnight")
async def check(group_id: str) -> bool:
return not await TaskInfo.is_block("morning_goodnight", group_id)
# 早上好
@@ -10,6 +10,7 @@ from nonebot_plugin_saa import Image, MessageFactory, Text
from nonebot_plugin_session import EventSession
from zhenxun.models.group_console import GroupConsole
from zhenxun.models.task_info import TaskInfo
from zhenxun.services.log import logger
from zhenxun.utils.platform import PlatformUtils
@@ -41,8 +42,9 @@ class BroadcastManage:
error_count = 0
for group in group_list:
try:
if not await GroupConsole.is_block_task(
group.group_id, "broadcast", group.channel_id
if not await TaskInfo.is_block(
group.group_id,
"broadcast", # group.channel_id
):
target = PlatformUtils.get_target(
bot, None, group.channel_id or group.group_id