Fix writing yaml to strings

Author: raboof

gateway/gateway.py

@@ -7,6 +7,7 @@
 
 import os
 from datetime import date, timedelta
+from io import StringIO
 from pathlib import Path
 from typing import Dict, NotRequired, TypedDict
 
@@ -74,6 +75,11 @@ def write_yaml(path: Path, yaml_dict: dict | list):
         yaml = ruyaml.YAML()
         yaml.dump(yaml_dict, file)
 
+def to_yaml_string(yaml_dict: dict | list):
+    yaml = ruyaml.YAML()
+    stream = StringIO()
+    yaml.dump(yaml_dict, stream)
+    return stream.getvalue()
 
 def write_str(path: Path, content: str):
     with open(path, "w") as file:
@@ -190,11 +196,9 @@ def update_actions(dummy_path: Path, actions_path: Path):
     actions: ActionsYAML = load_yaml(actions_path)
 
     update_refs(steps, actions)
-    yaml = ruyaml.YAML()
-    gha_print(yaml.dump(actions), "Generated List")
+    gha_print(to_yaml_string(actions), "Generated List")
     write_yaml(actions_path, actions)
 
-
 def create_pattern(actions: ActionsYAML) -> list[str]:
     """
     Create a pattern list of valid action references.
@@ -228,8 +232,7 @@ def update_patterns(pattern_path: Path, list_path: Path):
     actions: ActionsYAML = load_yaml(list_path)
     patterns = create_pattern(actions)
     comment = f"# This file was generated from {pattern_path} by gateway/gateway.py. DO NOT UPDATE MANUALLY.\n"
-    yaml = ruyaml.YAML()
-    patterns_str = comment + yaml.safe_dump(patterns)
+    patterns_str = comment + to_yaml_string(patterns)
     gha_print(patterns_str, "Generated Patterns")
     write_str(pattern_path, patterns_str)
 
@@ -283,6 +286,5 @@ def clean_actions(actions_path: Path):
     """
     actions: ActionsYAML = load_yaml(actions_path)
     remove_expired_refs(actions)
-    yaml = ruyaml.YAML()
-    gha_print(yaml.safe_dump(actions), "Cleaned Actions")
+    gha_print(to_yaml_string(actions), "Cleaned Actions")
     write_yaml(actions_path, actions)