[fix] OrderedLoader from SafeLoader

This commit is contained in:
ljf 2021-08-17 17:33:23 +02:00
parent ccb6dc54b1
commit 5518931c02

View file

@ -26,14 +26,14 @@ ACTIONSMAP_FILE = os.path.join(THIS_SCRIPT_DIR, "../data/actionsmap/yunohost.yml
def ordered_yaml_load(stream): def ordered_yaml_load(stream):
class OrderedLoader(yaml.Loader): class OrderedLoader(yaml.SafeLoader):
pass pass
OrderedLoader.add_constructor( OrderedLoader.add_constructor(
yaml.resolver.BaseResolver.DEFAULT_MAPPING_TAG, yaml.resolver.BaseResolver.DEFAULT_MAPPING_TAG,
lambda loader, node: OrderedDict(loader.construct_pairs(node)), lambda loader, node: OrderedDict(loader.construct_pairs(node)),
) )
return yaml.safe_load(stream, OrderedLoader) return yaml.load(stream, OrderedLoader)
def main(): def main():