fix(etl): Split memory usage

[csit.git] / docs / model / current / schema / yaml2json.py

diff --git a/docs/model/current/schema/yaml2json.py b/docs/model/current/schema/yaml2json.py
index 6927bc0..6899928 100644 (file)
--- a/docs/model/current/schema/yaml2json.py
+++ b/docs/model/current/schema/yaml2json.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Cisco and/or its affiliates.
+# Copyright (c) 2023 Cisco and/or its affiliates.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at:
@@ -11,10 +11,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Utility to convert from .schema.yaml to .schema.json.
-
-TODO: Read the input file name from command line argument.
-"""
+"""Utility to convert from .schema.yaml to .schema.json."""
 
 import glob
 import json
@@ -23,7 +20,5 @@ import yaml
 
 for filename in glob.glob(u"*.schema.yaml"):
     name = filename[:-5]
-    with open(f"{name}.yaml", u"rt") as fin:
-        data = yaml.load(fin.read())
-    with open(f"{name}.json", u"wt") as fout:
-        json.dump(data, fout, indent=2)
+    with open(f"{name}.yaml", u"r") as fin, open(f"{name}.json", u"w") as fout:
+        json.dump(yaml.safe_load(fin.read()), fout, indent=2)