X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=docs%2Fmodel%2Fcurrent%2Fschema%2Fyaml2json.py;h=6899928847d2cf76e98ea3dbdf04d75a338f1b97;hb=5a5f05618e7c1fd9e7983a90f0ba75b492e548d9;hp=6927bc0172e1601c7a6ec2d192a5e4d3d1179186;hpb=01d8f262afc567c3d49a23c3cb2cdeaced8a6887;p=csit.git

diff --git a/docs/model/current/schema/yaml2json.py b/docs/model/current/schema/yaml2json.py
index 6927bc0172..6899928847 100644
--- 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)