欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  后端开发

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

程序员文章站 2022-04-06 22:57:52
...
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__方法。


我们来举一个