#####HTML
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>login</title>
</head>
<body>
<form action="/login" method="post">
<p><input type="text" name="user" placeholder="用户名"/></p>
<p><input type="text" name="pwd" placeholder="密码"/></p>
<p><input type="text" name="code" placeholder="验证码"/></p>
<p><img src="/check_code" onclick = 'Ceck(this)'>{{status}}</p>
<p><input type="submit" value="提交"/>
</form>
<script>
function Ceck(arg){
arg.src += '?'
}
</script>
</body>
</html>
####代码
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import tornado.ioloop
import tornado.web
container = {}
class Session:
def __init__(self,handler):
self.handler = handler
self.random_str = None
# 生成随机字符串
def __random_star(self):
import hashlib
import time
obj = hashlib.md5()
obj.update(bytes(str(time.time()),encoding='utf-8'))
random_str = obj.hexdigest()
return random_str
# 在用户浏览器上设置cookie
def __setitem__(self,key,value):
if not self.random_str:
# 获取用户上的cookie
random_str = self.handler.get_cookie('__kakaka__')
# 如果用户没有 cookie 那么就创建一份
if not random_str:
random_str = self.__random_star()
container[random_str] = {}
# 如果有
else:
# 如果 random_str 在 container.keys()里面 就pass
if random_str in container.keys():
pass
# 否则 就重新创建一份
else:
random_str = self.__random_star()
container[random_str] = {}
self.random_str = random_str
container[self.random_str][key] = value
self.handler.set_cookie('__kakaka__',self.random_str)
# 获取用户浏览器上的cookie
def __getitem__(self,key):
random_str = self.handler.get_cookie('__kakaka__')
#判断random_str是否为空
if not random_str:
return None
# 到container里面获取下random_str 看是否存在
user_info_dict = container.get(random_str,None)
if not user_info_dict:
return None
val = user_info_dict.get(key)
return val
class BaseHandler(tornado.web.RequestHandler):
def initialize(self):
self.session = Session(self)
class IndexHandler(BaseHandler):
def get(self, *args, **kwargs):
# 获取用户传入的值 如果等于将执行操作
if self.get_argument('u',None)in ['alex','eric']:
self.session['is_login'] = True
self.session['name']= self.get_argument('u',None)
else:
self.write('请登录')
class ManagerHandler(BaseHandler):
def get(self, *args, **kwargs):
# 获取用户随机字符串 判断是否是true
val = self.session['is_login']
if val:
self.write(self.session['name'])
else:
self.write('失败')
class LoginHandler(BaseHandler):
def get(self, *args, **kwargs):
self.render('login.html',status = "")
def post(self, *args, **kwargs):
user = self.get_argument('user',None)
pwd = self.get_argument('pwd',None)
code = self.get_argument('code',None)
ceck_code = self.session['CeckCode']
if ceck_code.upper() == code.upper():
self.write('验证码正确')
else:
self.render('login.html',status = '验证码错误')
class CodeHandler(BaseHandler):
def get(self, *args, **kwargs):
import io
import check_code
mstream = io.BytesIO()
# 创建图片 并写入验证码
img,code = check_code.create_validate_code()
# 为每个用户保存验证码
self.session['CeckCode'] = code
# 将图片写入IO mstream 中
img.save(mstream,"GIF")
# 发送图片到客户端
self.write(mstream.getvalue())
pass
settings = {
'template_path':'views',
'static_path':'statics',
'cookie_secret':'skjfsajdklfjalskj'
}
application = tornado.web.Application([
(r"/index",IndexHandler),
(r"/manager",ManagerHandler),
(r"/login",LoginHandler),
(r"/check_code",CodeHandler),
],**settings)
if __name__ == '__main__':
application.listen(8888)
tornado.ioloop.IOLoop.instance().start()