web框架--tornado之验证码实例

tornado随机生成图片验证码

用python生成随机验证码需要借鉴一个插件，和一个io模块，实现起来也非常容易，当然也需要借鉴session来判断验证码是否错误，下面写一段用户登录验证带验证码的。

一、目录结构

二、login.html

 <!DOCTYPE html>

 <html lang="en">

 <head>

     <meta charset="UTF-8">

     <title>Title</title>

 </head>

 <body>

     <h1>欢迎登录</h1>

     <form method="POST", action="/login" enctype="multipart/form-data">

         <p><input name="username" type="text" placeholder="用户名"></p>

         <p><input name="pwd" type="password" placeholder="密码"></p>

         <p>

             <input name="code" type="text" placeholder="验证码">

             <img src="/check_code" onclick="ChangeCode();" id="imgCode">

         </p>

         <input type="submit" value="登录"/>

         <span style="color: red">{{state}}</span>

     </form>

 <script type="text/javascript">

     function ChangeCode() {   //当点击图片的时候，会刷新图片，这一段代码就可以实现

         var code = document.getElementById('imgCode');

         code.src += '?';

     }

 </script>

 </body>

 </html>

三、app.py

 #!/usr/bin/env python

 # -*- coding: utf- -*-

 import tornado.ioloop

 import tornado.web

 import io

 import check_code

 container = {}

 class Session:

     def __init__(self, handler):

         self.handler = handler

         self.random_str = None

     def __genarate_random_str(self):

         import hashlib

         import time

         obj = hashlib.md5()

         obj.update(bytes(str(time.time()), encoding='utf-8'))

         random_str = obj.hexdigest()

         return random_str

     def __setitem__(self, key, value):

         #在container中加入随机字符串

         #定义专属于自己的数据

         #在客户端中写入随机字符串

         #判断，请求的用户是否已有随机字符串

         if not self.random_str:

             random_str = self.handler.get_cookie('__session__')

             if not random_str:

                 random_str = self.__genarate_random_str()

                 container[random_str] = {}

             else:

                 #客户端有随机字符串

                 if random_str in container.keys():

                     pass

                 else:

                     random_str = self.__genarate_random_str()

                     container[random_str] = {}

             self.random_str = random_str

         container[self.random_str][key] = value

         self.handler.set_cookie("__session__", self.random_str)

     def __getitem__(self, key):

         #获取客户端的随机字符串

         #从container中获取专属于自己的数据

         #专属信息[key]

         random_str = self.handler.get_cookie("__session__")

         if not random_str:

             return None

         #客户端有随机字符串

         user_info_dict = container.get(random_str, None)

         if not user_info_dict:

             return None

         value = user_info_dict.get(key, None)

         return value

 class BaseHandler(tornado.web.RequestHandler):

     def initialize(self):

         self.session = Session(self)

 class LoginHandler(BaseHandler):

     def get(self, *args, **kwargs):

         self.render('login.html', state='')

     def post(self, *args, **kwargs):

         username = self.get_argument('username', None)

         pwd = self.get_argument('pwd', None)

         code = self.get_argument('code')

         check_code = self.session['CheckCode']

         if username == 'admin' and pwd == '' and code.upper() == check_code.upper():

             self.write('登录成功')

         else:

             self.render('login.html', state='用户信息或验证码不正确')

 class CheckCodeHandler(BaseHandler):

     def get(self, *args, **kwargs):

         mstream = io.BytesIO()

         img, code = check_code.create_validate_code()

         img.save(mstream, "GIF")

         self.session['CheckCode'] = code

         print(mstream.getvalue())

         self.write(mstream.getvalue())

 settings = {

     'template_path': 'views',

     'static_path': 'static',

     'cookie_secret': 'hafhdahqahefihdkasfka',

 }

 application = tornado.web.Application([

     (r"/login", LoginHandler),

     (r"/check_code", CheckCodeHandler),

 ], **settings)

 if __name__ == "__main__":

     application.listen()

     tornado.ioloop.IOLoop.instance().start()

四、check_code.py 和 Monaco.ttf 是一个插件，需要跟app.py放在同一个目录下。

五、运行结果图示

巴特西

web框架--tornado之验证码实例

最新文章

热门文章