Python——装饰器

装饰器

装饰器：
把一个函数当作参数,返回一个替代版的函数
本质就是一个返回函数的函数
作用：在不改变原函数的基础上,给函数增加功能
举例如下：
没有使用装饰器的情况：

def func1():
    print('~~~~~~~~~~~~~~')
    print('hello world')
def outer():
    func1()
    print('~~~~~~~~~~~~~~')
func1()
outer()

使用装饰器：
举例一：

import time
def decorator(func):    #装饰器
    def wrapper(*args,**kwargs):    #装饰器内部的函数
        print(time.time())     #linux系统的时间，从1970年1月1日开始计时
        func(*args,**kwargs)     #调用的函数
    return wrapper

@decorator    #给函数添加装饰器
def f1():
    print('This is a function')
f1()   

举例二：装饰器设置可变形参

import time
def decorator(func):
    def wrapper(*args,**kwargs):
        print(time.time())
        func(*args,**kwargs)
    return wrapper

@decorator
def f2(func_name1,func_name2):
    print('This is function ' +func_name1)
    print('This is function ' +func_name2)
f2('test1','test2')

举例三：装饰器设置关键字参数

import time
def decorator(func):
    def wrapper(*args,**kwargs):
        print(time.time())
        func(*args,**kwargs)
    return wrapper

@decorator
def f3(func_name1,func_name2,**kwargs):
    print('This is function '+func_name1)
    print('This is function '+func_name2)
    print(kwargs)
f3('test1','test2',a=1,b=2,c='hello')

装饰器练习

练习一：装饰器实现一个函数计时器
1.被装饰的函数有返回值
2.如何保留被装饰函数的函数名和帮助信息
代码如下：

import random
import string
import time
import functools

li = [random.choice(string.ascii_letters) for i in range(100)]
print(li)

def timec(fun):
    """这是一个装饰器timec"""
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):   #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():
    """这是一个join_add函数"""
    print(','.join(li))

con_add()
join_add()

print(con_add.__doc__)
print(con_add.__name__)

@timec
def fun_list(n):
    return [2 * i for i in range(n)]

@timec
def fun_map(n):
    return list(map(lambda x:x*2,range(n)))

print(fun_list(100))
print(fun_map(100))

运行结果如下：

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

1. 创建add_log装饰器， 被装饰的函数打印日志信息；
2. 日志格式为: [字符串时间] 函数名: xxx， 运行时间：xxx, 运行返回值结果:xxx

代码如下：

import time
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,3)

练习三：
判断是否为管理员root登陆，如果为root登陆，则显示相应的信息
inspect.getcallargs返回一个字典,key值:形参 value值:对应的实参

import functools
import inspect
def is_admin(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        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('root')

执行结果：