[docs] Use yaml.safe_load() in build_*.py scripts (#2432)

Co-authored-by: itchyny <[email protected]>
jqlang · Jun 5, 2023 · d178ebd · d178ebd
1 parent b7511b9
commit d178ebd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docs/build_manpage.py b/docs/build_manpage.py
@@ -187,7 +187,7 @@ def __write_raw(self, dat):
 
 def load_yml_file(fn):
   with open(fn) as f:
-    return yaml.load(f, Loader=yaml.Loader)
+    return yaml.safe_load(f)
 
 def dedent_body(body):
   lines = [re.sub(r'^  (\S)', r'\1', l) for l in body.split('\n')]

diff --git a/docs/build_mantests.py b/docs/build_mantests.py
@@ -2,7 +2,7 @@
 import yaml
 
 with open("content/manual/manual.yml") as f:
-  manual = yaml.load(f)
+  manual = yaml.safe_load(f)
   for section in manual.get('sections', []):
     for entry in section.get('entries', []):
       for example in entry.get('examples', []):

diff --git a/docs/build_website.py b/docs/build_website.py
@@ -16,7 +16,7 @@
 
 def load_yml_file(fn):
   with open(fn) as f:
-    return yaml.load(f, Loader=yaml.Loader)
+    return yaml.safe_load(f)
 
 env.filters['search_id'] = lambda input: input.replace(r'`', '')
 env.filters['section_id'] = lambda input: re.sub(r"[^a-zA-Z0-9_]", '', input)