PYTHON---装饰器

装饰器

1.需求：用户登陆验证的装饰器 is_login
        1.如果用户登陆成功，则执行被装饰的函数
        1.如果用户登陆不成功，则执行登陆函数

"""
import functools
login_users = ['admin', 'root']
def is_login(fun):
    @functools.wraps(fun)
    def wrapper(*args, **kwargs):
        # 判断写博客这个用户是否登陆成功
        if kwargs.get('name') in login_users:
            res = fun(*args, **kwargs)
            return res
        else:
            res = login()
            return res
    return wrapper
def login():
    return '登陆。。。'
@is_login
def writeBlog(name):
    return '编写博客。。'
print writeBlog(name='adminss')

 

2.探究多个装饰器的执行顺序

"""
def decorator_a(func):
    print 'Get in decorator_a'

    def inner_a(*args, **kwargs):
        print 'Get in inner_a'
        res = func(*args, **kwargs)
        return res
    return inner_a
def decorator_b(func):
    print 'Get in decorator_b'

    def inner_b(*args, **kwargs):
        print 'Get in inner_b'
        res = func(*args, **kwargs)
        return res
    return inner_b
# 当有多个装饰器时，从下到上调用装饰器
@decorator_b
@decorator_a
def f(x):
    print 'Get in f'
    return x * 2
f(1)

3.带有参数的装饰器

import time
import functools

def log(kind):
    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] 函数名:%s,运行时间:%.5f,运行返回值结果:%d' %(kind,time.ctime(),fun.__name__,end_time-start_time,res)
            return res
        return wrapper
    return add_log

@log('westos')
# log('dubug) --> 返回值是add_log
# add = add_log(add)
def add(x,y):
    time.sleep(1)
    return x+y
# wrapper(1,2)
print add(1,2)

4. 在我们实际的应用场景中，会采用多个装饰器先验证是否登陆成功再验证权限

"""
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
        if inspect_res.get('name') == 'root':
            res = fun(*args, **kwargs)
            return res
        else:
            print 'Error：no permisson add studnet'

    return wrapper
login_session = ['root', 'admin', 'redhat']
def is_login(fun):
    @functools.wraps(fun)
    def warrper(*args, **kwargs):
        if args[0] in login_session:
            res = fun(*args, **kwargs)
            return res
        else:
            print 'Error:%s未登陆!' % args[0]
    return warrper
@is_login
@is_root
def add_student(name):
    print '添加学生信息。。'
add_student('root')