aboutsummaryrefslogtreecommitdiff
path: root/modules/styles.py
diff options
context:
space:
mode:
authorpapuSpartan <30642826+papuSpartan@users.noreply.github.com>2023-05-03 02:21:50 -0500
committerpapuSpartan <30642826+papuSpartan@users.noreply.github.com>2023-05-03 02:21:50 -0500
commitf08ae961157d33051b5cd09ba5c77b779096ef69 (patch)
tree038e0f511d51816c0a971d6093da0420b5677ae0 /modules/styles.py
parentdff60e2e74964a8b02b75ecd8cf8007ef67a9712 (diff)
parent335428c2c8139dfe07ba096a6defa75036660244 (diff)
resolve merge conflicts and swap to dev branch for now
Diffstat (limited to 'modules/styles.py')
-rw-r--r--modules/styles.py12
1 files changed, 5 insertions, 7 deletions
diff --git a/modules/styles.py b/modules/styles.py
index 990d5623..9ed85991 100644
--- a/modules/styles.py
+++ b/modules/styles.py
@@ -72,16 +72,14 @@ class StyleDatabase:
return apply_styles_to_prompt(prompt, [self.styles.get(x, self.no_style).negative_prompt for x in styles])
def save_styles(self, path: str) -> None:
- # Write to temporary file first, so we don't nuke the file if something goes wrong
- fd, temp_path = tempfile.mkstemp(".csv")
+ # Always keep a backup file around
+ if os.path.exists(path):
+ shutil.copy(path, path + ".bak")
+
+ fd = os.open(path, os.O_RDWR|os.O_CREAT)
with os.fdopen(fd, "w", encoding="utf-8-sig", newline='') as file:
# _fields is actually part of the public API: typing.NamedTuple is a replacement for collections.NamedTuple,
# and collections.NamedTuple has explicit documentation for accessing _fields. Same goes for _asdict()
writer = csv.DictWriter(file, fieldnames=PromptStyle._fields)
writer.writeheader()
writer.writerows(style._asdict() for k, style in self.styles.items())
-
- # Always keep a backup file around
- if os.path.exists(path):
- shutil.move(path, path + ".bak")
- shutil.move(temp_path, path)