测试组/脚本/Change_password/修改数据库的哈希密码和原始密码做桌面客户端.py
@@ -31,6 +31,89 @@
DEFAULT_NEW_PASSWORD = 'Baoyi@1341'
PROTECTED_USER_ID = 1
# Redis配置
REDIS_HOST = '192.168.6.168'
REDIS_PORT = 6002
REDIS_PASSWORD = None
class RedisRefresher:
    """Redis缓存刷新器"""
    def __init__(self):
        self.redis_client = None
    def connect_redis(self):
        """连接Redis服务器"""
        try:
            # 尝试导入redis库
            import redis
            self.redis_client = redis.Redis(
                host=REDIS_HOST,
                port=REDIS_PORT,
                password=REDIS_PASSWORD,
                decode_responses=True
            )
            # 测试连接
            self.redis_client.ping()
            return True
        except ImportError:
            print("错误: 未安装redis库,请使用 'pip install redis' 安装")
            return False
        except Exception as e:
            print(f"连接Redis失败: {str(e)}")
            return False
    def refresh_all_redis(self):
        """刷新所有Redis数据库"""
        try:
            if not self.redis_client:
                if not self.connect_redis():
                    return False, "无法连接Redis服务器"
            # 获取所有数据库数量
            try:
                # 尝试获取配置信息
                config = self.redis_client.config_get('databases')
                db_count = int(config.get('databases', 16))
            except:
                # 如果无法获取配置,默认使用16个数据库
                db_count = 16
            refreshed_dbs = 0
            total_keys = 0
            # 遍历所有数据库并刷新
            for db_index in range(db_count):
                try:
                    # 切换到指定数据库
                    temp_client = self.redis_client.connection_pool.get_connection()
                    temp_client.send_command('SELECT', db_index)
                    temp_client.read_response()
                    # 获取当前数据库的key数量
                    temp_client.send_command('DBSIZE')
                    key_count = temp_client.read_response()
                    if key_count > 0:
                        # 清空当前数据库
                        temp_client.send_command('FLUSHDB')
                        temp_client.read_response()
                        refreshed_dbs += 1
                        total_keys += key_count
                        print(f"已刷新数据库 {db_index}: 清除了 {key_count} 个键")
                    self.redis_client.connection_pool.release(temp_client)
                except Exception as e:
                    print(f"刷新数据库 {db_index} 时出错: {str(e)}")
                    continue
            return True, f"成功刷新 {refreshed_dbs} 个Redis数据库,共清除 {total_keys} 个键"
        except Exception as e:
            return False, f"刷新Redis缓存时发生错误: {str(e)}"
class DatabaseUpdater(QThread):
    """数据库更新器类,运行在单独的线程中"""
@@ -52,6 +135,7 @@
        self.updaters = []
        self.log_records = []
        self.is_running = True
        self.redis_refresher = RedisRefresher()
    def run(self):
        """主运行方法"""
@@ -82,9 +166,9 @@
            # 创建数据库更新器实例
            self.updaters = []
            for _, row in df_filtered.iterrows():
                project, host, database, user, password = row[:5]
                self.log_signal.emit(f"准备处理项目:{project} | 数据库:{database}")
                updater = ProjectUpdater(project, host, database, user, password)
                project, host, port, database, user, password = row[:6]
                self.log_signal.emit(f"准备处理项目:{project} | 数据库:{database} | 端口:{port}")
                updater = ProjectUpdater(project, host, database, user, password, port)
                self.updaters.append(updater)
            # 执行更新任务
@@ -116,6 +200,14 @@
                self.log_signal.emit(f"所有更新日志已保存至 {DEFAULT_LOG_CSV_PATH}")
            else:
                self.log_signal.emit("无任何更新记录生成")
            # 数据库更新完成后,自动刷新Redis缓存
            self.log_signal.emit("开始刷新Redis缓存...")
            success, message = self.redis_refresher.refresh_all_redis()
            if success:
                self.log_signal.emit(f"✅ {message}")
            else:
                self.log_signal.emit(f"❌ {message}")
        except Exception as e:
            self.error_signal.emit(f"运行过程中发生错误: {str(e)}")
@@ -162,12 +254,13 @@
class ProjectUpdater:
    """单个项目更新器"""
    def __init__(self, project, host, database, user, password):
    def __init__(self, project, host, database, user, password, port=3306):
        self.project = project
        self.host = host
        self.database = database
        self.db_user = user
        self.db_password = password
        self.port = port
        self.updated_users = []
        self.log_records = []
@@ -176,7 +269,7 @@
        try:
            conn = mysql.connector.connect(
                host=self.host,
                port=3306,
                port=self.port,
                user=self.db_user,
                password=self.db_password,
                database=self.database,