Python 详解修饰器附带 js使用修饰器

修饰器

功能

修饰器的主要功能是，在不改变已有代码的情况下，为某一个类，方法等扩展功能

首先看这样一段代码

def foo():

    for i in range(10):

        print(i)

foo()

添加需求打印日志

现在需要在不改变这段代码前提下,计算出这段代码的运行时间日志

import time

def log(fun):

    print('开始时间:{}'.format(time.time()))

    fun()

    print('结束时间:{}'.format(time.time()))

def foo():

    for i in range(10):

        print(i)

log(foo)

优化调用方式处理多次调用

我们将foo函数当成一个参数传递给了log(),在def log(fun)函数中，fun()就等同于foo(),好，现在可以显示运行的时间日志了，但是这样就改变了调用方式需要使用log(foo)来调用，假设代码中使用了100次foo()现在需要打印每个foo()的运行日志，全部这样改是不现实的，我们需要优化一下

import time

def log(fun):

    def run_log():

        print('开始时间:{}'.format(time.time()))

        fun()

        print('结束时间:{}'.format(time.time()))

    return run_log

def foo():

    for i in range(10):

        print(i)

foo = log(foo)

foo()

现在我们改造了下代码，foo = log(foo)时，调用log()方法，内部定义了一个run_log()方法，然后返回给了foo，此时foo等于run_log，调用foo相当于调用run_log 这样,即实现了需要打印日志的需求，又可以不用去修改已有代码

传参

现在的foo()方法可以打印0到9数字，但是，由于业务需求变更，现在需要给foo()方法传递一个值，例如foo(100)，就需要打印出0-99的数字，同时还要打印日志，所以我们需要再次优化代码

import time

def log(fun):

    def run_log(num):

        print('开始时间:{}'.format(time.time()))

        fun(num)

        print('结束时间:{}'.format(time.time()))

    return run_log

def foo(num):

    for i in range(num):

        print(i)

foo = log(foo)

foo(100)

根据之前的逻辑，我们已经知道新的foo()等于run_log(),所以我们给foo(100)传递100的值时，实际上等同于run_log(100)，所以我们直接在def foo(num)中是接受不到的，我们需要在run_log(num)中接受到参数，传给fun(num)，这样新的函数就可以接受参数了

返回参数

新的需求又来了，需要在foo()中返回一个所有数字的累加之和，而我们现有的foo函数实际上是run_log，所以我们再来改造一下代码

import time

def log(fun):

    def run_log(num):

        print('开始时间:{}'.format(time.time()))

        info = fun(num)

        print('结束时间:{}'.format(time.time()))

        return info

    return run_log

def foo(num):

    add_num = 0

    for i in range(num):

        add_num += i

        print(i)

    return add_num

foo = log(foo)

x = foo(100)

print(x)

由于现在的foo()等同于run_log(),run_log()中的fun()相当于foo()，所以foo中返回的值传到了info中然后我们把info返回，x就可以接受到从run_log中传递出来的参数

好了，到这里我们就实现了一个阉割版修饰器

通用性

假如需要你再给一个新的函数foo2打印日志，代码如下

import time

def log(fun):

    def run_log(num):

        print('开始时间:{}'.format(time.time()))

        info = fun(num)

        print('结束时间:{}'.format(time.time()))

        return info

    return run_log

def foo(num):

    add_num = 0

    for i in range(num):

        add_num += i

        print(i)

    return add_num

def foo2(num, num2):

    add_num = 0

    for i in range(num, num2):

        add_num += i

        print(i)

    return add_num

foo = log(foo)

x = foo(100)

print(x)

由于我们的修饰器只能接收一个参数，而foo2需要两个参数，现有代码无法实现，所以我们要继续升级代码

import time

def log(fun):

    def run_log(*args,**kwargs):

        print('开始时间:{}'.format(time.time()))

        info = fun(*args,**kwargs)

        print('结束时间:{}'.format(time.time()))

        return info

    return run_log

def foo(num):

    add_num = 0

    for i in range(num):

        add_num += i

        print(i)

    return add_num

def foo2(num, num2):

    add_num = 0

    for i in range(num, num2):

        add_num += i

        print(i)

    return add_num

foo = log(foo)

foo2 = log(foo2)

x = foo(100)

x2 = foo2(50,100)

print(x)

print(x2)

我们使用了*args，这样就可以接收任意数量的参数，可以满足我们的所有需求

语法糖

但是每次使用前都要写一个 foo = log(foo)这样的赋值操作，代码并不美观不符合python的代码风格，所以python给提供了一种语法糖，可以用@log来替代,修改后代码如下

import time

def log(fun):

    def run_log(*args,**kwargs):

        print('开始时间:{}'.format(time.time()))

        info = fun(*args,**kwargs)

        print('结束时间:{}'.format(time.time()))

        return info

    return run_log

@log

def foo(num):

    add_num = 0

    for i in range(num):

        add_num += i

        print(i)

    return add_num

@log

def foo2(num, num2):

    add_num = 0

    for i in range(num, num2):

        add_num += i

        print(i)

    return add_num

x = foo(100)

x2 = foo2(50, 100)

print(x)

print(x2)

我们只需要在定义函数时，在上面添加一句@修饰器名就相当于完成了函数名 = 修饰器名(函数名)这样的操作

现在就已经是一个标准的修饰器了

扩展

同理，我们用js也可以写一个修饰器

function log(fun) {

	function run_log(...ags) {

		console.log('==============')

		let info = fun(...ags)

		console.log('==============')

		return info

	}

	return run_log

}

foo = log(foo)

function foo(num, num1) {

	let x = 0

	for (let i = num; i < num1; i++) {

		console.log(i)

		x += i

	}

	return x

}

foo2 = log(foo2)

function foo2(num) {

	let x = 0

	for (let i = 0; i < num; i++) {

		console.log(i)

		x += i

	}

	return x

}

foo(10, 100)

foo2(10)

巴特西

Python 详解修饰器附带 js使用修饰器

修饰器

功能

添加需求打印日志

优化调用方式处理多次调用

传参

返回参数

通用性

语法糖

扩展

最新文章

热门文章

巴特西

Python 详解修饰器 附带 js使用修饰器

修饰器

功能

添加需求 打印日志

优化调用方式 处理多次调用

传参

返回参数

通用性

语法糖

扩展

最新文章

热门文章

Python 详解修饰器附带 js使用修饰器

添加需求打印日志

优化调用方式处理多次调用