python进阶13——手动实现一个支持切片的类型

自己写一个可以切片的类型

这里我们实现一个Group类型的定义，它含有多个属性（leader，staffs，group_name），其中staffs是一个list序列，切片也是在这个staffs的list上操作，经过切片的依然是Group类型。

切片协议

要实现一个可切片的类，应该实现以下抽象方法：
len(self)：返回序列的长度，即元素的个数。
getitem(self, index)：根据索引获取序列中的元素。支持单个索引和切片索引。
此外staffs是mutsequence，最好也实现相应的抽象方法：
setitem(self, index, value)：根据索引设置序列中的元素的值。支持单个索引和切片索引。
delitem(self, index)：根据索引删除序列中的元素。支持单个索引和切片索引。

代码

class Group:
    def __init__(self,leader,group_name,staffs) -> None:
        self.leader = leader
        self.group_name = group_name
        self.staffs = staffs

    def __getitem__(self,index):
        cls = type(self)
        if isinstance(index,slice):
            return cls(leader = self.leader,group_name = self.group_name,staffs = self.staffs[index])
        if isinstance(index,numbers.Integral):
            return cls(leader = self.leader,group_name = self.group_name,staffs = [self.staffs[index]])
        
    def __len__(self):
        return len(self.staffs)
    
    def __setitem__(self,index,value):
        if isinstance(index,slice):
            self.staffs[index] = value
        else :
            self.staffs[index] = value

    def __delitem__(self,index):
        if isinstance(index,slice):
            del self.staffs[index]
        else:
            del self.staffs[index]
    def print_staff(self):
        print(self.staffs)
    

group1 = Group('mark','study_group',['kare','hourse','bob','Ash','tracer','Gengi','Ion','Mercy'])
group1.print_staff()
group1[:2].print_staff()