diff --git a/DB.py b/DB.py index 2c3feda..ce339e1 100644 --- a/DB.py +++ b/DB.py @@ -293,10 +293,22 @@ class DBVidcon: """批量执行完后手动提交。""" self.conn.commit() - @mysql_retry() def close(self): - self.cursor.close() - self.conn.close() + try: + if self.cursor: + self.cursor.close() + finally: + if self.conn: + self.conn.close() + try: + if hasattr(self, "redis") and self.redis: + # redis-py ≥ 4.2 推荐直接调用 .close() + if hasattr(self.redis, "close"): + self.redis.close() + else: + self.redis.connection_pool.disconnect() + except Exception as e: + print("[Redis close error]", e) @redis_retry(max_retries=3) def get_proxy(self, region_code: str) -> str: