python中实现定制类的特殊方法总结


Posted in Python onSeptember 28, 2014

看到类似__slots__这种形如__xxx__的变量或者函数名就要注意,这些在Python中是有特殊用途的。

__slots__我们已经知道怎么用了,__len__()方法我们也知道是为了能让class作用于len()函数。

除此之外,Python的class中还有许多这样有特殊用途的函数,可以帮助我们定制类。

__str__

我们先定义一个Student类,打印一个实例:

>>> class Student(object):

...     def __init__(self, name):

...         self.name = name

...

>>> print Student('Michael')

<__main__.Student object at 0x109afb190>

打印出一堆<__main__.Student object at 0x109afb190>,不好看。

怎么才能打印得好看呢?只需要定义好__str__()方法,返回一个好看的字符串就可以了:

>>> class Student(object):

...     def __init__(self, name):

...         self.name = name

...     def __str__(self):

...         return 'Student object (name: %s)' % self.name

...

>>> print Student('Michael')

Student object (name: Michael)

这样打印出来的实例,不但好看,而且容易看出实例内部重要的数据。

但是细心的朋友会发现直接敲变量不用print,打印出来的实例还是不好看:

>>> s = Student('Michael')

>>> s

<__main__.Student object at 0x109afb310>

这是因为直接显示变量调用的不是__str__(),而是__repr__(),两者的区别是__str__()返回用户看到的字符串,而__repr__()返回程序开发者看到的字符串,也就是说,__repr__()是为调试服务的。

解决办法是再定义一个__repr__()。但是通常__str__()和__repr__()代码都是一样的,所以,有个偷懒的写法:

class Student(object):

    def __init__(self, name):

        self.name = name

    def __str__(self):

        return 'Student object (name=%s)' % self.name

    __repr__ = __str__

__iter__

如果一个类想被用于for ... in循环,类似list或tuple那样,就必须实现一个__iter__()方法,该方法返回一个迭代对象,然后,Python的for循环就会不断调用该迭代对象的next()方法拿到循环的下一个值,直到遇到StopIteration错误时退出循环。

我们以斐波那契数列为例,写一个Fib类,可以作用于for循环:

class Fib(object):

    def __init__(self):

        self.a, self.b = 0, 1 # 初始化两个计数器a,b
    def __iter__(self):

        return self # 实例本身就是迭代对象,故返回自己
    def next(self):

        self.a, self.b = self.b, self.a + self.b # 计算下一个值

        if self.a > 100000: # 退出循环的条件

            raise StopIteration();

        return self.a # 返回下一个值

现在,试试把Fib实例作用于for循环:

>>> for n in Fib():

...     print n

...

1

1

2

3

5

...

46368

75025

__getitem__

Fib实例虽然能作用于for循环,看起来和list有点像,但是,把它当成list来使用还是不行,比如,取第5个元素:

>>> Fib()[5]

Traceback (most recent call last):

  File "<stdin>", line 1, in <module>

TypeError: 'Fib' object does not support indexing

要表现得像list那样按照下标取出元素,需要实现__getitem__()方法:
class Fib(object):

    def __getitem__(self, n):

        a, b = 1, 1

        for x in range(n):

            a, b = b, a + b

        return a

现在,就可以按下标访问数列的任意一项了:
>>> f = Fib()

>>> f[0]

1

>>> f[1]

1

>>> f[2]

2

>>> f[3]

3

>>> f[10]

89

>>> f[100]

573147844013817084101

但是list有个神奇的切片方法:
>>> range(100)[5:10]

[5, 6, 7, 8, 9]

对于Fib却报错。原因是__getitem__()传入的参数可能是一个int,也可能是一个切片对象slice,所以要做判断:
class Fib(object):

    def __getitem__(self, n):

        if isinstance(n, int):

            a, b = 1, 1

            for x in range(n):

                a, b = b, a + b

            return a

        if isinstance(n, slice):

            start = n.start

            stop = n.stop

            a, b = 1, 1

            L = []

            for x in range(stop):

                if x >= start:

                    L.append(a)

                a, b = b, a + b

            return L

现在试试Fib的切片:
>>> f = Fib()

>>> f[0:5]

[1, 1, 2, 3, 5]

>>> f[:10]

[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]

但是没有对step参数作处理:
>>> f[:10:2]

[1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89]

也没有对负数作处理,所以,要正确实现一个__getitem__()还是有很多工作要做的。

此外,如果把对象看成dict,__getitem__()的参数也可能是一个可以作key的object,例如str。

与之对应的是__setitem__()方法,把对象视作list或dict来对集合赋值。最后,还有一个__delitem__()方法,用于删除某个元素。

总之,通过上面的方法,我们自己定义的类表现得和Python自带的list、tuple、dict没什么区别,这完全归功于动态语言的“鸭子类型”,不需要强制继承某个接口。

__getattr__

正常情况下,当我们调用类的方法或属性时,如果不存在,就会报错。比如定义Student类:

class Student(object):
    def __init__(self):

        self.name = 'Michael'

调用name属性,没问题,但是,调用不存在的score属性,就有问题了:
>>> s = Student()

>>> print s.name

Michael

>>> print s.score

Traceback (most recent call last):

  ...

AttributeError: 'Student' object has no attribute 'score'

错误信息很清楚地告诉我们,没有找到score这个attribute。

要避免这个错误,除了可以加上一个score属性外,Python还有另一个机制,那就是写一个__getattr__()方法,动态返回一个属性。修改如下:

class Student(object):
    def __init__(self):

        self.name = 'Michael'
    def __getattr__(self, attr):

        if attr=='score':

            return 99

当调用不存在的属性时,比如score,Python解释器会试图调用__getattr__(self, 'score')来尝试获得属性,这样,我们就有机会返回score的值:
>>> s = Student()

>>> s.name

'Michael'

>>> s.score

99

返回函数也是完全可以的:
class Student(object):
    def __getattr__(self, attr):

        if attr=='age':

            return lambda: 25

只是调用方式要变为:
>>> s.age()

25

注意,只有在没有找到属性的情况下,才调用__getattr__,已有的属性,比如name,不会在__getattr__中查找。

此外,注意到任意调用如s.abc都会返回None,这是因为我们定义的__getattr__默认返回就是None。要让class只响应特定的几个属性,我们就要按照约定,抛出AttributeError的错误:

class Student(object):
    def __getattr__(self, attr):

        if attr=='age':

            return lambda: 25

        raise AttributeError('\'Student\' object has no attribute \'%s\'' % attr)

这实际上可以把一个类的所有属性和方法调用全部动态化处理了,不需要任何特殊手段。

这种完全动态调用的特性有什么实际作用呢?作用就是,可以针对完全动态的情况作调用。

举个例子:

现在很多网站都搞REST API,比如新浪微博、豆瓣啥的,调用API的URL类似:

http://api.server/user/friends
http://api.server/user/timeline/list

如果要写SDK,给每个URL对应的API都写一个方法,那得累死,而且,API一旦改动,SDK也要改。

利用完全动态的__getattr__,我们可以写出一个链式调用:

class Chain(object):
    def __init__(self, path=''):

        self._path = path
    def __getattr__(self, path):

        return Chain('%s/%s' % (self._path, path))
    def __str__(self):

        return self._path

试试:
>>> Chain().status.user.timeline.list

'/status/user/timeline/list'

这样,无论API怎么变,SDK都可以根据URL实现完全动态的调用,而且,不随API的增加而改变!

还有些REST API会把参数放到URL中,比如GitHub的API:

GET /users/:user/repos

调用时,需要把:user替换为实际用户名。如果我们能写出这样的链式调用:
Chain().users('michael').repos

就可以非常方便地调用API了。有兴趣的童鞋可以试试写出来。

__call__

一个对象实例可以有自己的属性和方法,当我们调用实例方法时,我们用instance.method()来调用。能不能直接在实例本身上调用呢?类似instance()?在Python中,答案是肯定的。

任何类,只需要定义一个__call__()方法,就可以直接对实例进行调用。请看示例:

class Student(object):

    def __init__(self, name):

        self.name = name
    def __call__(self):

        print('My name is %s.' % self.name)

调用方式如下:
>>> s = Student('Michael')

>>> s()

My name is Michael.

__call__()还可以定义参数。对实例进行直接调用就好比对一个函数进行调用一样,所以你完全可以把对象看成函数,把函数看成对象,因为这两者之间本来就没啥根本的区别。

如果你把对象看成函数,那么函数本身其实也可以在运行期动态创建出来,因为类的实例都是运行期创建出来的,这么一来,我们就模糊了对象和函数的界限。

那么,怎么判断一个变量是对象还是函数呢?其实,更多的时候,我们需要判断一个对象是否能被调用,能被调用的对象就是一个Callable对象,比如函数和我们上面定义的带有__call()__的类实例:

>>> callable(Student())

True

>>> callable(max)

True

>>> callable([1, 2, 3])

False

>>> callable(None)

False

>>> callable('string')

False

通过callable()函数,我们就可以判断一个对象是否是“可调用”对象。

小结

Python的class允许定义许多定制方法,可以让我们非常方便地生成特定的类。

Python 相关文章推荐
python 实现文件的递归拷贝实现代码
Aug 02 Python
浅谈Python由__dict__和dir()引发的一些思考
Oct 30 Python
matplotlib.pyplot画图 图片的二进制流的获取方法
May 24 Python
python使用selenium登录QQ邮箱(附带滑动解锁)
Jan 23 Python
Python 3.8新特征之asyncio REPL
May 28 Python
Numpy之将矩阵拉成向量的实例
Nov 30 Python
python实现图像拼接功能
Mar 23 Python
python小程序基于Jupyter实现天气查询的方法
Mar 27 Python
使用jupyter notebook直接打开.md格式的文件
Apr 10 Python
Django ForeignKey与数据库的FOREIGN KEY约束详解
May 20 Python
如何快速一次性卸载所有python包(第三方库)呢
Oct 20 Python
关于Python 解决Python3.9 pandas.read_excel(‘xxx.xlsx‘)报错的问题
Nov 28 Python
python之wxPython菜单使用详解
Sep 28 #Python
python中lambda函数 list comprehension 和 zip函数使用指南
Sep 28 #Python
python之wxPython应用实例
Sep 28 #Python
Python实现从url中提取域名的几种方法
Sep 26 #Python
Python实现的一个简单LRU cache
Sep 26 #Python
python网络编程实例简析
Sep 26 #Python
python的re模块应用实例
Sep 26 #Python
You might like
php查询相似度最高的字符串的方法
2015/03/12 PHP
Yii 框架使用Forms操作详解
2020/05/18 PHP
ztree获取当前选中节点子节点id集合的方法
2015/02/12 Javascript
深入浅析同源策略和跨域访问
2015/11/26 Javascript
详解AngularJS过滤器的使用
2016/03/11 Javascript
深入理解jQuery之事件移除
2016/06/02 Javascript
jquery动态创建div与input的实例代码
2016/10/12 Javascript
js常用的继承--组合式继承
2017/03/06 Javascript
AngularJS 控制器 controller的详解
2017/10/17 Javascript
改变vue请求过来的数据中的某一项值的方法(详解)
2018/03/08 Javascript
如何在vue里添加好看的lottie动画
2018/08/02 Javascript
JavaScript实现预览本地上传图片功能完整示例
2019/03/08 Javascript
JavaScript中的一些实用小技巧总结
2019/04/07 Javascript
vue实现分环境打包步骤(给不同的环境配置相对应的打包命令)
2019/06/04 Javascript
node基于async/await对mysql进行封装
2019/06/20 Javascript
VsCode与Node.js知识点详解
2019/09/05 Javascript
JS精确判断数据类型代码实例
2019/12/18 Javascript
vue+ts下对axios的封装实现
2020/02/18 Javascript
angular中的post请求处理示例详解
2020/06/30 Javascript
python通过文件头判断文件类型
2015/10/30 Python
Python基于numpy灵活定义神经网络结构的方法
2017/08/19 Python
python解决js文件utf-8编码乱码问题(推荐)
2018/05/02 Python
python中redis查看剩余过期时间及用正则通配符批量删除key的方法
2018/07/30 Python
Django项目之Elasticsearch搜索引擎的实例
2019/08/21 Python
wxpython多线程防假死与线程间传递消息实例详解
2019/12/13 Python
Django ORM 查询表中某列字段值的方法
2020/04/30 Python
Origins悦木之源英国官网:雅诗兰黛集团高端植物护肤品牌
2017/11/06 全球购物
出口公司经理求职简历中的自我评价
2013/10/13 职场文书
前台接待员岗位职责
2014/01/02 职场文书
护士岗位求职应聘自荐书范文
2014/02/12 职场文书
说明书怎么写
2014/05/06 职场文书
文秘专业应届生求职信
2014/05/26 职场文书
团员自我评价范文
2015/03/10 职场文书
贫民窟的百万富翁观后感
2015/06/09 职场文书
志愿者服务宣传标语口号
2015/12/26 职场文书
Java spring单点登录系统
2021/09/04 Java/Android