Browse Source

yaml.safe_load to avoid warning from pyyaml

See https://pyyaml.docsforge.com/master/documentation/#loading-yaml
pull/269/head
Mathieu Bernard 4 years ago
parent
commit
01b8460b39
  1. 2
      patacrep/build.py
  2. 2
      patacrep/content/__init__.py
  3. 2
      patacrep/content/include.py
  4. 4
      patacrep/songbook/__init__.py
  5. 3
      patacrep/templates.py
  6. 2
      test/test_book/test_compilation.py
  7. 4
      test/test_content/test_content.py

2
patacrep/build.py

@ -380,6 +380,6 @@ def config_model(key):
""" """
model_path = pkg_datapath('templates', 'songbook_model.yml') model_path = pkg_datapath('templates', 'songbook_model.yml')
with encoding.open_read(model_path) as model_file: with encoding.open_read(model_path) as model_file:
data = yaml.load(model_file) data = yaml.safe_load(model_file)
return data.get(key, {}) return data.get(key, {})

2
patacrep/content/__init__.py

@ -238,7 +238,7 @@ def validate_parser_argument(raw_schema):
Will raise `ContentError` if the schema is not respected. Will raise `ContentError` if the schema is not respected.
""" """
schema = Rx.make_schema(yaml.load(raw_schema)) schema = Rx.make_schema(yaml.safe_load(raw_schema))
def wrap(parse): def wrap(parse):
"""Wrap the parse function""" """Wrap the parse function"""

2
patacrep/content/include.py

@ -74,7 +74,7 @@ def parse(keyword, config, argument):
filepath, filepath,
encoding=config['book']['encoding'] encoding=config['book']['encoding']
) as content_file: ) as content_file:
new_content = yaml.load(content_file) new_content = yaml.safe_load(content_file)
except Exception as error: # pylint: disable=broad-except except Exception as error: # pylint: disable=broad-except
new_contentlist.append_error(ContentError( new_contentlist.append_error(ContentError(
keyword="include", keyword="include",

4
patacrep/songbook/__init__.py

@ -26,13 +26,13 @@ def open_songbook(filename):
try: try:
with patacrep.encoding.open_read(filename) as songbook_file: with patacrep.encoding.open_read(filename) as songbook_file:
user_songbook = yaml.load(songbook_file) user_songbook = yaml.safe_load(songbook_file)
if 'encoding' in user_songbook.get('book', []): if 'encoding' in user_songbook.get('book', []):
with encoding.open_read( with encoding.open_read(
filename, filename,
encoding=user_songbook['book']['encoding'] encoding=user_songbook['book']['encoding']
) as songbook_file: ) as songbook_file:
user_songbook = yaml.load(songbook_file) user_songbook = yaml.safe_load(songbook_file)
except Exception as error: # pylint: disable=broad-except except Exception as error: # pylint: disable=broad-except
raise patacrep.errors.SongbookError(str(error)) raise patacrep.errors.SongbookError(str(error))

3
patacrep/templates.py

@ -240,7 +240,8 @@ class TexBookRenderer(Renderer):
variables[templatename] = {} variables[templatename] = {}
for variables_string in match: for variables_string in match:
try: try:
variables[templatename].update(yaml.load(variables_string)) variables[templatename].update(
yaml.safe_load(variables_string))
except ValueError as exception: except ValueError as exception:
raise errors.TemplateError( raise errors.TemplateError(
exception, exception,

2
test/test_book/test_compilation.py

@ -166,7 +166,7 @@ class FileTest(unittest.TestCase, metaclass=dynamic.DynamicTest):
"""Compile songbook "on the fly": without a physical songbook file.""" """Compile songbook "on the fly": without a physical songbook file."""
with open(base + ".yaml", mode="r", encoding="utf8") as sbfile: with open(base + ".yaml", mode="r", encoding="utf8") as sbfile:
sbyaml = yaml.load(sbfile) sbyaml = yaml.safe_load(sbfile)
outputdir = os.path.dirname(base) outputdir = os.path.dirname(base)
outputname = os.path.basename(base) outputname = os.path.basename(base)

4
test/test_content/test_content.py

@ -47,7 +47,7 @@ class FileTest(unittest.TestCase, metaclass=dynamic.DynamicTest):
"""Test that `base.source` produces the correct file list""" """Test that `base.source` produces the correct file list"""
sourcename = "{}.source".format(base) sourcename = "{}.source".format(base)
with open(sourcename, mode="r", encoding="utf8") as sourcefile: with open(sourcename, mode="r", encoding="utf8") as sourcefile:
sbcontent = yaml.load(sourcefile) sbcontent = yaml.safe_load(sourcefile)
outputdir = os.path.dirname(base) outputdir = os.path.dirname(base)
config = cls._generate_config(sbcontent, outputdir, base) config = cls._generate_config(sbcontent, outputdir, base)
@ -60,7 +60,7 @@ class FileTest(unittest.TestCase, metaclass=dynamic.DynamicTest):
if not os.path.exists(controlname): if not os.path.exists(controlname):
raise Exception("Missing control:" + str(controlname).replace("'", '"')) raise Exception("Missing control:" + str(controlname).replace("'", '"'))
with open(controlname, mode="r", encoding="utf8") as controlfile: with open(controlname, mode="r", encoding="utf8") as controlfile:
controllist = yaml.load(controlfile) controllist = yaml.safe_load(controlfile)
self.assertEqual(controllist, sourcelist) self.assertEqual(controllist, sourcelist)

Loading…
Cancel
Save