python - 在pyyaml中表示具有相同基类的不同类的实例

Question

我有一组单元测试，想将每次测试运行的结果存储为 YAML 文件以供进一步分析。以 YAML 格式转储数据以多种方式满足我的需求。但是测试属于不同的花色，结果有不同的父类，这是我的例子:

>>> rz # shorthand for result
[<upstream_suite.fftest.SimpleTest testMethod=test_is_fsType_not_set>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_A>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_A_a_glyph_instance>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_ended_ufo>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_folder>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_fsType_eq_1>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_it_exists>]
# etc.

每个测试都可以有自己的父类。

>>> type(rz[0]) == type(rz[1])
False

我已经使用 __getstate__ 方法为所有类添加了我自己的基类，但由于某些原因它不起作用:

>>> rz[0].__getstate__()
{'targets': ['upstream'], 'methodDoc': 'Is the OS/2 table fsType set to 0?', 'methodName': 'test_is_fsType_not_set', 'tool': 'FontForge', 'name': 'upstream_suite.fftest'}
>>> yaml.safe_dump(rz[0])
*** RepresenterError: cannot represent an object: test_is_fsType_not_set (upstream_suite.fftest.SimpleTest)

文档说我可以为每个类创建代表，所以我试了一下并写了这段代码:

import yaml
from somepath import MyTestCase

def repr_testcase(dumper, data):
    return dumper.represent_mapping(u'tag:yaml.org,2002:map', {
        'methodDoc': data._testMethodDoc,
        'tool': data.tool,
        'name': data.name,
        'methodName': data._testMethodName,
        'targets': data.targets,
        })

# Register representer 
yaml.SafeDumper.add_representer(MyTestCase, repr_testcase)

同时:

>>> isinstance(rz[0], MyTestCase)
True

即使在那之后 safe_dump 也无法正常工作。但是如果我做了这个把戏，那么它就会起作用，我可以得到结果:

>>> yaml.SafeDumper.add_representer(type(rz[0]), repr_testcase)
>>> print(yaml.safe_dump(rz[0]))
methodDoc: Is the OS/2 table fsType set to 0?
methodName: test_is_fsType_not_set
name: checker.upstream_suite.fontforge_suite.fftest
targets: [upstream]
tool: FontForge

但这意味着我需要记录所有可能出现的情况，嗯……至少，这看起来并不明智。实际上，我真正的 result 是更复杂的字典、从不同地方收集的测试等。

>>> for x in rz: yaml.SafeDumper.add_representer(type(x), repr_testcase)
>>> yaml.safe_dump(rz)
... here is expected result

所以，问题是，我在 __getstate__ 上做错了什么，或者我如何使 pyyaml 与特定基类的所有实例一起工作，只有一个代表注册。

Answer 1

我找到了完成这项工作的方法。简短的回答是，还有另一种注册代表的方法，它称为 add_multi_representer。所以修复是:

# Change this line:
# yaml.SafeDumper.add_representer(MyTestCase, repr_testcase)

# to this line:
yaml.SafeDumper.add_multi_representer(MyTestCase, repr_testcase)

更新。多代表允许实现我想要实现的目标:为基类注册代表。