Python之装饰器（综合练习）

装饰器

案例1：

创建装饰器，要求如下：

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

代码：

import time
import functools

# 定义装饰器
def add_log(fun):
    # 保留被装饰函数的函数名和帮助信息文档
    @functools.wraps(fun)
    def inter(*args,**kwargs):
        # time.time():  秒数
        start_time = time.time()
        # 由于被装饰函数有返回值
        res = fun(*args,**kwargs)
        end_time = time.time()
        # 运行返回值结果:即为函数的返回值,fun.__name__:返回函数名
        print('[%s] 函数名:%s, 运行时间:%.6f, 运行返回值结果:%d'
              %(time.ctime(),fun.__name__,end_time - start_time,res))
        # 所以必须return返回函数返回值
        return res
    return inter

# 添加装饰
@add_log
# 定义被装饰函数(求任意两个数的和)
def add(x,y):
    # 休眠1秒
    time.sleep(1)
    return x+y

# 调用函数
print(add(1,10))

# 输出时间; time.ctime():以正常的格式显示时间
# print(time.ctime())

运行结果：

案例2：

需求：
仅root用户可以有权限执行以下函数，当普通用户执行时会报错

#提示: 如何返回函数的实参?
      inspect.getcallargs(函数名,参数):以字典的形式返回函数的参数 ( {形参:实参} )
      get方法:返回字典指定key的value值

代码：

import functools
import inspect

# 定义装饰器
def is_root(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        # inspect.getcallargs:以字典的形式返回函数的参数
        # key值为:形参,value值为:形参对应的实参
        inspect_res = inspect.getcallargs(fun,*args,**kwargs)
        print('inspect_res的返回值为:%s' % inspect_res)
        # get方法:返回字典指定key的value值
        if inspect_res.get('name') == 'root':
            res = fun(*args,**kwargs)
            return res
        else:
             print('ERROR:you are not a root user')
    return  wrapper

# 添加装饰
@is_root
# 定义被装饰的函数
def add_student(name):
    print('添加学生信息...')

# 调用函数
add_student('westos')

运行结果：

改进：（需要用到多个装饰器）

先让用户登陆，再判断用户是否由权限执行添加信息功能

代码：

import functools
import inspect

userlist = ['root','admin','student']

# 定义装饰器1(判断用户是否登陆成功)
def user_login(fun):
    @functools.wraps(fun)
    def logins(*args,**kwargs):
        for i in args:
            if i not in userlist:
                print('Error:登陆失败')
            else:
            	print('登陆成功')
                res = fun(*args,**kwargs)
                return res
    return logins

# 定义装饰器2(判断用户是否为root用户)
def is_root(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        # inspect.getcallargs:以字典的形式返回函数的参数
        # key值为:形参,value值为:形参对应的实参
        inspect_res = inspect.getcallargs(fun,*args,**kwargs)
        print('inspect_res的返回值为:%s' % inspect_res)
        # get方法:返回字典指定key的value值
        if inspect_res.get('name') == 'root':
            res = fun(*args,**kwargs)
            return res
        else:
             print('ERROR:you are not a root user')
    return  wrapper

# 添加装饰(装饰器的内部功能函数从上到下依次执行)
@user_login
@is_root
# 定义被装饰的函数
def add_student(name):
    print('添加学生信息...')

# 调用函数
add_student('admin')

运行结果：

案例3：

编写装饰器required_ints, 条件如下:
    1). 确保函数接收到的每一个参数都是整数;
        # 提示：如何判断变量的类型?
          type(s)             # 输出s的类型
          isinstance(s,str)   # 判断s是否为字符串类型
    2). 如果参数不是整形数，抛出异常: 参数必须为整形;
        # 提示：如何抛出异常?
          raise TypeError(“参数必须为整形”)

代码：

import functools
# 定义装饰器
def required_ints(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        # 遍历整个想要求和的元组
        for i in args + tuple(kwargs.values()):
            # 判断每个元素是否为整型,如果不是,则抛出异常
            if not isinstance(i,int):
                # raise TypeError(): 抛出异常
                raise TypeError('参数必须为整形')
        else:
            # 调用函数,并返回函数返回值
            res = fun(*args,**kwargs)
            return res
    return wrapper
    
# 添加装饰
@required_ints
# 定义被装饰函数:
# *args:可变参数  **kwargs:关键字参数
def add(*args,**kwargs):
    # args:字典类型   kwargs.values():元组类型
    # 由于数据类型不同,所以不能直接用+来连接,必须先转换数据类型
    return sum(args + tuple(kwargs.values()))

# 注意：如果有字典类型的实参，则必须写在最后边
print(add(1,2.0,a=3,b=5))