mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from tortoise import fields
|
|
|
|
from zhenxun.services.db_context import Model
|
|
|
|
from .group_console import GroupConsole
|
|
|
|
|
|
class TaskInfo(Model):
|
|
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
|
"""自增id"""
|
|
module = fields.CharField(255, description="被动技能模块名")
|
|
"""被动技能模块名"""
|
|
name = fields.CharField(255, description="被动技能名称")
|
|
"""被动技能名称"""
|
|
status = fields.BooleanField(default=True, description="全局开关状态")
|
|
"""全局开关状态"""
|
|
run_time = fields.CharField(255, null=True, description="运行时间")
|
|
"""运行时间"""
|
|
run_count = fields.IntField(default=0, description="运行次数")
|
|
"""运行次数"""
|
|
|
|
class Meta:
|
|
table = "task_info"
|
|
table_description = "被动技能基本信息"
|
|
|
|
@classmethod
|
|
async def is_block(cls, module: str, group_id: str | None) -> bool:
|
|
"""判断被动技能是否被禁用
|
|
|
|
参数:
|
|
module: 被动技能模块名
|
|
group_id: 群组id
|
|
|
|
返回:
|
|
bool: 是否被禁用
|
|
"""
|
|
if task := await cls.get_or_none(module=module):
|
|
if task.status:
|
|
return True
|
|
if group_id:
|
|
return await GroupConsole.is_block_task(group_id, module)
|
|
return False
|