Python函数使用：使用装饰器来增强函数功能

在 Python 中，使用装饰器来增强函数功能是一种常见的方法。装饰器可以在不修改函数源代码的情况下，给函数增加新的功能，这使得代码更加清晰简洁。本文将介绍 Python 中如何使用装饰器来增强函数功能。

1. 什么是装饰器

装饰器是一种特殊的 Python 函数，它可以在不修改原函数的情况下，给函数增加新的功能。装饰器的作用是将功能划分为独立的部分，从而使代码更加模块化和可复用。

装饰器的语法如下：

@decorator
def function():
    pass

其中，decorator 是装饰器函数，function 是被装饰的函数。装饰器函数接受被装饰函数作为参数，并返回一个新的函数，这个新的函数具有装饰器所增加的功能。

2. 装饰器的应用

装饰器可以用来实现很多功能，例如：

2.1 计算函数执行时间

装饰器可以用来计算函数的执行时间，这对于优化代码有很大帮助。下面是一个计算函数执行时间的装饰器：

import time

def timeit(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} used {end_time-start_time:.4f} seconds")
        return result
    return wrapper

以上代码中，timeit 是一个装饰器函数，使用 time 模块来计算函数的执行时间，然后返回一个新的函数 wrapper。被装饰的函数可以通过 @timeit 命令来调用装饰器：

@timeit
def myfunc():
    time.sleep(1)

此时，每次调用 myfunc 函数时，都会打印出函数执行的时间。

2.2 检查函数参数

装饰器可以用来实现参数检查的功能，比如检查函数参数的类型和值是否正确。下面是一个参数检查的装饰器：

def check_arg(*types):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for i, arg in enumerate(args):
                if types[i] != type(arg):
                    raise TypeError(f"argument {i+1} should be {types[i].__name__}")
            result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator

以上代码中，check_arg 是一个装饰器工厂函数，它返回一个参数检查装饰器。装饰器中的 *types 参数可以接受多个参数类型，如 int、str 等。被装饰的函数可以使用 @check_arg(int, str) 命令来检查函数的参数类型是否正确。

2.3 授权访问

装饰器可以用来授权访问，比如检查用户是否有访问权限。下面是一个访问授权的装饰器：

def authorize(access_level):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if check_access_level(access_level):
                result = func(*args, **kwargs)
                return result
            else:
                raise PermissionError("you don't have access to this resource")
        return wrapper
    return decorator

以上代码中，authorize 是一个装饰器工厂函数，它返回一个访问授权装饰器。装饰器中的 access_level 参数表示访问级别，被装饰的函数可以使用 @authorize("admin") 命令来限制只有管理员才能访问函数。

3. 总结

使用装饰器来增强函数功能，是 Python 编程中的常用技巧。装饰器可以划分函数的责任，使代码更加简洁和模块化。常见的装饰器应用包括计算函数执行时间、检查函数参数、授权访问等。熟练掌握装饰器的使用，可以使 Python 代码更加优雅和灵活。