Python进阶之：__getitem__

'''
有时，我们希望可以使用 obj[n] 这种方式对实例对象进行取值，比如对斐波那契数列，
我们希望可以取出其中的某一项，这时我们需要在类中实现 __getitem__ 方法，比如下面的例子：
'''

# class Fib(object):
#     def __getitem__(self, item):
#         a,b = 1,1
#         for x in range(item):
#             a,b = b,a+b
#         return a
#
#
# fib = Fib()
# print(fib[100])

'''
我们还想更进一步，希望支持 obj[1:3] 这种切片方法来取值，这时 __getitem__ 方法传入的参数可能是一个整数，也可能是一个切片对象 slice，
因此，我们需要对传入的参数进行判断，可以使用 isinstance 进行判断，改后的代码如下：
'''

class Fib(object):
    def __getitem__(self, item):
        if isinstance(item,slice):
            a,b = 1,1
            start,stop = item.start,item.stop
            l = []
            for x in range(stop):
                if x >= start:
                    l.append(a)
                a,b = b,a+b
            return l
        if isinstance(item,int):
            a,b = 1,1
            for x in range(item):
                a,b = b,a+b
            return a

fib = Fib()
print(fib[5:10])