Django登录验证

def login(func):
def wrapper(request, *args, **kw):
if 'uname' not in request.COOKIES:
response = redirect('login')
response.set_cookie('url', request.get_full_path()) #记录用户浏览的URL
return response
else:
return func(request, *args, **kw)
return wrapper

def login(request):
if request.method == 'GET':
return render(request, 'login.html', {})
if request.method == 'POST':
username = request.POST['username']
password = request.POST['pwd']
count = UserInfo.objects.filter(username=username).count()
if count == 0:
return redirect(to='login')
else:
user = UserInfo.objects.get(username=username)
if user.password == password:
url = request.COOKIES.get('url', '/index')#如果'url'存在返回url，否则跳转index
response = redirect(to=url)
response.set_cookie('uname', username, max_age=3600)
return response
else:
return redirect(to='login')