拿着吧你！关于装饰器的魅力和强大

2023-12-01 00:19:26

修饰器是一种用来修改类或函数行为的强大工具，它们可以用来做许多事情，从类型检查到日志记录。装饰器可以让你在不修改类或函数本身的情况下修改它们的行为。

Python装饰器

Python装饰器是一种函数，它接受一个函数作为参数并返回一个新的函数。这个新的函数在执行时，会先执行装饰器的代码，然后执行函数本身的代码。装饰器可以通过@符号来使用。例如，以下代码使用装饰器来记录函数调用的时间：

def timing_decorator(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print('Function {} took {} seconds to run.'.format(func.__name__, end - start))
        return result
    return wrapper

@timing_decorator
def my_function():
    # Do something

当my_function被调用时，timing_decorator将被首先执行。timing_decorator将记录my_function的执行时间，然后执行my_function本身的代码。

JavaScript装饰器

JavaScript装饰器与Python装饰器类似，但它们使用不同的语法。JavaScript装饰器是通过一个函数来定义的，这个函数接受一个函数作为参数并返回一个新的函数。这个新的函数在执行时，会先执行装饰器的代码，然后执行函数本身的代码。装饰器可以通过@符号来使用。例如，以下代码使用装饰器来记录函数调用的时间：

const timingDecorator = (func) => {
    return (...args) => {
        const start = performance.now();
        const result = func(...args);
        const end = performance.now();
        console.log(`Function ${func.name} took ${end - start} milliseconds to run.`);
        return result;
    };
};

@timingDecorator
function myFunction() {
    // Do something
}

当myFunction被调用时，timingDecorator将被首先执行。timingDecorator将记录myFunction的执行时间，然后执行myFunction本身的代码。

修饰类

装饰器也可以用来修饰类。类装饰器是一种函数，它接受一个类作为参数并返回一个新的类。这个新的类继承了原来的类的所有属性和方法，但它可能有一些额外的功能。例如，以下代码使用装饰器来添加一个日志记录功能到类：

def logging_decorator(cls):
    class Wrapper(cls):
        def __init__(self, *args, **kwargs):
            super().__init__(*args, **kwargs)
            self.logger = logging.getLogger(self.__class__.__name__)

        def __call__(self, *args, **kwargs):
            self.logger.info('Calling {} with args {} and kwargs {}.'.format(self.__class__.__name__, args, kwargs))
            return super().__call__(*args, **kwargs)

    return Wrapper

@logging_decorator
class MyClass:
    def __init__(self, name):
        self.name = name

    def greet(self):
        print('Hello, {}!'.format(self.name))

当MyClass被实例化时，logging_decorator将被首先执行。logging_decorator将创建一个新的类Wrapper，Wrapper继承了MyClass的所有属性和方法，但它还添加了一个日志记录功能。当Wrapper的__init__方法被调用时，它将创建一个日志记录器并将其存储在self.logger属性中。当Wrapper的greet方法被调用时，它将记录一个日志消息，然后调用MyClass的greet方法。

修饰函数

装饰器也可以用来修饰函数。函数装饰器是一种函数，它接受一个函数作为参数并返回一个新的函数。这个新的函数在执行时，会先执行装饰器的代码，然后执行函数本身的代码。装饰器可以通过@符号来使用。例如，以下代码使用装饰器来记录函数调用的时间：

const timingDecorator = (func) => {
    return (...args) => {
        const start = performance.now();
        const result = func(...args);
        const end = performance.now();
        console.log(`Function ${func.name} took ${end - start} milliseconds to run.`);
        return result;
    };
};

@timingDecorator
function myFunction() {
    // Do something
}