Python装饰器

小半 • 2024年3月21日下午12:04 • Python • 阅读 50

装饰器（Decorator）是Python中一种强大而灵活的特性，它可以在不修改原函数代码的情况下，对函数进行扩展、增强或修改功能。本文将详细介绍Python装饰器的概念、用法以及常见的应用场景，帮助您深入理解和运用装饰器提升代码的可读性和可维护性。

一、装饰器的基本概念

在Python中，装饰器是一种高阶函数，它接受一个函数作为参数，并返回一个新的函数。装饰器通常用于在不修改原函数代码的情况下，对函数的行为进行扩展或修改。

def decorator(func):
    def wrapper():
        print("Before calling the function")
        func()
        print("After calling the function")
    return wrapper

@decorator
def greet():
    print("Hello, World!")

# 调用装饰后的函数
greet()

在上面的例子中，decorator函数就是一个装饰器，它接受一个函数作为参数，并返回一个新的函数wrapper。通过在函数定义前使用@decorator语法糖，可以将greet函数传递给decorator装饰器，并将返回的新函数绑定到greet函数上，从而实现对greet函数的装饰。

二、装饰器的应用场景

认证与授权
装饰器可以用于实现认证与授权功能，例如检查用户是否已登录、是否具有特定权限等。

def login_required(func):
    def wrapper(*args, **kwargs):
        if check_login():
            return func(*args, **kwargs)
        else:
            return "Please login first"
    return wrapper

@login_required
def secret_page():
    return "This is a secret page"

# 调用需要认证的函数
print(secret_page())

三、日志记录

装饰器可以用于记录函数的调用信息，例如函数的参数、返回值等。

def log(func):
    def wrapper(*args, **kwargs):
        print(f"Calling {func.__name__} with args: {args}, kwargs: {kwargs}")
        result = func(*args, **kwargs)
        print(f"{func.__name__} returned: {result}")
        return result
    return wrapper

@log
def add(x, y):
    return x + y

# 调用带有日志记录的函数
print(add(3, 5))

四、性能分析

装饰器可以用于对函数的性能进行分析和优化，例如计算函数的执行时间等。

import time

def performance(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} took {end_time - start_time} seconds to execute")
        return result
    return wrapper

@performance
def fibonacci(n):
    if n <= 1:
        return n
    else:
        return fibonacci(n-1) + fibonacci(n-2)

# 调用带有性能分析的函数
print(fibonacci(30))