62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
import os
|
|
import shutil
|
|
import folder_paths
|
|
|
|
class CleanFolders:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {
|
|
"input_path": ("STRING", {"default": ""}),
|
|
"output_path": ("STRING", {"default": ""})
|
|
}}
|
|
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "clean"
|
|
CATEGORY = "comfyui_zip"
|
|
OUTPUT_NODE = True
|
|
|
|
def clean(self, input_path, output_path):
|
|
result_msg = []
|
|
|
|
# 获取基础目录
|
|
input_base = folder_paths.get_input_directory()
|
|
output_base = folder_paths.get_output_directory()
|
|
|
|
# 确保路径是字符串类型，如果是元组则取第一个元素
|
|
input_path = input_path[0] if isinstance(input_path, tuple) else str(input_path) if input_path else ""
|
|
output_path = output_path[0] if isinstance(output_path, tuple) else str(output_path) if output_path else ""
|
|
|
|
# 只有当提供了路径时才进行删除操作
|
|
if input_path:
|
|
input_full_path = os.path.join(input_base, input_path)
|
|
if os.path.exists(input_full_path):
|
|
try:
|
|
shutil.rmtree(input_full_path)
|
|
result_msg.append(f"已清理输入目录: {input_full_path}")
|
|
except Exception as e:
|
|
result_msg.append(f"清理输入目录失败: {str(e)}")
|
|
else:
|
|
result_msg.append(f"输入目录不存在: {input_full_path}")
|
|
else:
|
|
result_msg.append("未提供输入路径，跳过清理")
|
|
|
|
# 只有当提供了路径时才进行删除操作
|
|
if output_path:
|
|
output_full_path = os.path.join(output_base, output_path)
|
|
if os.path.exists(output_full_path):
|
|
try:
|
|
shutil.rmtree(output_full_path)
|
|
except Exception as e:
|
|
pass
|
|
|
|
return ("",)
|
|
|
|
# 注册节点
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CleanFolders": CleanFolders
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CleanFolders": "清理文件夹"
|
|
}
|