47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import os
|
|
import shutil
|
|
|
|
def find_replace(file_path, find_text, replace_text):
|
|
with open(file_path, 'r') as file:
|
|
content = file.read()
|
|
|
|
new_content = content.replace(find_text.replace('\\', '\\\\'), replace_text)
|
|
|
|
with open(file_path, 'w') as file:
|
|
file.write(new_content)
|
|
|
|
def delete_pycache_folders(start_dir='.'):
|
|
for root, dirs, files in os.walk(start_dir):
|
|
for dir_name in dirs:
|
|
if dir_name == '__pycache__':
|
|
folder_path = os.path.join(root, dir_name)
|
|
try:
|
|
shutil.rmtree(folder_path)
|
|
except Exception as e:
|
|
print("Error:",folder_path)
|
|
pass
|
|
|
|
def remove_file(file):
|
|
try:
|
|
os.remove(file)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
delete_pycache_folders()
|
|
|
|
remove_file("Fooocus/config.txt")
|
|
remove_file("Fooocus/modules/config.txt")
|
|
remove_file("percentage.txt")
|
|
remove_file("image.png")
|
|
|
|
folder_path = os.path.normpath("Fooocus/models/upscale_models")
|
|
if os.path.exists(folder_path):
|
|
shutil.rmtree(folder_path)
|
|
|
|
fooocus_dir = os.path.dirname(os.path.realpath(__file__)) + "\\Fooocus"
|
|
find_replace(fooocus_dir + "\\modules\\config.py", fooocus_dir + "\\config.txt", './config.txt')
|
|
find_replace(fooocus_dir + "\\modules\\config.py", fooocus_dir + "\\presets\\default.json", './presets/default.json')
|
|
|
|
os.system('git add .')
|
|
os.system('git commit -m "Update"')
|
|
os.system('git push --set-upstream origin main') |