3分钟带你了解Python装饰器(Decorators)离成为大佬又近了一步

装饰器(Decorators)是Python中一种强大的编程工具,它允许您在不修改被装饰函数源代码的情况下,增加、修改或扩展函数的功能。装饰器本质上是一个函数,它接受一个函数作为输入,并返回一个新的函数。装饰器在实际开发中常用于代码重用、日志记录、性能分析、权限检查等方面。

下面是一个装饰器的基本结构

Python 复制代码
def decorator_function(original_function):
    def wrapper_function(*args, **kwargs):
        # 添加装饰逻辑
        result = original_function(*args, **kwargs)
        # 返回被装饰函数的结果
        return result
    return wrapper_function

接下来我会举三个🌰🌰🌰来进一步带你们了解Python装饰器的用法,相信会对你们有所帮助👏👏👏(此处应该有掌声)

第一个🌰:

计时装饰器

这个装饰器可以用来计算函数的执行时间,比如在对一些函数做性能分析时,往往需要统计执行时间,有了装饰器我们就不需要写重复代码了,只需要给需要计算执行时间的方法加上装饰器即可。

Python 复制代码
# 这个装饰器可以用来计算函数的执行时间。
import time

# 这里是装饰器的定义
def timing_decorator(original_function):
    def wrapper_function(*args, **kwargs):
        start_time = time.time()
        result = original_function(*args, **kwargs)
        end_time = time.time()
        print(f"函数{original_function.__name__}执行使用了{end_time - start_time}秒")
        return result
    return wrapper_function

# @后面跟着的方法就是我们定义的装饰器了
@timing_decorator
def slow_function():
    time.sleep(2)

#这里调用了slow_function(),会输出方法所用时间
slow_function()

运行结果如下:

第二个🌰:

权限检验装饰器

这个装饰器可以用来检查用户是否有足够的权限执行某个操作。示例装饰器有1个入参,输入的是用户角色,比如只有角色是admin和root的才有对方法do_something()有操作的权限。

Python 复制代码
# 这个装饰器可以用来检查用户是否有足够的权限执行某个操作。

def permission_required(permission):
    def decorator(original_function):
        def wrapper_function(*args, **kwargs):
            if has_permission(permission):
                return original_function(*args, **kwargs)
            else:
                # 装饰器内部使用了 raise 关键字来抛出异常,以便在特定条件下阻止函数继续执行。
                raise PermissionError("没有操作权限") 
        return wrapper_function
    return decorator

def has_permission(user_role):
    user_roles = ['admin','root']
    if user_role in user_roles:
        return True
    else:
        return False
    
@permission_required("admin111")
def do_something(resource_id):
    # 有权限操作
    print(f"您有操作【{resource_id}】 的权限")

do_something('查看用户交易记录')

运行结果如下: 当@permission_required("admin111")的入参是admin111时输出结果如下:

@permission_required("admin")的入参是admin时输出结果如下:

第三个🌰:

日志装饰器

这个装饰器可以用来记录函数的调用和参数。

Python 复制代码
# 这个装饰器可以用来记录函数的调用和参数。
def log_arguments(original_function):
    def wrapper_function(*args, **kwargs):
        print(f"调用了:{original_function.__name__}方法,传入的args为: {args}传入的kwargs为: {kwargs}")
        result = original_function(*args, **kwargs)
        return result
    return wrapper_function

@log_arguments
def add(a, b):
    return a + b

result = add(3, 5)

运行结果如下:


备注

raise 是Python中用于引发异常的关键字。当您在代码中使用 raise 关键字时,您实际上是在抛出一个异常,从而指示程序出现了错误或异常情况。

在上述示例中,装饰器内部使用了 raise 关键字来抛出异常,以便在特定条件下阻止函数继续执行。

相关推荐
Deng94520131418 分钟前
基于Python的旅游数据可视化应用
python·numpy·pandas·旅游·数据可视化技术
2401_8786247920 分钟前
pytorch 自动微分
人工智能·pytorch·python·机器学习
胖达不服输23 分钟前
「日拱一码」021 机器学习——特征工程
人工智能·python·机器学习·特征工程
screenCui1 小时前
macOS运行python程序遇libiomp5.dylib库冲突错误解决方案
开发语言·python·macos
小眼睛羊羊2 小时前
pyinstaller打包paddleocr
python
java1234_小锋2 小时前
基于Python的旅游推荐协同过滤算法系统(去哪儿网数据分析及可视化(Django+echarts))
python·数据分析·旅游
蓝婷儿2 小时前
Python 机器学习核心入门与实战进阶 Day 4 - 支持向量机(SVM)原理与分类实战
python·机器学习·支持向量机
%d%d22 小时前
python 在运行时没有加载修改后的版本
java·服务器·python
amazinging3 小时前
北京-4年功能测试2年空窗-报培训班学测开-第四十七天
python·学习·selenium
Freak嵌入式3 小时前
一文速通 Python 并行计算:13 Python 异步编程-基本概念与事件循环和回调机制
开发语言·python·嵌入式·协程·硬件·异步编程