理解Python装饰器：常见用法和错误示例

原创深碍√TFBOYSˉ_ 2024-12-14 20:45 90阅读 0赞

装饰器是Python中一种强大的特性，它允许你在不修改原函数代码的情况下，增加或修改其功能。

**常见的装饰器使用方法：**

1. **@decorator**: 这是应用装饰器的最基本方式。`decorator`是你定义的装饰器函数。

```python
def my_decorator(func):
    def wrapper():
        print("Something is happening before function execution!")
        func()
        print("Something is happening after function execution!")
    return wrapper

@my_decorator
def say_hello():
    print("Hello!")

say_hello()  # 输出：
# Something is happening before function execution!
# Hello!
# Something is happening after function execution!
```

2. **参数传递**：在装饰器中，你可以通过参数向原函数传递额外信息。

```python
def log_decorator(level="info"):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print(f"logging at {level}:")
            func(*args, **kwargs))
            print("\n-- End Function Execution --\n")
        return wrapper
    return decorator

@log_decorator(level="debug"))
def add(a, b):
    return a + b

result = add(1, 2)
print(result)  # 输出：3

# logging output
for line in log_decorator.__call__(add)(1, 2)).split("\n"):
    print(line.strip())
```

**常见错误示例：**

1. **装饰器定义的混淆**：如果装饰器和被装饰函数名字相似，可能会导致混淆。

```python
def my_func():
    # ...

@my_decorator  # 错误：装饰器名与原函数相同
def my_func():  # 被装饰函数
    # ...

my_decorator(my_func())  # 正确使用
```

2. **忘记返回装饰后的函数**：如果在装饰器中没有正确地返回被装饰的函数，会导致无法应用装饰。

```python
# 错误示例1
def my_decorator(func):
    def wrapper():
        pass
        func()
    return wrapper  # 这里返回了wrapper

@my_decorator  # 装饰失败：装饰器没有返回原函数
def my_func():  # 被装饰函数
    print("Hello!")

my_func()  # 错误：无法应用装饰，原函数未变

# 正确示例2
def my_decorator(func):
    def wrapper():
        print("Something is happening before function execution!")
        func()
        print("Something is happening after function execution!")
    return wrapper

@my_decorator
def my_func():  # 被装饰函数，这里返回了原函数
    print("Hello!")

my_func()  # 正确：应用装饰器后，原函数被替换为装饰后的版本。
```

以上就是Python装饰器的理解和常见错误示例。