python 类中的 __getitem__方法

经常会遇到python类中遇到一些特殊的方法,最常见的就是__ini__方法,想必大家都遇到过,今天来谈谈另外一种非常好用的类方法--__getitem__

来点干货, __getitem__的作用是什么呢?说白了就是类中一个特殊方法,类对象可以像字典对象那样根据key取值(dict['key']),如类对象Object['key'],系统会自动调用__getitem__方法,然后返回该方法定义return值,说到这里,有一点需要强调哦, 字典取值时key不存在,会抛出异常,而类对象的key是否存在,都会调用__getitem__方法并返回其规定的值,下面进入实例。

test = {
    'key1': 'value1',
    'key2': 'value2',
    'key3': 'value3',
    'key4': 'value4'
}

class Element:
    """获取元素"""
    def __init__(self):
        self.data = test

    def __getitem__(self, item):
        """获取属性"""
        data = self.data.get(item, '')
        if data:
            return data
        raise Exception('关键字不存在')


if __name__ == '__main__':
    print(Element()['key1'])
    print(Element()['key0'])

你可能感兴趣的:(python,开发语言)