daxcay
b5d49a1ebb
Decision was due to the increase of loading size of JDCN users who were not Using Reasearch nodes
103 lines
2.9 KiB
Python
103 lines
2.9 KiB
Python
import glob
|
|
import os
|
|
import shutil
|
|
|
|
|
|
def create_folder_if_not_exists(folder_path):
|
|
if not os.path.exists(folder_path):
|
|
os.makedirs(folder_path)
|
|
print(f"Folder '{folder_path}' created.")
|
|
else:
|
|
print(f"Folder '{folder_path}' already exists.")
|
|
|
|
def move_it(source_path, destination_dir, overwrite=False):
|
|
try:
|
|
create_folder_if_not_exists(destination_dir)
|
|
filename = os.path.basename(source_path)
|
|
destination_path = os.path.join(destination_dir, filename)
|
|
|
|
if os.path.exists(destination_path):
|
|
if overwrite:
|
|
shutil.move(source_path, destination_path)
|
|
else:
|
|
base, ext = os.path.splitext(filename)
|
|
i = 1
|
|
while True:
|
|
new_filename = f"{base}_{i}{ext}"
|
|
new_destination_path = os.path.join(destination_dir, new_filename)
|
|
if not os.path.exists(new_destination_path):
|
|
destination_path = new_destination_path
|
|
break
|
|
i += 1
|
|
shutil.move(source_path, destination_path)
|
|
else:
|
|
shutil.move(source_path, destination_path)
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
|
|
def create_folder_if_not_exists(folder_path):
|
|
if not os.path.exists(folder_path):
|
|
os.makedirs(folder_path)
|
|
print(f"Folder '{folder_path}' created.")
|
|
else:
|
|
print(f"Folder '{folder_path}' already exists.")
|
|
|
|
|
|
def get_files_in_folder(folder_path):
|
|
if not os.path.isdir(folder_path):
|
|
print("Folder does not exist.")
|
|
return []
|
|
files = glob.glob(os.path.join(folder_path, "*"))
|
|
return files
|
|
|
|
|
|
class JDCN_FileMover:
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
|
|
return {
|
|
"required": {
|
|
"FilePaths": ("STRING", {"forceInput": True}),
|
|
"OutputDirectory": ("STRING", {"default": "directory path"}),
|
|
"OverwriteFile": ("BOOLEAN", {"default": False}),
|
|
},
|
|
}
|
|
|
|
INPUT_IS_LIST = True
|
|
RETURN_TYPES = ("STRING",)
|
|
RETURN_NAMES = ("NewFilePaths",)
|
|
OUTPUT_IS_LIST = (True,)
|
|
OUTPUT_NODE = True
|
|
FUNCTION = "make_list"
|
|
CATEGORY = "🔵 JDCN 🔵"
|
|
|
|
def make_list(self, FilePaths, OutputDirectory, OverwriteFile):
|
|
|
|
if len(FilePaths) == 0:
|
|
print("Empty FileName string")
|
|
return ([],)
|
|
|
|
create_folder_if_not_exists(OutputDirectory[0])
|
|
|
|
for file in FilePaths:
|
|
move_it(file, OutputDirectory[0], OverwriteFile[0])
|
|
|
|
file_paths_new = get_files_in_folder(OutputDirectory[0])
|
|
|
|
return (file_paths_new,)
|
|
|
|
|
|
N_CLASS_MAPPINGS = {
|
|
"JDCN_FileMover": JDCN_FileMover,
|
|
}
|
|
|
|
N_DISPLAY_NAME_MAPPINGS = {
|
|
"JDCN_FileMover": "JDCN_FileMover",
|
|
}
|