首页 > 解决方案 > 如何将嵌套的字典键作为类的属性返回?

问题描述

我的班级中有一个函数:getNested(),它将一堆数据作为嵌套字典获取。将这些数据转换为我可以在实例化类时使用的属性的最佳实践是什么?例如; 在下面的示例中运行 test.nestedDict.A 行将理想地返回 {'Aa': ['1', '2', '3',], 'Ab':'item'}。

class MyClass( object ):
    def __init__( self ):
        self._nestedDict = None
        self.getNested()
    
    def getNested( self ):
        self._nestedDict = {'A':{'Aa': ['1', '2', '3',], 'Ab':'item'}, 'B':{'Ba': ['4', '5', '6',], 'Bb':'item2'} }
        
    @property    
    def nestedDict( self ):
        return self._nestedDict
        
test = MyClass()

test.nestedDict
# Result: {'A': {'Aa': ['1', '2', '3'], 'Ab': 'item'},'B': {'Ba': ['4', '5', '6'], 'Bb': 'item2'}} # 
test.nestedDict['A']
# Result: {'Aa': ['1', '2', '3'], 'Ab': 'item'} # 
test.nestedDict.A
# Error: AttributeError: line 1: 'dict' object has no attribute 'A' # 

标签: pythonpython-2.7classattributes

解决方案


实现您想要的一种方法是定义和使用一个从dict. 然后,在这个类中,将嵌套字典的键设置为类的属性。

这看起来像:

class Nested(dict):
    def __init__(self, dict_):
        super(Nested, self).__init__()
        for k, v in dict_.items():
            if isinstance(v, dict):
                v = Nested(v)
            setattr(self, k, v)
            self.update({k: v})


class MyClass:
    def __init__(self):
        self.set_nested()

    def set_nested(self):
        nested_dict = {'A': {'Aa': ['1', '2', '3'], 'Ab': 'item'},
                       'B': {'Ba': ['4', '5', '6'], 'Bb': 'item2'}}
        self._nested_dict = Nested(nested_dict)

    @property
    def nested_dict( self ):
        return self._nested_dict

然后你可以这样做:

>>> test = MyClass()
>>> test.nested_dict
{'A': {'Aa': ['1', '2', '3'], 'Ab': 'item'}, 'B': {'Ba': ['4', '5', '6'], 'Bb': 'item2'}}
>>> test.nested_dict.A
{'Aa': ['1', '2', '3'], 'Ab': 'item'}
>>> test.nested_dict.A.Aa
['1', '2', '3']
>>> test.nested_dict.A.Ab
'item'
>>> test.nested_dict['A']
{'Aa': ['1', '2', '3'], 'Ab': 'item'}

请注意,我允许自己更改变量和方法的名称以符合PEP8 样式


推荐阅读