python 魔术方法 : 让自定义类更像内置类型

Python的魔术方法是Python中那些预定义的像__XXX__类型的函数。

使用Python的魔术方法的最大优势在于python提供了简单的方法让对象可以表现得像内置类型一样。

__str__函数

__str__函数用于处理打印实例本身的时候的输出内容。如果没有覆写该函数，则默认输出一个对象名称和内存地址。

例如：

>>> class Student(object):
...     def __init__(self,name):
...             self._name = name
...
>>> print Student()

输出：

<__main__.Student 
object at 0x0000000002A929E8>.

那么我们如何让输出的结果可读性更高一点呢？我们可以覆写__str__函数。例如

>>> class Student(object):
...     def __init__(self, name):
...             self._name = name
...     def __str__(self):
...             return  "I'm a student, named %s" % self._name
...
>>> print Student("Charlie")

输出结果就是：

I'm a student, named Charlie.

我们将str()函数作用于该对象的时候，其实是调用了该对象的__str__函数。

__repr__ 函数

__repr__也是将对象序列化，但是__repr__更多的是给python编译器看的。__str__更多的是可读性(readable)。

我们将repr()函数作用于摸某一个对象的时候，调用的其实就是该函数的__repr__函数。

与repr()成对的是eval()函数。eval()函数是将序列化后的对象重新转为对象。前提是该对象实现了__repr__函数。

上面这一段话基于自己的理解，不知道对错。

>>> item = [1,2,3]
>>> repr(item)
'[1, 
2, 3]'
>>> other_item = eval(repr(item))
>>> other_item[1]
2

__iter__函数

我们经常对list或者tuple使用for…in…来迭代。那是list继承自Iterable。Iterable实现了__iter__函数。

要想将一个自定义的对象变成一个可迭代的对象，那么必须要实现两个方法：__iter__和next.

__iter__函数返回一个对象。迭代的时候则会不断地调用next函数拿到下一个值，直到捕获到StopIteration停止。

廖雪峰老师教程里写的是__next__方法，不知道为啥。

class Fib(object):
    def __init__(self):
        self.a, self.b = 0, 1
 
    def __iter__(self):
        return self
 
    def next(self):
        self.a, self.b = self.b, self.a + self.b
        if self.a > 10000:
            raise StopIteration
        return self.a
 
 
for i in Fib():
    print i

__getitem__函数

上面通过实现__iter__函数实现对象的迭代。

那么如何实现对象按下标取出元素呢。

这是通过实现对象的__getitem__方法。

我们来举一个