fix deployment scripts (#321)

* Use safe_load and context managers for file handling in render.py (#1)

Co-authored-by: Cursor Agent <[email protected]>

* Changes from background composer bc-925563b9-8fb2-4df9-b99e-ebd3342eb01e (#2)

Co-authored-by: Cursor Agent <[email protected]>

---------

Co-authored-by: Cursor Agent <[email protected]>

Author: caesar0301

deploy/render.py

@@ -29,7 +29,8 @@ def scan_core_data(core_dir, validate_link=False):
             data_file = os.path.join(core_dir, category, data_item)
 
             try:
-                data_obj = yaml.load(open(data_file))
+                with open(data_file, 'r') as f:
+                    data_obj = yaml.safe_load(f)
                 data_obj["_rawFileName"] = data_item
             except Exception as e:
                 raise RuntimeError("Failed to read YAML data: {}".format(e))
@@ -75,7 +76,8 @@ def do_validate_link(link):
 
     categories = scan_core_data(core_dir, validate_link=True)
 
-    rendered = Template(open(template_file).read()).render(categories=categories)
+    with open(template_file, 'r') as f:
+        rendered = Template(f.read()).render(categories=categories)
     with open(os.path.join(pdir, "index.rst"), "w") as of:
         of.write(rendered)
         of.write("\n")

tools/parse_rst.py

@@ -7,7 +7,7 @@
 from ruamel.yaml import YAML
 
 # Load template
-template = yaml.load(open('dataset.yml.template', 'rb').read())
+template = yaml.load(open('dataset.yml.template', 'rb').read(), Loader=yaml.Loader)
 
 category = None
 categories = {}