From 56f1d08bc9fd383d26c6626addfa725566141c68 Mon Sep 17 00:00:00 2001 From: liuwei Date: Fri, 16 May 2025 13:54:50 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8db=E7=AE=A1=E7=90=86=E7=B1=BB?= =?UTF-8?q?=E8=BF=9B=E8=A1=8C=E5=B7=A5=E5=85=B7=E5=88=9D=E5=A7=8B=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/robot_cmd/robot_command.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/utils/robot_cmd/robot_command.py b/utils/robot_cmd/robot_command.py index b183689..2fa6397 100644 --- a/utils/robot_cmd/robot_command.py +++ b/utils/robot_cmd/robot_command.py @@ -13,8 +13,10 @@ from enum import Enum from loguru import logger +from db.connection import DBConnectionManager + # 连接到本地 Redis 服务 -r = redis.StrictRedis(host='192.168.2.40', port=6379, db=0, decode_responses=True) +# r = redis.StrictRedis(host='192.168.2.40', port=6379, db=0, decode_responses=True) class PermissionStatus(Enum): @@ -59,6 +61,12 @@ class Feature(Enum): return self.description +def get_redis_connection(): + # 初始化时加载本地缓存 + db_manager: DBConnectionManager = DBConnectionManager.get_instance() + return db_manager.get_redis_connection() + + class GroupBotManager: """群机器人管理，支持本地缓存""" @@ -81,6 +89,7 @@ class GroupBotManager: @staticmethod def load_local_cache(): + r = get_redis_connection() """从 Redis 加载数据到本地缓存""" group_list = r.smembers("group:list") # 输出group_list 列表 @@ -100,6 +109,7 @@ class GroupBotManager: @staticmethod def save_to_redis(): + r = get_redis_connection() """将本地缓存保存回 Redis""" # 保存 group:list 到 Redis r.sadd("group:list", *GroupBotManager.local_cache["group_list"]) @@ -112,6 +122,7 @@ class GroupBotManager: @staticmethod def set_group_permission(group_id, feature: Feature, status: PermissionStatus): + r = get_redis_connection() """设置群组功能权限并更新本地缓存""" # 更新本地缓存 if group_id not in GroupBotManager.local_cache["group_permissions"]: @@ -185,6 +196,7 @@ class GroupBotManager: except KeyError: return "无效功能名称" + r = get_redis_connection() # 处理群机器人的启用和关闭（特别操作：更新 group:list） if feature == Feature.ROBOT: if action == "启用": @@ -283,7 +295,7 @@ class GroupBotManager: @staticmethod def remove_group(group_id): """一键清除某个群的所有设置，用于退群或关闭群时处理 - + Args: group_id: 群ID @@ -294,6 +306,7 @@ class GroupBotManager: if group_id not in GroupBotManager.local_cache["group_list"]: return f"群 {group_id} 不在机器人管理列表中" + r = get_redis_connection() # 从本地缓存中移除群组 GroupBotManager.local_cache["group_list"].remove(group_id) if group_id in GroupBotManager.local_cache["group_permissions"]: