mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-28 16:20:56 +08:00
8b16126e40
* bugfix:修复内存泄露和信号量饥饿问题 * 修复图片渲染按高度截断问题 * 优化图片渲染速度 * 权限检查去掉无效引用代码 * 添加uv支持 * 🚨 auto fix by pre-commit hooks * bugfix:修改gitignore换行 * bugfix:修复测试没有新生成uv.lock * 修复导入错误 * bugfix:移除重复调用 * 🚨 auto fix by pre-commit hooks * 清理残余poetry引用 * 更新uv安装方式 * 修复阿里云获取问题 * 增加资源下载提示 * 🚨 auto fix by pre-commit hooks * 修改资源下载为流式 * 🚨 auto fix by pre-commit hooks * 提高启动速度 * 移除bot.py支持 * 🚨 auto fix by pre-commit hooks * 优化win脚本逻辑 * 🚨 auto fix by pre-commit hooks * 清理残余无效逻辑 * 代码改进 * 🚨 auto fix by pre-commit hooks * 增加数据库迁移存在性检查 * 🚨 auto fix by pre-commit hooks * chore(test): 添加pytest超时控制和优雅关闭机制 - 在GitHub Actions工作流中添加作业级和步骤级超时限制，防止测试无限期挂起 - 添加pytest-timeout依赖并配置全局超时为120秒 - 在send_queue服务添加关闭钩子，确保worker任务正确取消 - 在priority_manager添加on_shutdown钩子，支持优先级生命周期的关闭阶段 * chore(lint): 禁用超长行的lint警告 * Modify restart logic for Windows platform * 🚨 auto fix by pre-commit hooks * bugfix:修复sys导入问题 * 清理无效结构 * bugfix:修复路径问题 * bugfix:修复shell语法传递给git导致资源获取失败问题 * 优化关闭显示 * bugfix:修复路径问题 * bugfix:增加路径安全 * bugfix:修复orm绕过问题 * 放宽numpy版本限制 * 修改重启方案 * bugfix:修复循环导入 * 优化逻辑 * Enhance disconnect function with error handling Added error handling for disconnect function and imported ConfigurationError. * Implement emergency restart mechanism Added emergency restart mechanism using atexit to ensure process restart even on severe exceptions during shutdown. * 🚨 auto fix by pre-commit hooks * 重启行为归一化 * 修复测试检测问题 * bugfix:修复测试侧类型报错问题 * 引入launcher机制 * 移除重启测试 * 收紧缓存调用路径 * 类型注解收敛 * 优化浏览器回收行为 * 优化浏览器渲染 * bugfix:解决重复关闭浏览器问题 --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: ManyManyTomato <93612024+ATTomatoo@users.noreply.github.com> Co-authored-by: AkashiCoin <l1040186796@gmail.com>
66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
from typing import ClassVar
|
|
|
|
from tortoise import fields
|
|
|
|
from zhenxun.services.cache.runtime_cache import PluginLimitMemoryCache
|
|
from zhenxun.services.db_context import Model
|
|
from zhenxun.utils.enum import LimitCheckType, LimitWatchType, PluginLimitType
|
|
|
|
|
|
class PluginLimit(Model):
|
|
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
|
"""自增id"""
|
|
module = fields.CharField(255, description="模块名")
|
|
"""模块名"""
|
|
module_path = fields.CharField(255, description="模块路径")
|
|
"""模块路径"""
|
|
plugin = fields.ForeignKeyField(
|
|
"models.PluginInfo",
|
|
related_name="plugin_limit",
|
|
on_delete=fields.CASCADE,
|
|
description="所属插件",
|
|
)
|
|
"""所属插件"""
|
|
limit_type = fields.CharEnumField(PluginLimitType, description="限制类型")
|
|
"""限制类型"""
|
|
watch_type = fields.CharEnumField(LimitWatchType, description="监听类型")
|
|
"""监听类型"""
|
|
status = fields.BooleanField(default=True, description="限制的开关状态")
|
|
"""限制的开关状态"""
|
|
check_type = fields.CharEnumField(
|
|
LimitCheckType, default=LimitCheckType.ALL, description="检查类型"
|
|
)
|
|
"""检查类型"""
|
|
result = fields.CharField(max_length=255, null=True, description="返回信息")
|
|
"""返回信息"""
|
|
cd = fields.IntField(null=True, description="cd")
|
|
"""cd"""
|
|
max_count = fields.IntField(null=True, description="最大调用次数")
|
|
"""最大调用次数"""
|
|
|
|
class Meta: # pyright: ignore [reportIncompatibleVariableOverride]
|
|
table = "plugin_limit"
|
|
table_description = "插件限制"
|
|
indexes: ClassVar = [("module", "status")]
|
|
|
|
@classmethod
|
|
async def create(cls, *args, **kwargs):
|
|
result = await super().create(*args, **kwargs)
|
|
await PluginLimitMemoryCache.upsert_from_model(result)
|
|
return result
|
|
|
|
@classmethod
|
|
async def update_or_create(cls, *args, **kwargs):
|
|
result = await super().update_or_create(*args, **kwargs)
|
|
await PluginLimitMemoryCache.upsert_from_model(result[0])
|
|
return result
|
|
|
|
async def save(self, *args, **kwargs):
|
|
await super().save(*args, **kwargs)
|
|
await PluginLimitMemoryCache.upsert_from_model(self)
|
|
|
|
async def delete(self, *args, **kwargs):
|
|
limit_id = self.id
|
|
await super().delete(*args, **kwargs)
|
|
await PluginLimitMemoryCache.remove_by_id(limit_id)
|