From f183549e7722a87f88957a37fe5f84f2f13e5b04 Mon Sep 17 00:00:00 2001
From: Maximilian Moser <maximilian.moser@tuwien.ac.at>
Date: Mon, 26 Feb 2024 12:41:46 +0100
Subject: [PATCH] Allow a simple list of formats in `store_formats()`
---
formatscaper/core/utils.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/formatscaper/core/utils.py b/formatscaper/core/utils.py
index e8d60f3..7ff9c8d 100644
--- a/formatscaper/core/utils.py
+++ b/formatscaper/core/utils.py
@@ -40,10 +40,13 @@ def load_formats(file_name: str) -> Dict[str, Format]:
return formats
-def store_formats(formats: Dict[str, Format], file_name: str) -> bool:
+def store_formats(formats: Dict[str, Format] | List[Format], file_name: str) -> bool:
"""Store the known formats to the given YAML file."""
try:
- updated_formats = [f.as_dict() for f in formats.values()]
+ if isinstance(formats, dict):
+ formats = formats.values()
+
+ updated_formats = [f.as_dict() for f in formats]
with open(file_name, "w") as formats_file:
yaml.dump(updated_formats, formats_file, sort_keys=False)
--
GitLab