Python装饰器（你想知道的这里都有）_装饰器特诊

最新推荐文章于 2025-02-10 17:17:07 发布

2301_82242410

最新推荐文章于 2025-02-10 17:17:07 发布

阅读量807

点赞数 17

文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/2301_82242410/article/details/139934465

版权

5.1 装饰器的使用场景

函数执行时间的统计
输出日志信息

5.2 装饰器实现已有函数执行时间的统计

import time


def decorator(func):
    def inner():
        # 获取时间距离1970-1-1 0:0:1的时间差
        begin = time.time()
        func()
        end = time.time()
        result = end - begin
        print(f'函数执行完成耗时：{result}')

    return inner


@decorator
def work():
    for i in range(10000):
        print(i)


work()

6. 通用装饰器的使用

通用装饰器：可以装饰任意类型的函数

使用装饰器装饰已有函数的时候，内部函数的类型和要装饰的已有函数的类型保持一致

6.1 装饰带有参数的函数

def decorator(func):
    def inner(num1, num2):
        print('正在努力执行加法计算')
        func(num1, num2)

    return inner


@decorator
def add\_num(num1, num2):
    result = num1 + num2
    print(f'结果为：{result}')


add_num(1, 2)

6.2 装饰带有参数、返回值的函数

def decorator(func):
    def inner(num1, num2):
        print('正在努力执行加法计算')
        num = func(num1, num2)
        return num

    return inner


@decorator
def add\_num(num1, num2):
    result = num1 + num2
    return result


result = add_num(1, 2)
print(f'结果为：{result}')

6.3 装饰带有不定长参数、返回值的函数

def decorator(func):
    def inner(\*args, \*\*kwargs):
        print('正在努力执行加法计算')
        # \*args:把元组里面的每一个元素，按照位置参数的方式进行传参
        # \*\*kwargs:把字典里面的每一个键值对，按照关键字的方式进行传参
        num = func(\*args, \*\*kwargs)
        return num

    return inner


@decorator
def add\_num(\*args, \*\*kwargs):
    result = 0
    for value in args:
        result += value
    for value in kwargs.values():
        result += value
    return result


result = add_num(1, 2, a=3)
print(f'结果为：{result}')

7. 多个装饰器的使用

多个装饰器的装饰过程：由内到外的一个装饰过程，先执行内部的装饰器，在执行外部的装饰器。

def make\_div(func):
    print('make\_div装饰器执行了')

    def inner():
        result = '<div>' + func() + '</div>'
        return result

    return inner


def make\_p(func):
    print('make\_p装饰器执行了')

    def inner():
        result = '<p>' + func() + '</p>'
        return result

    return inner


# 原理剖析：content = make\_div(make\_p(content))
# 分布拆解：content = make\_p(content)，内部装饰器完成，content = make\_p.inner
# content = make\_div(make\_p.inner)
@make_div
@make_p
def content():
    return '人生苦短，我用python'


c = content()
print(c)

make_p装饰器执行了
make_div装饰器执行了
<div><p>人生苦短，我用python</p></div>

8. 带有参数的装饰器

带有参数的装饰器就是使用装饰器装饰函数的时候可以传入指定参数，语法格式: @装饰器(参数,…)

使用带有参数的装饰器，其实是在装饰器外面又包裹了一个函数，使用该函数接收参数，返回是装饰器，因为 @ 符号需要配合装饰器实例使用。

def return\_decorator(flag):
    # 装饰器只能接收一个参数并且是函数类型
    def decorator(func):
        def inner(a, b):
            if flag == '+':
                print('正在努力执行加法计算')
            elif flag == '-':
                print('正在努力执行减法计算')
            func(a, b)

        return inner

    # 当调用函数的时候可以返回一个装饰器decorator
    return decorator


@return_decorator('+')  # decorator = return\_decorator('+'), @decorator => add\_num = decorator(add\_num)
def add\_num(a, b):
    result = a + b
    print(result)


@return_decorator('-')
def sub\_num(a, b):
    result = a - b
    print(result)


add_num(1, 2)
sub_num(1, 2)

正在努力执行加法计算
3
正在努力执行减法计算
-1

9. 类装饰器的使用

类装饰器：使用类装饰已有函数

class MyDecorator(object):
    def \_\_init\_\_(self, func):
        self.__func = func

    # 实现\_\_call\_\_方法，表示对象是一个可调用对象，可以像调用函数一样进行调用
    def \_\_call\_\_(self, \*args, \*\*kwargs):
        # 对已有函数进行封装
        print('马上就有下班啦')
        self.__func()


@MyDecorator  # @MyDecorator => show = MyDecorator(show)
def show():
    print('快要下雪啦')


# 执行show，就相当于执行MyDecorator类创建的实例对象，show() => 对象()
show()