10

使用此数据结构：

d = {
    (2,3,4): {
        'a': [1,2], 
        'b': 'Hello World!',
        'c': 'Voilà!'
    }
}

我想得到这个 YAML：

%YAML 1.2
---
[2,3,4]:
  a:
    - 1
    - 2
  b: Hello World!
  c: 'Voilà!'

不幸的是，我得到了这种格式：

$ print ruamel.yaml.dump(d, default_flow_style=False, line_break=1, explicit_start=True, version=(1,2))
%YAML 1.2
---
? !!python/tuple
- 2
- 3
- 4
: a:
  - 1
  - 2
  b: Hello World!
  c: !!python/str 'Voilà!'

即使使用 . 我也无法配置我想要的输出safe_dump。如果不对输出进行手动正则表达式工作，我该如何做到这一点？

我发现的唯一丑陋的解决方案是：

def rep(x):
    return repr([int(y) for y in re.findall('^\??\s*-\s*(\d+)', x.group(0), re.M)]) + ":\n"
print re.sub('\?(\s*-\s*(\w+))+\s*:', rep, 
    ruamel.yaml.dump(d, default_flow_style=False, line_break=1, explicit_start=True, version=(1,2)))
4

1 回答 1

11

新的 ruamel.yaml API

您无法使用 获得想要的东西ruamel.yaml.dump()，但是使用具有更多控件的新 API，您可以非常接近。

import sys
import ruamel.yaml


d = {
    (2,3,4): {
        'a': [1,2], 
        'b': 'Hello World!',
        'c': 'Voilà!'
    }
}

def prep(d):
    if isinstance(d, dict):
        needs_restocking = False
        for idx, k in enumerate(d):
            if isinstance(k, tuple):
                needs_restocking = True
            try:
                if 'à' in d[k]:
                    d[k] = ruamel.yaml.scalarstring.SingleQuotedScalarString(d[k])
            except TypeError:
                pass
            prep(d[k])
        if not needs_restocking:
            return
        items = list(d.items())
        for (k, v) in items:
            d.pop(k)
        for (k, v) in items:
            if isinstance(k, tuple):
                k = ruamel.yaml.comments.CommentedKeySeq(k)
            d[k] = v
    elif isinstance(d, list):
        for item in d:
            prep(item)

yaml = ruamel.yaml.YAML()
yaml.indent(mapping=2, sequence=4, offset=2)
yaml.version = (1, 2)
data = prep(d)
yaml.dump(d, sys.stdout)

这使：

%YAML 1.2
---
[2, 3, 4]:
  a:
    - 1
    - 2
  b: Hello World!
  c: 'Voilà!'

仍然没有简单的方法来抑制序列项之前的空格，因此如果不[2,3,4]付出[2, 3, 4]很大的努力就无法安装。

原答案：

ruamel.yaml.dump()如果不对内部进行重大返工，您将无法准确获得您想要的输出 。

  • 您喜欢的输出具有缩进 2 用于顶级映射的值（键a、b等）和缩进 4 用于作为键值的序列元素a（-在 2 个位置被推入。那至少需要区分映射和序列的缩进级别（如果不是针对单个集合），这是不平凡的。
  • 您的序列输出从,（逗号，空格）“正常”流样式发出的内容压缩为,. IIRC 这目前不受任何参数的影响，并且由于您在发出集合时几乎没有上下文知识，因此很难“在发出作为键的序列时不包括空格”。另一个选项dump()需要更改几个源文件和类。

不太困难的问题，并附有解决方案的指示：

  • 您的 tuple 必须神奇地转换为序列才能摆脱 tag !!python/tuple。由于您不想影响所有元组，因此 IMO 最好通过创建一个子类tuple并将其表示为序列来完成（仅当实际用作键时，才可选择将此类元组表示为列表）。您可以使用comments.CommentedKeySeq它（假设ruamel.yaml>=0.12.14它在使用时具有适当的表示支持ruamel.yaml.round_trip_dump()
  • 您的关键是，在发射前测试时，不是一个简单的关键，因此它得到一个'？'（问号，空格）表示一个复杂的映射键。. 您必须更改发射器，以便SequenceStartEvent启动一个简单的键（如果它具有流样式而不是块样式）。另一个问题是，这样的 SequenceStartEvent 然后将被“测试”以具有style属性（这可能表明明确需要“？”键）。这需要改变emitter.py:Emitter.check_simple_key()和emitter.py:Emitter.expect_block_mapping_key()。
  • 你的标量字符串值得c到引号，而你的标量字符串值b没有。您只能通过将它们设置为不同的类型，才能在 ruamel.yaml 中获得这种输出差异。例如通过使其类型scalarstring.SingleQuotedScalarString()（并使用round_trip_dump()）。

如果你这样做：

import sys
import ruamel.yaml
from ruamel.yaml.comments import CommentedMap, CommentedKeySeq
assert ruamel.yaml.version_info >= (0, 12, 14)

data = CommentedMap()
data[CommentedKeySeq((2, 3, 4))] = cm = CommentedMap()
cm['a'] = [1, 2]
cm['b'] = 'Hello World!'
cm['c'] = ruamel.yaml.scalarstring.SingleQuotedScalarString('Voilà!')

ruamel.yaml.round_trip_dump(data, sys.stdout, explicit_start=True, version=(1, 2))

你会得到：

%YAML 1.2
---
[2, 3, 4]:
  a:
  - 1
  - 2
  b: Hello World!
  c: 'Voilà!'

除了现在一致的缩进级别 2 之外，流样式序列中的额外空格以及所需的 使用round_trip_dump，将使您尽可能接近您想要的，而无需进行重大返工。

上面的代码是否丑陋当然是个人喜好问题。

当使用ruamel.yaml.round_trip_load(preserve_quotes=True).

如果不需要控制引号，并且映射键的顺序也不重要，那么您也可以修补正常的转储程序：

def my_key_repr(self, data):
    if isinstance(data, tuple):
        print('data', data)
        return self.represent_sequence(u'tag:yaml.org,2002:seq', data,
                                       flow_style=True)
    return ruamel.yaml.representer.SafeRepresenter.represent_key(self, data)

ruamel.yaml.representer.Representer.represent_key = my_key_repr

然后你可以使用正常的序列：

data = {}
data[(2, 3, 4)] = cm = {}
cm['a'] = [1, 2]
cm['b'] = 'Hello World!'
cm['c'] = 'Voilà!'

ruamel.yaml.dump(data, sys.stdout, allow_unicode=True, explicit_start=True, version=(1, 2))

会给你：

%YAML 1.2
---
[2, 3, 4]:
  a: [1, 2]
  b: Hello World!
  c: Voilà!

请注意，您需要round_trip_dump()使用allow_unicode=True.

¹免责声明：我是ruamel.yaml的作者。

于 2016-09-21T08:15:00.170 回答