fix: ✨ less strict csv parsing

closes #122
2023-11-08 18:28:27 +01:00
parent 87e301d120
commit d5c4c5f264
1 changed files with 13 additions and 6 deletions
@@ -1,5 +1,4 @@
-import csv
-import shutil
+import csv, shutil
 from pathlib import Path

 import folder_paths
@@ -157,13 +156,21 @@ class StylesLoader:
            for file in files:
                with open(file, "r", encoding="utf8") as f:
                    parsed = csv.reader(f)
-                    for row in parsed:
+                    for i, row in enumerate(parsed):
                        log.debug(f"Adding style {row[0]}")
                        try:
-                            cls.options[row[0]] = (row[1], row[2])
-                        except Exception:
+                            name, positive, negative = (row + [None] * 3)[:3]
+                            positive = positive or ""
+                            negative = negative or ""
+                            if name is not None:
+                                cls.options[name] = (positive, negative)
+                            else:
+                                # Handle the case where 'name' is None
+                                log.warning(f"Missing 'name' in row {i}.")
+
+                        except Exception as e:
                            log.warning(
-                                f"There was an error while parsing {file}, make sure it respects A1111 format, i.e 3 columns name, positive, negative"
+                                f"There was an error while parsing {file}, make sure it respects A1111 format, i.e 3 columns name, positive, negative:\n{e}"
                            )
                            continue