redis 链接超时断开的修改

apps/auto_convert3d.py

@@ -158,7 +158,7 @@ def team_check(r):
         print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), 'redis连接异常，5秒后重试')
         print(e)
         time.sleep(5)
-        r = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
+        r = create_redis_connection()
         return
     repair_id = r.lpop('model:IndependentRepairTeamcheckGLBQueue')
     if repair_id is None: return
@@ -239,12 +239,32 @@ def objToGlbAndUpload(pid):
     oss_client.put_object_from_file(f'glbs/print/{glb_filename}', os.path.join(workdir, "print", pid, glb_filename))
 
 
+def create_redis_connection():
+    """创建 Redis 连接，若连接失败则重试"""
+    while True:
+        try:
+            r = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
+            # 尝试进行一次操作，检查连接是否有效
+            r.ping()  # ping 操作是一个简单的连接测试
+            print("Redis连接成功！")
+            return r
+        except ConnectionError:
+            print("Redis连接失败，正在重试...")
+            time.sleep(5) 
+
+
 def main(r):
     print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), '模型生成程序 start')
     while True:  
-        team_check(r)
-        # platform_check(r)
-        #make3d4print_task(r)                    
+        try:
+            team_check(r)
+            # platform_check(r)
+            #make3d4print_task(r)     
+        except Exception as e:
+            print('出现异常:', e)
+            time.sleep(5)
+            r = create_redis_connection()
+            continue               
 
 if __name__ == '__main__':
     atexit.register(common.notify,"R13-修模师审核通过，生成打印任务程序停止")
@@ -266,5 +286,5 @@ if __name__ == '__main__':
 
     blenderbin = find_blender_bin_path()
 
-    r = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
+    r = create_redis_connection()
     main(r)