python 之 装饰器(Decorators)

2024-03-22 18:44
文章标签 python 装饰 decorators

本文主要是介绍python 之 装饰器(Decorators),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

装饰器本质上也是一个函数,他可以让其他函数在不需要做任何代码变动的前提下,增加额外的功能,可以接受一个函数作为参数,并返回一个新的函数。

1. 装饰器的定义

装饰器的基本语法是使用@符号,后面跟着装饰器的名称。这个符号应该放在函数定义的上方。

2. 装饰器的示例代码

装饰器代码示例:

def my_decorator(func):  def wrapper():  print("Before function call")  func()  print("After function call")  return wrapper  @my_decorator  
def say_hello():  print("Hello!")  say_hello()

在这个例子中,my_decorator是一个装饰器。当它被应用到say_hello函数上时,它会返回一个新的函数wrapper。这个新的函数在调用原始函数say_hello之前和之后都打印一条消息。因此,当你调用say_hello()时,实际上是在调用wrapper()函数,它会首先打印"Before function call",然后调用say_hello()打印"Hello!“,最后打印"After function call”。

3. 常见的装饰器用法

  • 日志记录:装饰器可以用来在函数执行前后添加日志记录。这有助于跟踪函数的调用情况,以及在出现问题时进行调试。
import logging  def log_decorator(func):  def wrapper(*args, **kwargs):  logging.info(f"Calling {func.__name__} with args: {args} and kwargs: {kwargs}")  result = func(*args, **kwargs)  logging.info(f"{func.__name__} returned: {result}")  return result  return wrapper  @log_decorator  
def add_numbers(a, b):  return a + b  # 测试  
add_numbers(3, 4)
  • 性能分析:通过装饰器,你可以测量函数的执行时间,从而分析代码的性能。这对于优化性能瓶颈非常有用。
import time  def timing_decorator(func):  def wrapper(*args, **kwargs):  start_time = time.time()  result = func(*args, **kwargs)  end_time = time.time()  print(f"Function {func.__name__} took {end_time - start_time:.6f} seconds to execute.")  return result  return wrapper  @timing_decorator  
def slow_function():  time.sleep(1)  # 模拟耗时操作  # 测试  
slow_function()
  • 权限检查:装饰器可以用于实现权限检查,确保只有具有特定权限的用户才能调用特定的函数或方法。
def permission_decorator(func):  def wrapper(*args, **kwargs):  if not has_permission():  # 假设这是检查权限的函数  print("Permission denied!")  return None  return func(*args, **kwargs)  return wrapper  def has_permission():  # 这里应该是检查用户权限的逻辑  # 返回True表示有权限,返回False表示无权限  return True  # 示例中直接返回True,实际中需要根据用户或环境判断  @permission_decorator  
def sensitive_operation():  print("Executing sensitive operation...")  # 测试  
sensitive_operation()
  • 缓存结果:对于计算成本较高的函数,可以使用装饰器来缓存其结果,以便在后续调用时直接返回缓存的结果,从而提高效率。
def cache_decorator(func):  cache = {}  def wrapper(*args):  if args in cache:  return cache[args]  result = func(*args)  cache[args] = result  return result  return wrapper  @cache_decorator  
def fibonacci(n):  if n <= 1:  return n  return fibonacci(n-1) + fibonacci(n-2)  # 测试  
print(fibonacci(10))  # 第一次计算,结果会被缓存  
print(fibonacci(10))  # 第二次计算,直接从缓存中获取结果,不会重新计算
  • 函数注册:装饰器可以用于自动注册函数到某个注册表或容器中,方便后续的管理和调用。

  • 函数参数校验:装饰器可以在函数调用前对参数进行校验,确保传入的参数符合预期,从而避免在函数内部进行繁琐的参数检查。

  • 事务管理:对于需要数据库操作的函数,装饰器可以用来实现事务管理,确保在出现异常时能够回滚事务,保持数据的一致性。

  • 异常处理:装饰器可以用来统一处理函数抛出的异常,例如记录异常信息、发送报警邮件等。

  • 函数替换或增强:装饰器可以在不修改原始函数代码的情况下,替换或增强函数的功能。这对于扩展现有库或框架的功能非常有用。

4. pytest 框架中装饰器的用法

  • 组织测试用例:装饰器可以帮助你根据特定的条件或属性对测试用例进行分组或标记。例如,你可以使用@pytest.mark.skipif装饰器来跳过某些特定条件下不应该运行的测试用例。同样,你可以使用@pytest.mark.xfail来标记预期失败的测试用例。

  • 参数化测试:pytest提供了@pytest.mark.parametrize装饰器,允许你为测试函数提供多组参数和预期结果,从而执行参数化测试。这可以极大地减少冗余代码,并提高测试覆盖率。

  • 依赖管理和测试顺序:虽然pytest默认不保证测试函数的执行顺序,但你可以使用装饰器(结合其他机制)来管理测试之间的依赖关系。例如,你可以使用pytest-ordering插件和相应的装饰器来指定测试的执行顺序。

  • 自定义钩子:pytest具有一个强大的插件系统,允许你通过编写自定义插件来扩展其功能。在这些插件中,你可以定义自己的装饰器来实现特定的测试逻辑或行为。

  • 简化复杂逻辑:对于需要在多个测试用例中重复使用的复杂逻辑,你可以将其封装在装饰器中,从而减少测试代码的冗余。这使得测试用例更加简洁和易于维护。

这篇关于python 之 装饰器(Decorators)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/835975

相关文章

使用Python绘制蛇年春节祝福艺术图

《使用Python绘制蛇年春节祝福艺术图》:本文主要介绍如何使用Python的Matplotlib库绘制一幅富有创意的“蛇年有福”艺术图,这幅图结合了数字,蛇形,花朵等装饰,需要的可以参考下... 目录1. 绘图的基本概念2. 准备工作3. 实现代码解析3.1 设置绘图画布3.2 绘制数字“2025”3.3

python使用watchdog实现文件资源监控

《python使用watchdog实现文件资源监控》watchdog支持跨平台文件资源监控,可以检测指定文件夹下文件及文件夹变动,下面我们来看看Python如何使用watchdog实现文件资源监控吧... python文件监控库watchdogs简介随着Python在各种应用领域中的广泛使用,其生态环境也

Python中构建终端应用界面利器Blessed模块的使用

《Python中构建终端应用界面利器Blessed模块的使用》Blessed库作为一个轻量级且功能强大的解决方案,开始在开发者中赢得口碑,今天,我们就一起来探索一下它是如何让终端UI开发变得轻松而高... 目录一、安装与配置:简单、快速、无障碍二、基本功能:从彩色文本到动态交互1. 显示基本内容2. 创建链

Java调用Python代码的几种方法小结

《Java调用Python代码的几种方法小结》Python语言有丰富的系统管理、数据处理、统计类软件包,因此从java应用中调用Python代码的需求很常见、实用,本文介绍几种方法从java调用Pyt... 目录引言Java core使用ProcessBuilder使用Java脚本引擎总结引言python

python 字典d[k]中key不存在的解决方案

《python字典d[k]中key不存在的解决方案》本文主要介绍了在Python中处理字典键不存在时获取默认值的两种方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,... 目录defaultdict:处理找不到的键的一个选择特殊方法__missing__有时候为了方便起见,

使用Python绘制可爱的招财猫

《使用Python绘制可爱的招财猫》招财猫,也被称为“幸运猫”,是一种象征财富和好运的吉祥物,经常出现在亚洲文化的商店、餐厅和家庭中,今天,我将带你用Python和matplotlib库从零开始绘制一... 目录1. 为什么选择用 python 绘制?2. 绘图的基本概念3. 实现代码解析3.1 设置绘图画

Python pyinstaller实现图形化打包工具

《Pythonpyinstaller实现图形化打包工具》:本文主要介绍一个使用PythonPYQT5制作的关于pyinstaller打包工具,代替传统的cmd黑窗口模式打包页面,实现更快捷方便的... 目录1.简介2.运行效果3.相关源码1.简介一个使用python PYQT5制作的关于pyinstall

使用Python实现大文件切片上传及断点续传的方法

《使用Python实现大文件切片上传及断点续传的方法》本文介绍了使用Python实现大文件切片上传及断点续传的方法,包括功能模块划分(获取上传文件接口状态、临时文件夹状态信息、切片上传、切片合并)、整... 目录概要整体架构流程技术细节获取上传文件状态接口获取临时文件夹状态信息接口切片上传功能文件合并功能小

python实现自动登录12306自动抢票功能

《python实现自动登录12306自动抢票功能》随着互联网技术的发展,越来越多的人选择通过网络平台购票,特别是在中国,12306作为官方火车票预订平台,承担了巨大的访问量,对于热门线路或者节假日出行... 目录一、遇到的问题?二、改进三、进阶–展望总结一、遇到的问题?1.url-正确的表头:就是首先ur

基于Python实现PDF动画翻页效果的阅读器

《基于Python实现PDF动画翻页效果的阅读器》在这篇博客中,我们将深入分析一个基于wxPython实现的PDF阅读器程序,该程序支持加载PDF文件并显示页面内容,同时支持页面切换动画效果,文中有详... 目录全部代码代码结构初始化 UI 界面加载 PDF 文件显示 PDF 页面页面切换动画运行效果总结主