Question

想象一下，我有以下Traits对象：

from traits.api import Int, HasTraits, Instance

class Foo(HasTraits):
    a = Int(2)
    b = Int(5)

class Bar(HasTraits):
    c = Int(7)
    foo = Instance(Foo,())

Bar将允许我通过以下方式访问a上的Foo属性：

bar = Bar()
bar.foo.a
>>> 2

是否有标准方法可以将bar作为表单的嵌套字典返回：

print bar_as_dict 
>>> {'c':7, 'foo':{'a':2, 'b':5}}

我基本上试图提取特定类型的对象上的所有子特征。我们的用例是我们有深度嵌套的具有绘图特征的HasTrait对象，我们正在尝试动态查找特定对象上的所有绘图。我们有一个函数可以从嵌套字典中返回图，但我们需要传入HasTrait对象，因此将它们格式化为嵌套字典会很棒。如果有另一种方法来动态检查HasTraits对象的堆栈并返回某种类型的所有特征，那也可以。

Here's a link to the HasTraits API ......无法直接从中解决这个问题。

解决方案尝试

我已尝试使用.traits()方法，但会返回这些CTrait个对象。

print bar.traits()
>>>{'trait_added': <traits.traits.CTrait object at 0x8b7439c>,
 'c': <traits.traits.CTrait object at 0x8b29e9c>, 
'foo': <traits.traits.CTrait object at 0x8b29e44>, 
'trait_modified': <traits.traits.CTrait object at 0x8b74344>}

我不期待评价：

isinstance(bar.traits()['c'], int)
>>> False

但在Pieter的建议之后，这有效：

print bar.traits()['c'].is_trait_type(Int)
>>> True

现在的问题是如何以递归方式执行此操作。

Answer 1

我在following a similar question后对嵌套字典的递归没有特征值

进行了解释

def flatten_traitobject(traitobject, *types):
    node_map = {}
    node_path = [] 
    def nodeRecursiveMap(traitobject, node_path): 
        for key in traitobject.editable_traits():
            val = traitobject.get(key)[key]
            for type in types:
                if isinstance(val, types[0]):
                    node_map['.'.join(node_path + [key])] = val 
            try:
                nodeRecursiveMap(val, node_path + [key])
            except (AttributeError, TypeError):
                pass
    nodeRecursiveMap(traitobject, node_path)
    return node_map

测试它，我可以选择只保留整数和/或只保留Foo类型的对象

bar=Bar()
print 'RETAINING ONLY INTEGER TYPES'
print flatten_traitobject(bar, int)
print '\nRETAINTING ONLY FOO OBJECT TYPES'
print flatten_traitobject(bar, Foo)

>>> RETAINING ONLY INTEGER TYPES
>>> {'c': 7, 'foo.b': 5, 'foo.a': 2}

>>> RETAINTING ONLY FOO OBJECT TYPES
>>> {'foo': <__main__.Foo object at 0x8f9308c>}

Foo代表我追求的特殊绘图特征，所以我现在可以有效地返回它们。

Enthought Traits.HasTraits类作为嵌套字典？

解决方案尝试

1 个答案: