pyyaml 加载十进制数

Question

yaml.load 将数字加载为 Python 浮点数。我找不到直接的方法来覆盖它。

比较json.load，如果您想将浮点数解析为decimal.Decimal，则允许parse_float=Decimal。

有什么方法可以用 PyYAML 完成这个吗？或者 YAML 规范的某些属性不建议这样做？

Answer 1

你可以这样做：

def decimal_constructor(loader, node):
    value = loader.construct_scalar(node)
    return Decimal(value)

yaml.add_constructor(u'!decimal', decimal_constructor)

这允许您加载小数，但是 仅当它们在 YAML 文档中以 !decimal 标记为前缀时 。但是，您可以使用自定义 Resolver 将所有数字解析为 !decimal:

class MyResolver(BaseResolver):
    pass

MyResolver.add_implicit_resolver(
        '!decimal',
        re.compile(r'''^(?:[-+]?(?:[0-9][0-9_]*)\.[0-9_]*(?:[eE][-+][0-9]+)?
                    |\.[0-9_]+(?:[eE][-+][0-9]+)?
                    |[-+]?[0-9][0-9_]*(?::[0-5]?[0-9])+\.[0-9_]*
                    |[-+]?\.(?:inf|Inf|INF)
                    |\.(?:nan|NaN|NAN))$''', re.X),
        list('-+0123456789.'))

您应该从 default resolver 复制其他隐式解析器。然后，您需要定义一个使用解析器的 Loader。

class MyLoader(Reader, Scanner, Parser, Composer, SafeConstructor, MyResolver):
    def __init__(self, stream):
        Reader.__init__(self, stream)
        Scanner.__init__(self)
        Parser.__init__(self)
        Composer.__init__(self)
        SafeConstructor.__init__(self)
        MyResolver.__init__(self)

上面，我们将您的构造函数添加到默认加载程序。将该行更改为：

yaml.add_constructor(u'!decimal', decimal_constructor, MyLoader)

最后，使用

加载 YAML

yaml.load(stream, MyLoader)

pyyaml 加载十进制数

pyyaml load number as decimal

python

yaml

pyyaml