lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-04 20:49:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

update some codes related to win32

Browse Source
This commit is contained in:
lllyasviel
2024-08-21 03:23:34 -07:00
committed by GitHub
parent 3e253012b5
commit c73ff3724c
2 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
modules_forge/forge_space.py
View File

@@ -43,25 +43,36 @@ def find_free_port(server_name, start_port=None):
port += 1
def long_path_prefix(path):
if os.name == 'nt' and not path.startswith("\\\\?\\") and not os.path.exists(path):
return f"\\\\?\\{path}"
return path
def remove_dir(dir_path):
dir_path = long_path_prefix(dir_path)
for root, dirs, files in os.walk(dir_path, topdown=False):
for name in files:
file_path = os.path.join(root, name)
file_path = long_path_prefix(file_path)
try:
os.remove(os.path.join(root, name))
os.remove(file_path)
except Exception as e:
print(f"Error removing file {os.path.join(root, name)}: {e}")
print(f"Error removing file {file_path}: {e}")
for name in dirs:
dir_to_remove = os.path.join(root, name)
dir_to_remove = long_path_prefix(dir_to_remove)
try:
os.rmdir(os.path.join(root, name))
os.rmdir(dir_to_remove)
except Exception as e:
print(f"Error removing directory {os.path.join(root, name)}: {e}")
print(f"Error removing directory {dir_to_remove}: {e}")
try:
os.rmdir(dir_path)
print(f"Deleted: {dir_path}")
except:
print(f'Something went wrong when trying to delete a folder. You may try to manually delete the folder [{dir_path}].')
except Exception as e:
print(f"Error removing directory {dir_path}: {e}. You may try to manually delete the folder.")
return