Python 学习笔记9（装饰器，decorator）

31 装饰器

装饰器可以对一个函数、方法或者类进行加工，是一种高级的python语法。

装饰函数

接收一个可调用对象作为输入参数，并返回一个新的可调用对象.
把函数传递给装饰器，然后增加新的功能，返回一个新的函数重赋值给原函数
语法：

def decorator1... # 定义装饰器

@decorator1 # 定义函数前，@装饰器

def function1 ＃定义要加工的函数

例子：

def mydecorator(myfunction):

    def new_myfunction(a,b):

        print("modify myfunction:")   ＃ 1

        return myfunction(a,b)

    return new_myfunction

@mydecorator

def myfunction(x,y):

    print("old myfunction")            ＃2

    return x*y

print(myfunction(6,7))

顺序：先执行装饰器里面的语句1，再执行原函数的语句2。

含参的装饰器

在上面的例子中，@mydecorator默认后面的函数是它的唯一参数；

实际上，装饰器也可以含有其它参数，如@mydecorator(a)

装饰类

一个装饰器可以接收一个类，并返回一个类，从而起到加工类的效果。

例子：

def mydecorator(myclass):

    class newClass:

        def __init__(self,age):

            self.display_time=0

            self.wrapped=myclass(age)                               #把原类赋值给新类

        def display(self):

            self.display_time+=1

            print("this is ",self.display_time,"th display time")   #1

            self.wrapped.display()                                  # 调用原类的函数

    return newClass

@mydecorator

class Bird:

    def __init__(self,age):

        self.age=age

    def display(self):

        print("My age is ", self.age)                              #2

eagleLord=Bird(5)

for i in range(4):

    eagleLord.display()

顺序与装饰函数类似。先装饰类1，后原类2。

巴特西

Python 学习笔记9（装饰器，decorator）

31 装饰器

装饰函数

含参的装饰器

装饰类

最新文章

热门文章