python文件路径|极客教程

python文件路径

1. 引言

装饰器是Python中非常有用的一种编程技巧，它允许我们在不修改源代码的情况下，对已有函数、方法或类进行扩展或修改。本文将详细介绍装饰器的定义、用法和实际应用，帮助读者深入了解装饰器在Python编程中的重要性和灵活性。

2. 装饰器的定义

装饰器本质上是一个callable对象（函数、方法或类），它接收一个函数作为输入，并返回一个新的函数。装饰器的作用是在被装饰的函数（或方法、类）执行前后添加额外的功能。装饰器可以显式地使用@符号来应用，也可以通过函数调用的形式来应用。

3. 装饰器的基本用法

3.1 使用装饰器包装函数

下面是一个简单的装饰器示例，用于在函数执行前后打印日志信息：

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print("开始执行函数:", func.__name__)
        result = func(*args, **kwargs)
        print("函数执行完毕:", func.__name__)
        return result
    return wrapper

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

print(add(1, 2))  # 输出：开始执行函数: add; 函数执行完毕: add; 3

3.2 使用装饰器包装类方法

装饰器不仅可以用于函数，还可以用于类的方法。下面是一个示例，使用装饰器记录类方法的执行时间：

import time

def runtime_decorator(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"函数 {func.__name__} 的执行时间为：{end_time - start_time:.4f} 秒")
        return result
    return wrapper

class MyClass:
    @runtime_decorator
    def heavy_computation(self, num):
        # 模拟耗时的计算
        time.sleep(2)
        return num * num

my_obj = MyClass()
my_obj.heavy_computation(10)  # 输出：函数 heavy_computation 的执行时间为：2.0000 秒

3.3 使用装饰器包装类

装饰器不仅可以用于函数、方法，也可以用于类本身。下面是一个示例，使用装饰器给类增加一个新的方法：

def new_method_decorator(cls):
    cls.new_method = lambda self: print("这是一个新方法")
    return cls

@new_method_decorator
class MyClass:
    def existing_method(self):
        print("这是一个已有的方法")

my_obj = MyClass()
my_obj.existing_method()  # 输出：这是一个已有的方法
my_obj.new_method()       # 输出：这是一个新方法

4. 装饰器的实际应用

装饰器在实际的开发中非常有用，可以应用于许多场景。下面介绍几个常见的应用场景。

4.1 记录日志

装饰器可以用于记录函数、方法的执行日志，便于排查问题和了解程序的运行情况。示例代码如下：

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print(f"开始执行函数: {func.__name__}")
        result = func(*args, **kwargs)
        print(f"函数执行完毕: {func.__name__}")
        return result
    return wrapper

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

add(1, 2)  # 输出：开始执行函数: add; 函数执行完毕: add

4.2 记录函数执行时间

装饰器可以用于记录函数、方法的执行时间，便于优化性能和定位慢速部分。示例代码如下：

import time

def runtime_decorator(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"函数 {func.__name__} 的执行时间为：{end_time - start_time:.4f} 秒")
        return result
    return wrapper

@runtime_decorator
def heavy_computation(num):
    # 模拟耗时的计算
    time.sleep(2)
    return num * num

heavy_computation(10)  # 输出：函数 heavy_computation 的执行时间为：2.0000 秒

4.3 权限校验

装饰器可以用于验证用户的权限，确保只有有权的用户可以访问某些函数或方法。示例代码如下：

def login_required(func):
    def wrapper(*args, **kwargs):
        if is_logged_in():
            return func(*args, **kwargs)
        else:
            return "您尚未登录，无权访问该功能"
    return wrapper

@login_required
def secret_operation():
    return "这是一个需要登录的秘密操作"

print(secret_operation())  # 输出：您尚未登录，无权访问该功能