正文

如何用Python函数装饰器轻松增强你的代码功能，提高代码可读性和复用性

/0 浏览量
1104

在Python编程中，函数装饰器是一种非常强大的特性，它允许我们以不修改原始函数代码的方式，给函数增加额外的功能。使用装饰器可以提高代码的可读性、复用性，并使代码更加模块化。下面，我将详细介绍一下如何使用Python函数装饰器。

什么是装饰器？

装饰器本质上是一个返回函数的函数。它允许你在不改变函数定义的情况下，为函数添加新的功能。装饰器的语法如下：

def decorator_name(func):
    def wrapper():
        # 在这里可以添加功能
        func()
        # 在这里也可以添加功能
    return wrapper

@decorator_name
def original_function():
    print("This is the original function.")

在上面的例子中，decorator_name 是一个装饰器，它返回了一个名为 wrapper 的函数。当我们调用 original_function() 时，实际上是调用了 wrapper()。这样，我们就可以在 wrapper() 函数中添加任何想要在 original_function() 执行前后运行的功能。

装饰器的应用场景

1. 访问控制

使用装饰器可以实现简单的访问控制，例如：

def private(func):
    def wrapper(*args, **kwargs):
        if not isinstance(args[0], User):
            raise PermissionError("Access denied")
        return func(*args, **kwargs)
    return wrapper

@private
def private_function(user, message):
    print(f"{user.name} says: {message}")

2. 日志记录

装饰器可以用来添加日志记录功能：

import logging

def log(func):
    def wrapper(*args, **kwargs):
        logging.info(f"Calling {func.__name__} with args: {args} and kwargs: {kwargs}")
        result = func(*args, **kwargs)
        logging.info(f"Finished {func.__name__} with result: {result}")
        return result
    return wrapper

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

3. 函数计时

装饰器可以用来测量函数的执行时间：

import time

def timer(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 run.")
        return result
    return wrapper

@timer
def heavy_computation():
    time.sleep(2)

4. 缓存结果

装饰器可以用来缓存函数的返回值，从而提高效率：

def memoize(func):
    cache = {}

    def wrapper(*args):
        if args in cache:
            return cache[args]
        result = func(*args)
        cache[args] = result
        return result
    return wrapper

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

总结

函数装饰器是Python中一个非常实用且强大的特性。通过使用装饰器，我们可以轻松地为函数添加额外功能，提高代码的可读性和复用性。掌握装饰器可以帮助我们编写更加优雅、高效的代码。

-- 展开阅读全文 --