monorail
/
Archipelago
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Generate: disallow duplicate mapping keys in input files

This commit is contained in:
Fabian Dill 2022-01-19 04:26:25 +01:00
parent 5df4d2f2fd
commit b4f17e67d0
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Utils.py
View File

@ -26,7 +26,7 @@ class Version(typing.NamedTuple):
__version__ = "0.2.3" __version__ = "0.2.3"
version_tuple = tuplize_version(__version__) version_tuple = tuplize_version(__version__)
from yaml import load, dump, safe_load from yaml import load, dump, SafeLoader
try: try:
from yaml import CLoader as Loader from yaml import CLoader as Loader
@ -117,7 +117,19 @@ def open_file(filename):
subprocess.call([open_command, filename]) subprocess.call([open_command, filename])
parse_yaml = safe_load # from https://gist.github.com/pypt/94d747fe5180851196eb#gistcomment-4015118 with some changes
class UniqueKeyLoader(SafeLoader):
def construct_mapping(self, node, deep=False):
mapping = set()
for key_node, value_node in node.value:
key = self.construct_object(key_node, deep=deep)
if key in mapping:
raise KeyError(f"Duplicate key {key!r} found in YAML. Already found keys: {mapping}.")
mapping.add(key)
return super().construct_mapping(node, deep)
parse_yaml = functools.partial(load, Loader=UniqueKeyLoader)
unsafe_parse_yaml = functools.partial(load, Loader=Loader) unsafe_parse_yaml = functools.partial(load, Loader=Loader)