Python装饰器、Django自定义登录验证

装饰器是一个很著名的设计模式,经常被用于有切面需求的场景,较为经典的有插入日志、性能测试、事务处理等。
装饰器其实就是一个工厂函数,它接受一个函数为参数,然后返回一个新函数,其闭包中包含了原函数

1、简单装饰器:

def deco(func):
    def wrapper():
        print "start"
        func() #调用函数
        print "end"
    return wrapper

@deco
def myfun():
    print "run"

myfun()

由于装饰器函数返回的是原函数的闭包wrapper,实际上被装饰后的函数就是wrapper,其运行方式就和wrapper一样。相当于:
myfun=deco(myfun)

2、装饰任意参数的函数:

def deco(func):
    def warpper(*args,**kw):
        print "start"
        func(*args,**kw)
        print "end"
    return warpper
	
@deco
def myfun1(param1):
    print "run with param %s"%(param1)
	
@deco
def myfun2(param1,param2):
    print "run with param %s and %s"%(param1,param2)
	
myfun1("something")
myfun2("something","otherthing")

# start
# run with param something
# end
# start
# run with param something and otherthing
# end

 3、带参数的装饰器

装饰器接受一个函数作为参数,这个毋庸置疑。但是有时候我们需要装饰器接受另外的参数。此时需要再加一层函数,实际上是定义了一个生成装饰器的工厂函数,调用它,搭配需要的参数,来返回合适的装饰器

def log(text):
    def deco(func):
        def wrapper(*args,**kw):
            print text
            func(*args,**kw)
            print text + " again"
        return wrapper
    return deco
	
@log("hello")
def myfun(message):
    print message
	
myfun("world")

# hello
# world
# hello again

log=log("hello"),把返回的deco函数赋值给log,此时log相当于其包含text=“hello”的闭包
myfun=log(myfun),相当于把myfun传入了deco函数,并且返回wrapper,并赋值给myfun,此时myfun相当于其装饰后的闭包。
整体来看是myfun=log("hello")(myfun)

4、装饰器带类参数

class locker:  
    def __init__(self):  
        print("locker.__init__() should be not called.")
	
	@staticmethod  
    def acquire():  
        print("locker.acquire() called.(这是静态方法)")
		
	@staticmethod  
    def release():  
        print("  locker.release() called.(不需要对象实例)")
		
def deco(cls):  
    '''''cls 必须实现acquire和release静态方法'''
	def _deco(func):  
        def __deco(): 
			print("before %s called [%s]." % (func.__name__, cls))  
            cls.acquire()  
            try:  
                return func()  
            finally:  
                cls.release()
		return __deco  
    return _deco
	
@deco(locker)  
def myfunc():  
    print(" myfunc() called.")
	
myfunc()  
myfunc()

 5、django自定义装饰器实现登录验证

def Check_Login(func):  #自定义登录验证装饰器
    def warpper(request,*args,**kwargs):
        is_login = request.session.get('IS_LOGIN', False)
        if is_login:
            func(request,*args,**kwargs)
        else:
            return HttpResponseRedirect("/polls/login_user")
    return warpper

def login_user(request):
    if request.method == 'POST':
        form = LoginForm(request.POST)
        if form.is_valid():
            all_data = form.clean()   #获取post数据,例如 {'username': u'yang1', 'password': 111}
            exist = User.objects.filter(username = all_data['Form_username'],password = all_data['Form_password']).first()
            if exist:
                request.session['IS_LOGIN'] = True  #设置session的随机字段值
                request.session['uname'] = exist.username   #设置uname字段为登录用户
                return HttpResponseRedirect('/polls/home')
            else:
                return HttpResponse("账户或密码错误")
    else:
        form = LoginForm()
    return render(request, 'polls/login_user.html', {'form': form})

@Check_Login
def home(request):
        username = request.session.get('uname', False)   #获取登录用户名
        return render(request, 'polls/home.html', {'username': username}) #用户名渲染到前端页面

 

posted @ 2018-05-02 00:01  逆流而上的鱼丶  阅读(2458)  评论(0编辑  收藏  举报