[已解决]python怎么重载下标运算符(不知道是不是这么叫的)

Stubborn

1. class Group(object):
   
2.     """支持切片操作,实现这个协议，需要实现一下魔法函数"""
   
3.     def __init__(self, name, company, staffs):
   
4.         self.name = name
   
5.         self.company = company
   
6.         self.staffs = staffs
   
7. 
   
8.     def __reversed__(self):
   
9.         self.staffs.reverse()
   
10. 
    
11.     def __getitem__(self, item):
    
12.         """
    
13.         注意，如果是切片，item则是一个slice,如果是取值，item是一个int
    
14.         在对实例化对象进行索引，切片在这里实现对应的逻辑
    
15.         """
    
16.         import numbers
    
17.         cls = type(self)
    
18.         if isinstance(item, slice):
    
19.             return cls(name=self.name, company=self.company, staffs=self.staffs[item])
    
20.         elif isinstance(item, numbers.Integral):
    
21.             return cls(name=self.name, company=self.company, staffs=[self.staffs[item]])
    
22. 
    
23.     def __len__(self):
    
24.         """在对实例对象求长度"""
    
25.         return len(self.staffs)
    
26. 
    
27.     def __iter__(self):
    
28.         """返回一个迭代器，支持被遍历"""
    
29.         return iter(self.staffs)
    
30. 
    
31.     def __contains__(self, item):
    
32.         """实例对象支持in操作，"""
    
33.         if item in self.staffs:
    
34.             return True
    
35.         return False

复制代码