在Python中,如果我们想实现创建类似于序列和映射的类(可以迭代以及通过[下标]返回元素),可以通过重写魔法方法__getitem__
、__setitem__
、__delitem__
、__len__
方法去模拟。
魔术方法的作用:
__getitem__
(self,key):返回键对应的值。
__setitem__
(self,key,value):设置给定键的值
__delitem__
(self,key):删除给定键对应的元素。
__len__
():返回元素的数量
''' desc:尝试定义一种新的数据类型 等差数列 ''' class ArithemeticSequence: def __init__(self, start=0, step=1): print('Call function __init__') self.start = start self.step = step self.myData = {} # 定义获取值的方法 def __getitem__(self, key): print('Call function __getitem__') try: return self.myData[key] except KeyError: return self.start + key * self.step # 定义赋值方法 def __setitem__(self, key, value): print('Call function __setitem__') self.myData[key] = value # 定义获取长度的方法 def __len__(self): print('Call function __len__') return len(self.myData) # 定义删除元素的方法 def __delitem__(self, key): print('Call function __delitem__') del self.myData[key] s = ArithemeticSequence(1, 2) print(s[0]) print(s[1]) print(s[2]) print(s[3])# 这里应该执行self.start+key*self.step,因为没有3这个key s[3] = 100 # 进行赋值 print(s[3]) # 前面进行了赋值,那么直接输出赋的值100 print(len(s)) del s[3] # 删除3这个key
输出结果:
Call function __init__ Call function __getitem__ 1 Call function __getitem__ 3 Call function __getitem__ 5 Call function __getitem__ 7 Call function __setitem__ Call function __getitem__ 100 Call function __len__ 1 Call function __delitem__
这些魔术方法的原理就是:当我们对类的属性item进行下标的操作时,首先会被__getitem__
()、__setitem__
()、__delitem__
()拦截,从而执行我们在方法中设定的操作,如赋值,修改内容,删除内容等等。