Python 装饰器练习

前面我们说了装饰器的概念以及如何使用，下面通过几个例题来巩固一下吧

1.用装饰器实现一个函数记时器

要点1：被装饰的函数有返回值

要点2：如何保留被装饰函数的函数名和帮助信息文档

i

mport time
import random
import string
import functools

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

def Timer(fun):
    """这是一个装饰器Timer"""
    # @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        """这是一个wrapper函数"""
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('运行时间为: %.5f' %(end_time - start_time))
        return res
    return wrapper

@Timer
def con_add():
    s = ''
    for i in li:
        s += (i + ',')
    print(s)

@Timer
def join_add():
    print(','.join(li))

@Timer
def fun_list(n):
    """这是fun_list函数"""
    return [i * 2 for i in range(n)]

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

#con_add()
#join_add()
print(fun_list(10))
print(fun_map(10))
#print(fun_list.__doc__)
#print(fun_list.__name__)

print(time.ctime())

运行结果：

['q', 'c', 'P', 'f', 'N', 'Y', 'N', 'P', 'v', 'f', 'U', 'l', 'M', 'l', 'V', 'u', 'P', 'S', 'l', 'f', 'O', 'X', 'K', 'v', 'Q', 'i', 'E', 'm', 'n', 'M', 'C', 'x', 'u', 'v', 'O', 'j', 'p', 'u', 'r', 'o', 'i', 'e', 'M', 'M', 'Z', 'F', 'R', 'a', 'z', 'B', 'S', 'F', 'G', 'Y', 'X', 'R', 'I', 'I', 't', 'f', 'C', 'e', 'B', 'B', 'G', 'n', 'r', 'y', 'M', 'z', 'j', 'k', 'W', 'H', 'I', 'Q', 'm', 'n', 'P', 'C', 'L', 'U', 'w', 'B', 'c', 's', 'l', 'U', 'u', 'i', 'u', 'N', 'p', 'l', 'Z', 'X', 'n', 'b', 'a', 'P']
运行时间为: 0.00000
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]
运行时间为: 0.00000
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]
Thu Apr 11 19:18:33 2019

2.多个装饰器练习

  编写装饰器required_types, 条件如下:
#     1). 当装饰器为@required_types(int,float)确保函数接收到的
每一个参数都是int或者float类型;
#     2). 当装饰器为@required_types(list)确保函数接收到的每一>个参数都是list类型;
#     3). 当装饰器为@required_types(str,int)确保函数接收到的每
一个参数都是str或者int类型;
#     4). 如果参数不满足条件， 打印 TypeError:参数必须为xxxx类
型
import functools


def required_types(*kinds):
    def required_int(fun):
        @functools.wraps(fun)
        def wrapper(*args, **kwargs):
            for i in args:
                if not isinstance(i, kinds):

                    raise TypeError('参数必须为%s,%s' % kinds)
            else:
                res = fun(*args, **kwargs)
                return res

        return wrapper

    return required_int


@required_types(float, float)
def add(a, b):
    return a + b


print(add(1.2, 2.1))

运行结果：

3.3
输入不是浮点型
  File "/home/kiosk/PycharmProjects/untitled/装饰器.py", line 26, in <module>
    print(add(1, 2.1))
  File "/home/kiosk/PycharmProjects/untitled/装饰器.py", line 11, in wrapper
    raise TypeError('参数必须为%s,%s' % kinds)
TypeError: 参数必须为<class 'float'>,<class 'float'>

3.多个装饰器练习

这里实现一个用户管理的功能，root用户和普通用户都能登陆，但只有root用户可以添加用户信息，为了实现这个功能要写两个装饰器，一个用来检测登陆，一个用来检测是否为root用户

import functools
import inspect

def is_admin(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        #inspect.getcallargs返回一个字典，key值是形参，value值
        #是对应的实参{'name':'root'}
        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('not root user!')
    return wrapper

login_session = ['root', 'redhat', 'gui']

def is_login(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        if args[0] in login_session:
            res = fun(*args,**kwargs)
            return res
        else:
            print('Error:%s未登录' %args[0])
    return wrapper



@is_login
@is_admin										#先检测是否为root用户在检测是否登陆
def add_student(name):
    print('添加学生信息...')

add_student('gui')

运行结果：

inspect的返回值: {'name': 'gui'}            #由于gui非root用户
not root user!