[oldboy-django][2深入django]登录注册页面验证码
后台生成随机验证码
#验证码生成 - 在login.html 中有一张验证码图片 设置图片的src url, 相当于发送一次get请求,返回的内容为图片渲染的内容 <div class="col-sm-5"> <input type="password" class="form-control" id="inputPassword3" placeholder="验证码"> </div> <div class="col-sm-5"> <img src="/check_code/" width="150px" height="34px" title="点击刷新验证码"> </div> - 随机生成图片内容 def check_code(request): from PIL import Image,ImageDraw from io import BytesIO f = BytesIO() # 从内存开辟一段空间 img = Image.new(mode="RGB", size=(150,34), color=(255, 255, 255)) draw = ImageDraw.Draw(img, mode='RGB') # 设置图片文本的字体 from PIL import ImageFont font = ImageFont.truetype("kumo.ttf", 25) #7 往图片添加内容设置每个字符不同的颜色 import random code_list = [] for i in range(5): char = chr(random.randint(65,90)) code_list.append(char) draw.text((i*20,10), char, fill=(random.randint(0,255),random.randint(0,255),random.randint(0,255)), font = font ) # 添加内容结束 img.save(f,'png') # 图片内容保存到内存中 data = f.getvalue() # 读取图片内容 return HttpResponse(data)
将验证码保存在session, 用户提交验证码和session的验证码进行比较,完成验证码的验证
- 将验证码保存在session,用来记住用户。以便后面的验证码验证 code = "".join(code_list) request.session["code"] = code - login验证码验证 def login(request): if request.method == 'GET': return render(request, 'login.html') else: input_code = request.POST.get("code") session_code = request.session.get('code') if input_code.upper() == session_code.upper(): return HttpResponse("code is correct") else: return HttpResponse("code is invalid")
将验证码封装
- 验证码封装 import random from PIL import (Image,ImageDraw,ImageFont,ImageFilter) def check_code(width=120, height=30, char_length=5, font_file='kumo.ttf', font_size=28): code = [] img = Image.new(mode='RGB', size=(width, height), color=(255, 255, 255)) draw = ImageDraw.Draw(img, mode='RGB') def rndChar(): """生成随机字母 :return: """ return chr(random.randint(65, 90)) def rndColor(): """生成随机颜色:return: """ return (random.randint(0, 255), random.randint(10, 255), random.randint(64, 255)) # 写文字 font = ImageFont.truetype(font_file, font_size) for i in range(char_length): char = rndChar() code.append(char) h = random.randint(0, 4) draw.text([i * width / char_length, h], char, font=font, fill=rndColor()) # 写干扰点 for i in range(40): draw.point([random.randint(0, width), random.randint(0, height)], fill=rndColor()) # 写干扰圆圈 for i in range(40): draw.point([random.randint(0, width), random.randint(0, height)], fill=rndColor()) x = random.randint(0, width) y = random.randint(0, height) draw.arc((x, y, x + 4, y + 4), 0, 90, fill=rndColor()) # 画干扰线 for i in range(5): x1 = random.randint(0, width) y1 = random.randint(0, height) x2 = random.randint(0, width) y2 = random.randint(0, height) draw.line((x1, y1, x2, y2), fill=rndColor()) img = img.filter(ImageFilter.EDGE_ENHANCE_MORE) return img, ''.join(code)
获取验证码后台视图
# 封装后的使用 def checkout(request): from io import BytesIO from utils.checkCode import check_code img, code = check_code() stream = BytesIO() img.save(stream, 'png') request.session['code'] = code return HttpResponse(stream.getvalue())
# 验证码生成, 如何使用PIL
https://www.liaoxuefeng.com/wiki/001374738125095c955c1e6d8bb493182103fac9270762a000/00140767171357714f87a053a824ffd811d98a83b58ec13000