Python之装饰器的使用

一、装饰器

装饰器：把一个函数当作参数，返回一个替代版的函数，本质就是一个返回函数的函数

作用：在不改变原函数的基础上，给函数增加功能

二、装饰器的实现

（1）包含可变参数的装饰器

import time
def decorator(func):        #定义装饰器
    def wrapper(*args,**kwargs):        #实际功能函数
        print(time.time())
        func(*args,**kwargs)
    return wrapper            #返回一个功能函数

@decorator        #使用@语法糖的方式调用
def f1():
    print('This is a function')
f1()        #调用函数

（2）包含关键字的装饰器

import time
def decorator(func):
    def wrapper(*args,**kwargs):
        print(time.time())
        func(*args,**kwargs)
    return wrapper
@decorator
def f(func_name1,func_name2,**kwargs):
    print('This is function ' + func_name1)
    print('This is function ' + func_name2)
    print(kwargs)
f('test1','test2',a=1,b=2,c='westos')

练习：用装饰器实现一个函数计时器

import random
import string
import functools
li = [random.choice(string.ascii_letters) for i in range(100)]
def timec(fun):
    """这是一个装饰器timec"""
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        """这是一个wrapper函数"""
        #args:元组 kwargs:字典
        #函数执行前
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('运行时间为: %.6f' %(end_time - start_time))
        return res
    return wrapper
@timec
def con_add():
    """这是一个con_add函数"""
    s = ''
    for i in li:
        s += (i + ',')
    print(s)
@timec
def join_add():
    print(','.join(li))
con_add()
join_add()
print(con_add.__doc__)        #帮助文档
print(con_add.__name__)        #函数名称

练习：创建装饰器， 要求如下：

1. 创建add_log装饰器， 被装饰的函数打印日志信息；

2. 日志格式为: [字符串时间] 函数名: xxx， 运行时间：xxx，运行返回值结果:xxx

import functools
def add_log(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('[%s] 函数名: %s, 运行时间: %.6f, 运行返回值结果: %d'
              %(time.ctime(),fun.__name__,end_time - start_time,res))
        return res
    return wrapper
@add_log
def add(x,y):
    time.sleep(1)
    return x + y
add(1,2)

练习：用户管理，root用户可添加或删除信息，其他用户则返回权限不允许

import functools
import inspect
def is_admin(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        #inspect.getcallargs返回一个字典,key值:形参 value值:对应的实参
        inspect_res = inspect.getcallargs(fun,*args,**kwargs)
        print('inspect的返回值: %s' %inspect_res)
        if inspect_res.get('name') == 'root':
            res = fun(*args,**kwargs)
            return res
        else:
            print('Permission denied!')
    return wrapper
@is_admin
def add_student(name):
    print('添加学生信息')
def del_student(name):
    print('删除学生信息')
add_student('westos')