python 装饰器 补充

2018-08-28  本文已影响0人  snailpy

重新理解python 装饰器

python 装饰器是一个函数,被装饰器所装饰的代码块最终也是一个函数
这个对于一般的情况:装饰器装饰函数来说,没什么意义,因为本来就是函数,装饰完毕后只是多了部分功能,但任然是函数。但是当使用装饰器装饰类的时候就要当心了,因为本来的类也被装饰成了函数。

装饰器装饰过的对象最终放回为函数

import functools

def log(cls):
    print('log----------')
    functools.wraps(cls)
    def wrapper(*args):
        return cls(*args)
    return wrapper

@log
class A(object):
    def __init__(self, a=1,b=2):
        self.data = a+b
        
a = A(3,4)
print(a.data)
print(type(a))
print(type(A))

输出:
log----------
7
<class '__main__.A'>
<class 'function'>

重点关注print(type(A)) 输出 是<class 'function'> 如果不使用装饰器是<class 'type'> 简单解释下, type()可以查看一个对象的类。且如果一个对象的类是type则表示这个对象是一个类对象,可以创建新的对象,否则都是普通对象。或者可以使用print(A)来直接输出对象对比下。未装饰前输出:<class '__main__.A'> 装饰后输出:<function log.<locals>.wrapper at 0x000000000B0F1400> 可见一个类在装饰器装饰后确实变成了一个函数对象。那么此时就要注意了,A对象不具备被继承的功能了:

...
@log
class A(object):
    def __init__(self, a=1,b=2):
        self.data = a+b
        
class B(A):
    pass
...

编译时,直接报错:TypeError: function() argument 1 must be code, not str,所以在 python 单例模式
中使用 装饰器实现单例时,就要小心了。

装饰器在 函数、类 定义的地方使用

先来思考一个问题,既然python 装饰器可以装饰类,是因为在python中一切皆对象,只要这个对象实现了__call__()方法就可以像函数一样被调用,也可以像函数一样被装饰,那么我们随便创建一个类的对象呢,只要这个对像实现了__call__()方法是不就可以像函数一样呢?

class A(object):
    def __init__(self, a=1,b=2):
        self.data = a+b
    
    def __call__(self):
        print('aaaaaaaaaaa')        

a = A(3,5)
print(a.data)
a()

输出
8
aaaaaaaaaaa

可见对象a确实可以像函数一样被调用,内容就是类中实现的__call__()方法。那么是否也可给a() 调用装饰器呢?大家不妨试试,SyntaxError: invalid syntax直接语法错误了。这是因为,装饰器在装饰函数、类对象时都是在这些对象定义的时候装饰的。而非调用的时候,但我们现在的a()是直接调用方法了,所以装饰器不能这样装饰

关于 __call__type

这里我们简单提一下type类的继承之间的区别。

当我们要想使一个对象可以像函数一样调用,则需要让这个对象实现__call__ 方法如下:

class A(object):
    def __call__(self):
        print('aaaaaaaaaaa')        

a = A()
a()

可见,因为a对象的类中实现了,__call__方法所以a()可用
那么要想让A()可用该如何,此时我们就应该在A 这个类对象的类中修改__call__方法了,但是A的类是type的内置类,所以我们最好不要改动了。

暂时到这。。。。。未完

上一篇下一篇

猜你喜欢

热点阅读