Update utils.py

utils.py

@@ -49,6 +49,20 @@ def update_file_info(root_dir):
             update_file_info(os.path.join(root, dir_name))
     return info
 
+#删除章节或者mask
+def delete_file(delete_dir:str)->str:
+    for root, dirs, files in os.walk(delete_dir, topdown=False):
+        for name in files:
+            file_path = os.path.join(root, name)
+            print("删除指定目录函数检查到文件,删除文件:", file_path)
+            os.remove(file_path)
+        for name in dirs:
+            dir_path = os.path.join(root, name)
+            print("删除指定目录函数检查到目录,删除目录:", dir_path)
+            os.rmdir(dir_path)
+    return "message:"+ "成功删除{delete_dir}目录下的所有文件".format(delete_dir=delete_dir)
+
+
 def random_chapter():
     # for root, info in configData.data['file_info'].items():
     #手动更新避免引入旧值-成功
@@ -141,7 +155,7 @@ def is_asp_task_valid()->bool:
         print("No image files found in the directory.")
         return False
 
-    
+
     return True
 
 #定时任务流程函数集合 ->返回None说明非正常执行情况
@@ -163,7 +177,7 @@ with gr.Blocks() as mangaManager:
     #上传文件选择列表
     file_selected = gr.File(label="待上传压缩章节",file_types=['.zip', '.tar', '.gz'])
     #上传zip的解压保存路径或mask的地址(必选) 根据需要选择
-    output_dir_gr = gr.Dropdown(label="上传zip的解压保存路径或mask的地址(必选)", choices=configData.data['default_values'])
+    output_dir_gr = gr.Dropdown(label="zip的解压保存路径或mask的地址/删除地址(必选)", choices=configData.data['default_values'])
     #压缩包上传按钮
     file_upload_btn = gr.Button("开始上传")
     #获取到返回的结果-可以是上传的,可以是查看files信息,也可以是别的
@@ -187,6 +201,11 @@ with gr.Blocks() as mangaManager:
     config_update_btn = gr.Button("更新config.json数据")
     config_update_btn.click(fn=update_config_json, inputs=[config_update_text], outputs=someResult)
 
+    #删除指定目录文件 -对上传保存的文件进行删除 ->比如章节和mask不匹配,但是一个个删太麻烦,干脆一起
+    file_delete_btn = gr.Button("开始删除")
+    file_delete_btn.click(fn=delete_file, inputs=[output_dir_gr], outputs=someResult)
+
+
 
 # 定义一个函数来处理输入的步骤转list然后执行,返回执行结果->taskResult/None
 # gradio输入0 1 2 3就行,会自动转"0 1 2 3"然后给process_task_list处理