装饰器学习笔记3:简单的函数装饰器

诶嘿,终于进入正题了(⊙﹏⊙)b

• 使用场景
   基于软件设计的开放封闭原则,必须在不对原代码修改的情况下来对代码进行功能扩展,这时就用到装饰器.

•  本质上,装饰器其实就是一个函数,是一个多层嵌套函数,这个函数实现了一个闭包.通过装饰器可以给其他函数进行功能扩展.

 有一个简单的函数:

def say():
    print('hello')

 这时我想做个记录,记录下每次执行的时间日期,可以这么搞

import time

def say():
    now = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
    with open('data','a+') as f:
        f.write('{}执行了'.format(now))
    print('hello')

 当我有多个函数都需要记录时,每一个函数进行修改比较麻烦,不但违反了封闭开放原则,而且还产生了重复代码,所以可以这样写:

import time

def log(func):
    def inner():
        now = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
        with open('data','a+') as f:
            f.write('{}执行了'.format(now))
        func()
    return inner

def say():
    print('hello')

say = log(say)
say()

 这样用一个函数来包装另外一个函数.就是个简单的装饰器了.
 不过这样还是比较麻烦,每个函数都要进行一次赋值,可以用@来精简一下.

import time

def log(func):
    def inner():
        now = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
        with open('data','a+') as f:
            f.write('{}执行了'.format(now))
        func()
    return inner

@log
def say():
    print('hello')

say()

 当有多个函数需要扩展时,只要每个函数def前增加@语法就可以了.这就完成了一个简单的装饰器.