Python函数(八)-装饰器(一)

装饰器通过函数来定义，用来装饰函数

装饰器的结构为高阶函数和内嵌函数

装饰器不改变被装饰函数的源代码和运行方式

如何实现这个效果呢？

# -*- coding:utf-8 -*-

__author__ = "MuT6 Sch01aR"

import time

def timer(func): #定义一个装饰器

    def deco():

        start_time = time.time()

        func()

        stop_time = time.time()

        print("the run time is %s"%(stop_time-start_time))

    return deco

def test1():

    time.sleep(3)

    print('in the test1')

test1 = timer(test1)

test1()

既没有改变被装饰函数的源代码，也没有改变它的运行方式

运行

这么写有些复杂，可以直接在函数前调用装饰器

调用装饰器的格式为：@装饰器名

# -*- coding:utf-8 -*-

__author__ = "MuT6 Sch01aR"

import time

def timer(func): #定义一个装饰器

    def deco():

        start_time = time.time()

        func()

        stop_time = time.time()

        print("the run time is %s"%(stop_time-start_time))

    return deco

@timer #相当于test1 = timer(test1)

def test1():

    time.sleep(3)

    print('in the test1')

test1()

运行

执行过程：

先走test1函数前的装饰器timer()，然后在timer()函数内走函数deco()，记录下start_time，然后deco()函数调用函数test1()，然后执行完函数test1()后，记录下stop_time，最后计算时间并打印

虽然装饰器不会修改被装饰函数的源代码和运行方式，但是不能直接返回函数的返回值

# -*- coding:utf-8 -*-

__author__ = "MuT6 Sch01aR"

import time

def timer(func): #定义一个装饰器

    def deco():

        start_time = time.time()

        func()

        stop_time = time.time()

        print("the run time is %s"%(stop_time-start_time))

    return deco

@timer #test1 = timer(test1)

def test1():

    time.sleep(3)

    print('in the test1')

    return "test1"

print(test1())

运行结果

打印的返回值为None，因为这个返回值为装饰器里的内嵌函数的返回值

如果需要打印被装饰函数的返回值的话，需要在装饰器里的内嵌函数中返回

# -*- coding:utf-8 -*-

__author__ = "MuT6 Sch01aR"

import time

def timer(func): #定义一个装饰器

    def doc():

        start_time = time.time()

        func()

        stop_time = time.time()

        print("the run time is %s"%(stop_time-start_time))

        return func()

    return doc

@timer #test1 = timer(test1)

def test1():

    time.sleep(3)

    print('in the test1')

    return "test1"

print(test1())

运行结果

巴特西

Python函数(八)-装饰器(一)

最新文章

热门文章