util: Use libyaml if possible

managarm · Nov 4, 2024 · 2a5905c · 2a5905c
1 parent a181735
commit 2a5905c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/y4/__main__.py b/y4/__main__.py
@@ -55,7 +55,14 @@ def main():
                     ctx.bind(tag, binding)
             else:
                 out = ctx.evaluate(root)
-                sys.stdout.write(yaml.dump(out, sort_keys=False, explicit_start=True))
+                sys.stdout.write(
+                    yaml.dump(
+                        out,
+                        sort_keys=False,
+                        explicit_start=True,
+                        Dumper=util.YamlDumper,
+                    )
+                )
 
 
 if __name__ == "__main__":

diff --git a/y4/util.py b/y4/util.py
@@ -4,7 +4,15 @@
 YAML_MAP_TAG = "tag:yaml.org,2002:map"
 YAML_STR_TAG = "tag:yaml.org,2002:str"
 
-YamlLoader = yaml.SafeLoader
+try:
+    YamlLoader = yaml.CSafeLoader
+except AttributeError:
+    YamlLoader = yaml.SafeLoader
+
+try:
+    YamlDumper = yaml.CSafeDumper
+except AttributeError:
+    YamlDumper = yaml.SafeDumper
 
 
 class InternalNode(yaml.Node):