python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python装饰器

深度解析Python装饰器常见用法与进阶技巧

作者:天天进步2015

Python 装饰器(Decorator)是提升代码可读性与复用性的强大工具,本文将深入解析 Python 装饰器的原理,常见用法,进阶技巧与最佳实践,希望可以帮助大家写出更具专业水准的 Python 代码

装饰器的基本原理

装饰器本质上是一个高阶函数,即:以函数为参数并返回新函数的函数。它可以在不修改原始函数代码的前提下,动态地为其添加功能。

def my_decorator(func):
    def wrapper(*args, **kwargs):
        print('Before function call')
        result = func(*args, **kwargs)
        print('After function call')
        return result
    return wrapper

@my_decorator
def say_hello():
    print('Hello, world!')

say_hello()

输出:

Before function call
Hello, world!
After function call

函数装饰器的常见用法

1.日志记录

import logging

def log_decorator(func):
    def wrapper(*args, **kwargs):
        logging.info(f'Calling {func.__name__}')
        return func(*args, **kwargs)
    return wrapper

2.权限校验

def require_admin(func):
    def wrapper(user, *args, **kwargs):
        if not user.is_admin:
            raise PermissionError('Admin required')
        return func(user, *args, **kwargs)
    return wrapper

3.性能分析

import time

def timeit(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print(f'{func.__name__} took {end - start:.4f}s')
        return result
    return wrapper

4.缓存机制

from functools import lru_cache

@lru_cache(maxsize=128)
def fib(n):
    return n if n < 2 else fib(n-1) + fib(n-2)

带参数的装饰器

带参数的装饰器需要再包一层函数:

def repeat(times):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(times):
                result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator

@repeat(3)
def greet():
    print('Hello!')

类装饰器与方法装饰器

类装饰器

可以用于增强类的功能:

def singleton(cls):
    instances = {}
    def get_instance(*args, **kwargs):
        if cls not in instances:
            instances[cls] = cls(*args, **kwargs)
        return instances[cls]
    return get_instance

@singleton
class Database:
    pass

方法装饰器

与函数装饰器类似,但要注意 self 的传递:

def method_logger(func):
    def wrapper(self, *args, **kwargs):
        print(f'Calling {func.__name__} of {self}')
        return func(self, *args, **kwargs)
    return wrapper

class MyClass:
    @method_logger
    def foo(self):
        print('foo called')

装饰器的嵌套与组合

多个装饰器可以叠加使用,执行顺序为自下而上:

@decorator_a
@decorator_b
def func():
    pass
# 等价于 func = decorator_a(decorator_b(func))

进阶技巧:保留元信息与类型提示

装饰器会改变被装饰函数的元信息(如__name__, __doc__),推荐用 functools.wraps 保留原信息:

from functools import wraps

def my_decorator(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)
    return wrapper

对于类型提示,可以直接在装饰器和被装饰函数中添加 type hints。

装饰器最佳实践

总结

Python 装饰器让我们以声明式、可复用的方式增强函数和类的行为。掌握装饰器的原理与用法,不仅能提升代码的可读性和复用性,更能让你的 Python 项目更具专业水准。建议在实际开发中多加练习,灵活运用装饰器解决实际问题。

到此这篇关于深度解析Python装饰器常见用法与进阶技巧的文章就介绍到这了,更多相关Python装饰器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文