extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 09:57:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Verify a newly written config file before setting that as the actual config file. Helps prevent a corrupted write from overwriting the config file

Browse Source
This commit is contained in:
cmdr2 2023-06-30 13:22:23 +05:30
parent 8618708fd1
commit 5a643c383b
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ui/easydiffusion/app.py
View File

@ -160,10 +160,17 @@ def setConfig(config):
yaml.indent(mapping=2, sequence=4, offset=2) yaml.indent(mapping=2, sequence=4, offset=2)
try: try:
f = open(config_yaml_path, "w", encoding="utf-8") f = open(config_yaml_path + ".tmp", "w", encoding="utf-8")
yaml.dump(config, f) yaml.dump(config, f)
finally: finally:
f.close() # do this explicitly to avoid NUL bytes (rare bug when using 'with') f.close() # do this explicitly to avoid NUL bytes (possible rare bug when using 'with')
# verify that the new file is valid, and only then overwrite the old config file
# helps prevent the rare NUL bytes error from corrupting the config file
yaml = YAML()
with open(config_yaml_path + ".tmp", "r", encoding="utf-8") as f:
yaml.load(f)
shutil.move(config_yaml_path + ".tmp", config_yaml_path)
except: except:
log.error(traceback.format_exc()) log.error(traceback.format_exc())