From b1b15524d96470f16095b0636570a6e1c2d26d94 Mon Sep 17 00:00:00 2001 From: liuwei Date: Wed, 9 Apr 2025 14:56:27 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E5=85=A5=E6=9D=83=E9=99=90=E5=88=A4?= =?UTF-8?q?=E6=96=AD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/decorator/points_decorator.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/utils/decorator/points_decorator.py b/utils/decorator/points_decorator.py index 181103c..3ca30e1 100644 --- a/utils/decorator/points_decorator.py +++ b/utils/decorator/points_decorator.py @@ -6,7 +6,7 @@ from typing import Callable, Dict, Any, Tuple, Union from db.connection import DBConnectionManager from db.points_db import PointsDBOperator, PointSource -from utils.robot_cmd.robot_command import Feature, PermissionStatus +from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager def points_reward_decorator(points_calculator: Union[int, Callable], source_type: str = "other", @@ -28,8 +28,9 @@ def points_reward_decorator(points_calculator: Union[int, Callable], source_type def wrapper(self, message: Dict[str, Any]) -> Tuple[bool, str]: # 检查权限 roomid = message.get("roomid", "") - if feature and roomid and hasattr(self, 'gbm'): - if self.gbm.get_group_permission(roomid, feature) == PermissionStatus.DISABLED: + + if feature and roomid: + if GroupBotManager.get_group_permission(roomid, feature) == PermissionStatus.DISABLED: return False, "没有权限" # 调用原始方法 @@ -113,8 +114,8 @@ def plugin_points_cost(points: int, description: str = None, feature: Feature = try: # 检查权限 roomid = message.get("roomid", "") - if feature and roomid and hasattr(self, 'gbm'): - if self.gbm.get_group_permission(roomid, feature) == PermissionStatus.DISABLED: + if feature and roomid: + if GroupBotManager.get_group_permission(roomid, feature) == PermissionStatus.DISABLED: return False, "没有权限" # 获取消息信息