Question

我有一个类是标准字典的子类：

class Result(dict):

    """ Dict-like object with special methods """

    def content(self):
        return self.__getitem__('_content')

    def attrs(self):
        return self.__getitem__('_attrs')

此对象中的示例表示：

{
    '_attrs': {'id': 1},
    'description': 'testtest',
    'calories': 1234,
    '_content': 'Sample content',
    'name': 'qwerty',
    'price': 12390
}

我希望我的班级在迭代时跳过带有下划线键的记录。

# data is Result()
>>> for item in data:
>>>    print(item)

'description'
'calories'
'name'
'price'

我怎样才能做到这一点？

更新除了正确的答案，我还覆盖了keys（）和items（）方法来隐藏下划线键，即使这些方法将在迭代中使用：

def __iter__(self):
    for key in self.keys():
        yield key

def keys(self):
    return [key for key in super().keys() if key not in ['_attrs', '_content']]

def items(self):
    return [(key, value) for key, value in super().items() if key not in ['_attrs', '_content']]

Answer 1

只需实施__iter__即可。使用.keys()非常重要，以避免无限递归：

class Result(dict):
    def __iter__(self):
        for key in self.keys():
            if not(isinstance(key, str) and key.startswith("_")):
                yield key

然后在迭代中跳过它：

>>> r=Result({1:1,"_foo":2, "bar":3})
>>> for item in r:
...     print(item)
...
1
bar

Answer 2

只需使用startswith方法。

>>> x = {'_content':3, 'data':5, '_roarr':'chen'}
>>> for i in x.keys():
...     if i.startswith('_'): pass
...     else: print(x[i])
... 
5

dict子类中的自定义迭代行为

2 个答案: