python - pyyaml safe_load : how to ignore local tags

Question

我正在使用 yaml.safe_load() 但我需要忽略标签 !v2 - 有没有办法做到这一点但仍然使用 safe_load () ?

Answer 1

扩展现有答案以支持忽略所有未知标签。

import yaml

class SafeLoaderIgnoreUnknown(yaml.SafeLoader):
    def ignore_unknown(self, node):
        return None 

SafeLoaderIgnoreUnknown.add_constructor(None, SafeLoaderIgnoreUnknown.ignore_unknown)

root = yaml.load(content, Loader=SafeLoaderIgnoreUnknown)