From 3120dcb82ea14fd4081aa4f39324a05ef83416e7 Mon Sep 17 00:00:00 2001
From: dongchangxi <458593490@qq.com>
Date: Sun, 5 Jan 2025 10:41:27 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A7=BF=E5=8A=BF=E4=B8=8A=E4=BC=A0=E5=A4=84?=
 =?UTF-8?q?=E7=90=86=E5=90=8E=EF=BC=8C=E5=88=A0=E9=99=A4=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E5=A4=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 new_studio_to_reg/get_new_studio_reg_info.py | 2 ++
 old_to_new_build/get_reg_info.py             | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/new_studio_to_reg/get_new_studio_reg_info.py b/new_studio_to_reg/get_new_studio_reg_info.py
index 352b340..732fbf5 100644
--- a/new_studio_to_reg/get_new_studio_reg_info.py
+++ b/new_studio_to_reg/get_new_studio_reg_info.py
@@ -116,6 +116,8 @@ def get_photo2_align_info(pid):
     r = create_redis_connection()
     r.rpush('ai:ai_build_mesh',pid)
 
+    #删除文件夹
+    shutil.rmtree(os.path.join(config.workdir, pid), ignore_errors=True)
 
 def create_redis_connection():
     """创建 Redis 连接，若连接失败则重试"""
diff --git a/old_to_new_build/get_reg_info.py b/old_to_new_build/get_reg_info.py
index c466509..61edc0b 100644
--- a/old_to_new_build/get_reg_info.py
+++ b/old_to_new_build/get_reg_info.py
@@ -109,6 +109,9 @@ def get_photo2_align_info(pid,flagDoXmps=False):
     #处理完成后塞入到指定的队列
     r = create_redis_connection()
     r.rpush('ai:ai_repeat_build',pid)
+    
+    #删除文件夹
+    shutil.rmtree(os.path.join(config.workdir, pid), ignore_errors=True)
 
 
 def create_redis_connection():