yaml.load接受第二个参数,一个加载器类(默认情况下为yaml.loader.Loader).预定义的加载器是许多其他加载器的混搭:
class Loader(Reader, Scanner, Parser, Composer, Constructor, Resolver):
def __init__(self, stream):
Reader.__init__(self, stream)
Scanner.__init__(self)
Parser.__init__(self)
Composer.__init__(self)
Constructor.__init__(self)
Resolver.__init__(self)
Constructor类是执行数据类型转换的类.一种(kludgy,但快速)覆盖布尔转换的方法可能是:
from yaml.constructor import Constructor
def add_bool(self, node):
return self.construct_scalar(node)
Constructor.add_constructor(u'tag:yaml.org,2002:bool', add_bool)
它覆盖了构造函数用于将布尔标记数据转换为Python布尔值的函数.我们在这里做的只是逐字地返回字符串.
但是,这会影响所有YAML加载,因为您将覆盖默认构造函数的行为.更合适的方法是创建一个派生自Constructor的新类,以及新的Loader对象获取自定义构造函数.