将 YAML 多行值转换为折叠块标量样式？

Question

使用ruamel.yaml我试图获得某种风格的 YAML，更具体地说，单行字符串:与使用折叠标量样式 ( |/ |-) 的多行字符串在同一行开始并且行被限制为一定数量的字符（换行）。

到目前为止，我的尝试深受来源中调用walk_tree的类似函数的影响：

#!/usr/bin/env python

import ruamel.yaml

from ruamel.yaml.scalarstring import ScalarString, PreservedScalarString

def walk_tree(base):
    from ruamel.yaml.compat import string_types

    if isinstance(base, dict):
        for k in base:
            v = base[k]
            if isinstance(v, string_types):
                v = v.replace('\r\n', '\n').replace('\r', '\n').strip()
                base[k] = ScalarString(v) if '\n' in v else v
            else:
                walk_tree(v)
    elif isinstance(base, list):
        for idx, elem in enumerate(base):
            if isinstance(elem, string_types) and '\n' in elem:
                print(elem) # @Anthon: this print is in the original code as well
                base[idx] = preserve_literal(elem)
            else:
                walk_tree(elem)

with open("input.yaml", "r") as fi:
    inp = fi.read()

loader=ruamel.yaml.RoundTripLoader
data = ruamel.yaml.load(inp, loader)

walk_tree(data)

dumper = ruamel.yaml.RoundTripDumper

with open("output.yaml", "w") as fo:
    ruamel.yaml.dump(data, fo, Dumper=dumper, allow_unicode=True)

但后来我得到一个例外：ruamel.yaml.representer.RepresenterError: cannot represent an object: …. 我没有得到任何异常，如果我更换ScalarString与PreservedScalarString作为是在原有的情况下walk_tree代码，但后来我又得到了文字块这是不是我想要的。

那么如何修复我的代码以使其正常工作？

Answer 1

该类ScalarString是 的基类LiteralScalarString，正如您发现的那样，它没有代表者。您应该将其设为/保留为 Python 字符串，因为它会适当地处理特殊字符（引用需要引用以符合 YAML 规范的字符串）。

假设您有这样的输入：

- 1
- abc: |
    this is a short string scalar with a newline
    in it
- "there are also a multiline\nsequence element\nin this file\nand it is longer"

您可能想做类似的事情：

import ruamel.yaml
from ruamel.yaml.scalarstring import LiteralScalarString, preserve_literal


def walk_tree(base):
    from ruamel.yaml.compat import string_types

    def test_wrap(v):
        v = v.replace('\r\n', '\n').replace('\r', '\n').strip()
        return v if len(v) < 72 else preserve_literal(v)

    if isinstance(base, dict):
        for k in base:
            v = base[k]
            if isinstance(v, string_types) and '\n' in v:
                base[k] = test_wrap(v)
            else:
                walk_tree(v)
    elif isinstance(base, list):
        for idx, elem in enumerate(base):
            if isinstance(elem, string_types) and '\n' in elem:
                base[idx] = test_wrap(elem)
            else:
                walk_tree(elem)

yaml = YAML()

with open("input.yaml", "r") as fi:
    data = yaml.load(fi)

walk_tree(data)

with open("output.yaml", "w") as fo:
    yaml.dump(data, fo)

获得输出：

- 1
- abc: "this is a short string scalar with a newline\nin it"
- |-
  there are also a multiline
  sequence element
  in this file
  and it is longer

一些注意事项：

• 使用LiteralScalarString优于PreservedScalarString。后者的名称是它是唯一保留的字符串类型时的残余。
• 您可能没有包含字符串的序列元素，因为您没有 import preserve_literal，尽管它仍然在复制的代码中使用。
• 我将“包装”代码提取到 test_wrap 中，供值和元素包装使用，最大行长度设置为 72 个字符。
• 该值data[1]['abc']加载为LiteralScalarString. 如果要保留现有的文字样式字符串标量，则应在测试 type 之前测试这些标量string_types。
• 我将新的 API 与一个实例一起使用YAML()
• 如果您将示例中的 72 增加到默认值 80 以上，则可能需要将该width属性设置为 1000 之类的值，以防止自动换行。( yaml.width = 1000)