1、安装pillow 并编写validatecode.py
from PIL import Image, ImageFont, ImageDraw, ImageFilter import random def validate_picture(): total = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ012345789' # 图片大小130x50 width = 130 height = 50 # 先生成一个新图片对象 im = Image.new('RGB', (width, height), 'white') # 设置字体 font = ImageFont.truetype('C:\\Windows\\Fonts\\Calibri.ttf', 50) # font = ImageFont.load_default().font # 创建draw对象 draw = ImageDraw.Draw(im) str1 = '' # 输入每一个文字 for item in range(5): text = random.choice(total) str1 += text draw.text((5+random.randint(4, 7)+20*item, 5+random.randint(3, 7)), text=text, fill='blue', font=font) # 划几根干扰线 for num in range(8): x1 = random.randint(0, width/2) y1 = random.randint(0, height/2) x2 = random.randint(0, width) y2 = random.randint(height/2, height) draw.line(((x1, y1), (x2, y2)), fill='black', width=1) # 模糊下,加个滤镜 im = im.filter(ImageFilter.FIND_EDGES) return im, str1 if __name__ == '__main__': im, str1 = validate_picture() print(str1)
2、编写view 函数
@admin.route('/login/code/') def get_code(): image, str1 = validate_picture() # 讲验证码图片以二进制形式写入内存,防止图片都放在文件夹中,占用磁盘空间 buf = BytesIO() image.save(buf, 'jpeg') buf_str = buf.getvalue() # 把二进制作为response发回前端,并设置头部字段 response = make_response(buf_str) response.headers['Content-Type'] = 'image/gif' # 验证码字符串存储在seesion中 session['image'] = str1 return response
3、html模板部分
<div class="page"> <div class="loginwarrp"> <div class="logo">管理员登陆</div> <div class="login_form"> {% for msg in get_flashed_messages() %} <p style="color: #FF0000">{{ msg }}</p> {% endfor %} <form id="Login" name="Login" method="post"> {{ form.csrf_token }} <li class="login-item"> <span>{{ form.name.label }}:</span> {{ form.name }} <span id="count-msg" class="error"></span> </li> {% for err in form.name.errors %} <dev class="col-md-12"> <font style="color: #FF0000">{{ err }}</font> </dev> {% endfor %} <li class="login-item"> <span>{{ form.pwd.label }}:</span> {{ form.pwd }} <span id="password-msg" class="error"></span> {% for err in form.pwd.errors %} <dev class="col-md-12"> <font style="color: #FF0000">{{ err }}</font> </dev> {% endfor %} </li> <span>{{ form.verify_code.label }}:</span> {{ form.verify_code }} </li> <img src="{{ url_for('admin.get_code') }}" onclick="this.src='/admin/login/code?'+Math.random()" border="0" class="verifyimg"/> <div class="clearfix"></div> <li class="login-sub"> <input type="submit" name="Submit" value="登录"/> <input type="reset" name="Reset" value="重置"/> </li> </form> </div> </div> </div>