🎨 修复bug,移除plugins文件夹

This commit is contained in:
HibiKier
2024-08-29 22:01:34 +08:00
parent e10fee74b3
commit 625f17a708
146 changed files with 580 additions and 18320 deletions
+60 -5
View File
@@ -1,6 +1,10 @@
from tortoise import fields
from typing import Any
from typing_extensions import Self
from tortoise import fields
from tortoise.backends.base.client import BaseDBAsyncClient
from zhenxun.models.task_info import TaskInfo
from zhenxun.services.db_context import Model
@@ -35,11 +39,64 @@ class GroupConsole(Model):
platform = fields.CharField(255, default="qq", description="所属平台")
"""所属平台"""
class Meta:
class Meta: # type: ignore
table = "group_console"
table_description = "群组信息表"
unique_together = ("group_id", "channel_id")
@classmethod
async def create(
cls, using_db: BaseDBAsyncClient | None = None, **kwargs: Any
) -> Self:
"""覆盖create方法"""
group = await super().create(using_db=using_db, **kwargs)
if modules := await TaskInfo.filter(default_status=False).values_list(
"module", flat=True
):
group.block_task = ",".join(modules) + "," # type: ignore
await group.save(using_db=using_db, update_fields=["block_task"])
return group
@classmethod
async def get_or_create(
cls,
defaults: dict | None = None,
using_db: BaseDBAsyncClient | None = None,
**kwargs: Any,
) -> tuple[Self, bool]:
"""覆盖get_or_create方法"""
group, is_create = await super().get_or_create(
defaults=defaults, using_db=using_db, **kwargs
)
if is_create and (
modules := await TaskInfo.filter(default_status=False).values_list(
"module", flat=True
)
):
group.block_task = ",".join(modules) + "," # type: ignore
await group.save(using_db=using_db, update_fields=["block_task"])
return group, is_create
@classmethod
async def update_or_create(
cls,
defaults: dict | None = None,
using_db: BaseDBAsyncClient | None = None,
**kwargs: Any,
) -> tuple[Self, bool]:
"""覆盖update_or_create方法"""
group, is_create = await super().update_or_create(
defaults=defaults, using_db=using_db, **kwargs
)
if is_create and (
modules := await TaskInfo.filter(default_status=False).values_list(
"module", flat=True
)
):
group.block_task = ",".join(modules) + "," # type: ignore
await group.save(using_db=using_db, update_fields=["block_task"])
return group, is_create
@classmethod
async def get_group(
cls, group_id: str, channel_id: str | None = None
@@ -67,9 +124,7 @@ class GroupConsole(Model):
返回:
bool: 是否超级用户指定群
"""
if group := await cls.get_group(group_id):
return group.is_super
return False
return group.is_super if (group := await cls.get_group(group_id)) else False
@classmethod
async def is_super_block_plugin(