nodes/daxcay-ComfyUI-JDCN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated move function overwrite bug

Browse Source
This commit is contained in:
akr1999
2024-03-21 20:38:56 +05:30
parent 4e74ce9470
commit cb3edcaacf
2 changed files with 49 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
JDCN_FileMover.py
+24 -17
View File
@@ -13,28 +13,35 @@ def create_folder_if_not_exists(folder_path):
def move_it(source_path, destination_dir, overwrite=False):
try:
# Ensure destination directory exists
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) and overwrite:
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)
# print(f"Image moved from '{source_path}' to '{destination_path}'")
# Check if the destination file exists
if os.path.exists(destination_path):
if overwrite:
# If overwrite is True, move the file directly
shutil.move(source_path, destination_path)
else:
# If overwrite is False, find a new filename with an incremented counter
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:
# If the destination file does not exist, move the file directly
shutil.move(source_path, destination_path)
except Exception as e:
print("")
# print(f"Error: {e}")
print(f"Error: {e}")
def create_folder_if_not_exists(folder_path):
JDCN_VHSFileMover.py
+25 -17
View File
@@ -10,27 +10,35 @@ def create_folder_if_not_exists(folder_path):
def move_it(source_path, destination_dir, overwrite=False):
try:
# Ensure destination directory exists
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) and overwrite:
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)
# print(f"Image moved from '{source_path}' to '{destination_path}'")
destination_path = os.path.join(destination_dir, filename)
# Check if the destination file exists
if os.path.exists(destination_path):
if overwrite:
# If overwrite is True, move the file directly
shutil.move(source_path, destination_path)
else:
# If overwrite is False, find a new filename with an incremented counter
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:
# If the destination file does not exist, move the file directly
shutil.move(source_path, destination_path)
except Exception as e:
print("")
# print(f"Error: {e}")
print(f"Error: {e}")
class JDCN_VHSFileMover: