fix: automatically add suffix to config-file

metaDMG-dev · Oct 17, 2022 · 75fcb19 · 75fcb19
1 parent bc233c7
commit 75fcb19
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/metaDMG/cli/cli_utils.py b/src/metaDMG/cli/cli_utils.py
@@ -422,6 +422,9 @@ def save_config_file(
                 print("Exiting")
                 raise typer.Abort()
 
+    if not ("yaml" in config_file.name or "yml" in config_file.name):
+        config_file = config_file.with_suffix(".yaml")
+
     with open(config_file, "w") as file:
         yaml.dump(config, file, sort_keys=False)
     if verbose: