Python的装饰器

一、概念

在Python中,装饰器本质是一个特殊的嵌套函数,它接收一个函数(被装饰的函数)做参数,并返回一个新的函数(装饰后的函数)

装饰器最大的作用:在不改变原有的函数(被装饰的函数)的基础上给他添加新的功能

二、基本用法

python 复制代码
# 装饰器
def out_fun(function):
    def in_fun():
        print("挂上一颗星星")
        function()
        print("挂上一个礼物盒")
    return in_fun

# 被装饰函数
def my_tree():
    print("简简单单一棵树")

# 第一种调用方式
out_fun(my_tree)()
# 问:第一种调用方式,被装饰函数真的装饰了吗
my_tree()       # 并没有新增功能

# 第二种调用方式
my_tree = out_fun(my_tree)
# 问:第二种调用方式,被装饰函数真的装饰了吗
my_tree()

三、装饰器语法糖

语法糖是由编程语言提供的,可以让代码更加简洁、高效、易读和易写。语法糖不改变不会带来新的功能,也不会改变编程的结果,但是使用它会更加方便。

python中有很多语法糖,已用的有:

  1. 列表推导式
  2. 集合推导式
  3. 字典推导式
  4. f-string print(f"{}")
  5. 解包
  6. 装饰器

语法格式:

@装饰器函数名

def 被装饰函数名:

代码块

其实就是在被装饰函数的上面加一句话------@装饰器

python 复制代码
# 装饰器
def out_fun(function):
    def in_fun():
        print("挂上一颗星星")
        function()
        print("挂上一个礼物盒")
    return in_fun

# 被装饰函数
@out_fun
def my_tree():
    print("简简单单一棵树")

my_tree()

四、被装饰函数有参数

若被装饰函数有参数,那么装饰器的内部函数也需要有参数,保证在内部调用被装饰函数的时候能正确传参。

1. 被装饰函数有一个参数

python 复制代码
def out_fun(function):
    def in_fun(x):
        print("挂上一颗星星")
        function(x)
        print("挂上一个礼物盒")
    return in_fun

# 被装饰函数
@out_fun
def my_tree(x):
    print(f"简简单单{x}棵树")

my_tree(3)

2.被装饰函数有未知个参数

python 复制代码
import time
# 装饰器
def out_fun(function):
    def in_fun(*args, **kwargs):
        start = time.time()
        function(*args, **kwargs)
        end = time.time()
        print(end - start)
    return in_fun

# 被装饰函数1
@out_fun
def fun1(x):
    time.sleep(1)
    print(x)
fun1(1)
# 被装饰函数2
@out_fun
def fun2(x, y, z):
    time.sleep(1)
    print(x + y + z)
fun2(1,2,3)

五、装饰器带参数

现在装饰器要带有参数,而且装饰器的外部函数要接收被装饰函数的函数名,内部函数要接收被装饰函数的参数,那么为了保证装饰器参数、被装饰函数参数的正确传递,我们在装饰器外部函数再嵌套一层函数,用于接收装饰器参数。

python 复制代码
# 装饰器
def decoration(dec_arg):
    def out_fun(function):
        def in_fun(x):
            print(dec_arg)
            print("挂上一颗星星")
            function(x)
            print("挂上一个礼物盒")
        return in_fun
    return out_fun

# 被装饰函数
@decoration("我是装饰器的参数")
def my_tree(x):
    print(f"简简单单{x}棵树")

my_tree(3)

六、装饰器嵌套

装饰器嵌套就是被装饰函数 可以 被多个装饰器装饰。

python 复制代码
# 装饰器1
def out_fun_1(function):
    def in_fun_1(*args, **kwargs):
        print("装饰器1开始调用")
        function(*args, **kwargs)
        print("装饰1结束调用")
    return in_fun_1

# 装饰器2
def out_fun_2(function):
    def in_fun_2(*args, **kwargs):
        print("装饰器2开始调用")
        function(*args, **kwargs)
        print("装饰2结束调用")
    return in_fun_2

# 嵌套装饰 被装饰函数
@out_fun_1
@out_fun_2
def my_tree(x):
    print(f"{x}棵树")

my_tree(3)

输出

python 复制代码
装饰器1开始调用
装饰器2开始调用
3棵树
装饰2结束调用
装饰1结束调用

七、类装饰器

除了可以自定义一个新的函数用作装饰器之外,也可以将一个类作为装饰器,为被装饰的函数添加新的功能。类装饰器通过实现类的__call__方法,使得类的实例可以被当作函数来调用,从而实现对其他函数的装饰。

八、装饰器的常见应用【了解即可】

1.记录日志

装饰器可以用来记录函数调用的详细信息,包括调用时间、参数、返回值等

python 复制代码
import logging
def log_decorator(func):
    def wrapper(*args, **kwargs):
        logging.basicConfig(filename='./app.log', level=logging.INFO, filemode='a',
                            format='%(name)s - %(levelname)s - %(asctime)s - %(message)s')
        logging.warning(f"calling function:{func.__name__} with args:{args} and "
                        f"kwargs:{kwargs}")
        ret = func(*args, **kwargs)
        logging.warning(f"function {func.__name__} returned: {ret}")
        return ret
    return wrapper

@log_decorator
def test():
    print('123')

test()

2. 性能检测

装饰器可以用来测量函数执行所需的时间,帮助识别性能瓶颈。

python 复制代码
import time
def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        ret = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} took {(end_time - start_time):.10f} seconds to run")
        return ret
    return wrapper

@timer
def process_list(my_list):
    for i in my_list:
        pass

my_list = list(range(10000))
process_list(my_list)

3.权限验证

装饰器可以用来检查用户是否有权限执行特定的函数或方法。

python 复制代码
class User:
    def __init__(self, username, role):
        self.username = username
        self.role = role

# 装饰器:检查用户是否具有管理员权限
def admin_only(func):
    def wrapper(user, *args, **kwargs):
        if user.role != 'admin':
            raise PermissionError(f"User {user.username} is not "
                                  f"authorized to perform this action")
        return func(user, *args, **kwargs)
    return wrapper

@admin_only
def delete_user(user, user_to_delete):
    """只有管理员可以删除用户"""
    print(f'User {user_to_delete.username} has been deleted by {user.username}')

# 测试代码
admin_user = User('Alice', 'admin')
normal_user = User('Bob', 'user')

# 尝试以管理员身份删除用户
delete_user(admin_user, normal_user)

# 尝试以普通用户身份删除用户
delete_user(normal_user, admin_user)
相关推荐
ad禥思妙想2 小时前
如何运行python脚本
开发语言·python
Matlab程序猿小助手2 小时前
【MATLAB源码-第218期】基于matlab的北方苍鹰优化算法(NGO)无人机三维路径规划,输出做短路径图和适应度曲线.
开发语言·嵌入式硬件·算法·matlab·机器人·无人机
威威猫的栗子2 小时前
用 Python 与 Turtle 创作属于你的“冰墩墩”!
开发语言·python·turtle
IT古董2 小时前
【机器学习】超简明Python基础教程
开发语言·人工智能·python·机器学习
黑客Ash2 小时前
网络安全知识点
开发语言·php
童先生2 小时前
如何将java项目打包成docker 镜像并且可运行
java·开发语言·docker
feilieren2 小时前
SpringBoot 2.x 整合 Redis
java·开发语言·spring
晓看天色*2 小时前
[JAVA]MyBatis框架—获取SqlSession对象
java·开发语言·前端
捕鲸叉2 小时前
C++创建型模式之生成器模式
开发语言·c++·建造者模式
qq_q9922502772 小时前
django基于python 语言的酒店推荐系统
后端·python·django