图片验证码

• check_code.py

from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
from PIL import ImageFilter
import random
import math
import string
import os

BASE_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

class CreateCheckCode:
    #生成几位数的验证码
    number = 4
    #生成验证码图片的高度和宽度
    size = (100,30)
    #背景颜色，默认为白色
    bg_color = (255,255,255)
    #字体颜色，默认为蓝色
    font_color = (0,0,255)
    #干扰线颜色。默认为红色
    line_color = (255,0,0)
    #是否要加入干扰线
    draw_line = True
    #加入干扰线条数的上下限
    line_number = (1,5)
    #字体路径和保存路径
    font_path = os.path.join(BASE_PATH, 'static/font/Monaco.ttf')
    save_path = os.path.join(BASE_PATH, 'static/img/check_code')

    def create_str(self):
        '''生成字符串:字母+数字'''
        source = list(string.ascii_letters)
        for index in range(0, 10):
            source.append(str(index))
        return ''.join(random.sample(source, self.number))

    def create_lines(self,draw,width,height):
        '''绘制干扰线'''
        begin = (random.randint(0, width), random.randint(0, height))
        end = (random.randint(0, width), random.randint(0, height))
        draw.line([begin, end], fill=self.line_color)

    def check_code(self, filename):
        width, height = self.size
        image = Image.new('RGBA', self.size, self.bg_color)  # 创建空白图片
        font = ImageFont.truetype(self.font_path, 25)  # 验证码的字体和字体大小
        draw = ImageDraw.Draw(image)  # 创建画笔
        text = self.create_str()  # 生成字符串
        # print('字符串：',text)
        font_width, font_height = font.getsize(text)
        # 填充字符串
        draw.text(((width - font_width)/self.number,(height-font_height)/self.number),text,font=font,fill=self.font_color)
        if self.draw_line:
            self.create_lines(draw, width, height)
            self.create_lines(draw, width, height)
            self.create_lines(draw, width, height)
        # 创建扭曲
        image = image.transform((width + 20, height + 10), Image.AFFINE, (1, -0.3, 0, -0.1, 1, 0), Image.BILINEAR)
        # 滤镜，边界加强
        image = image.filter(ImageFilter.EDGE_ENHANCE_MORE)
        # 保存验证码图片.png
        image.save('%s/%s.png' % (self.save_path, filename))
        # print("图片保存路径:", self.save_path)
        return text

• views.py

from utils.check_code import CreateCheckCode
import uuid


def check_code(request):
    '''生成验证码图片'''
    check_obj = CreateCheckCode()
    nid = str(uuid.uuid4())
    text = check_obj.check_code(nid)
    file_path = os.path.join('static/img/check_code','%s.png'%nid)
    f = open(file_path,'rb')
    data = f.read()
    f.close()
    request.session['check_code'] = text
    # 返回一个验证码图片到前端显示出来
    return HttpResponse(data)


def login(request):
    '''登陆'''
    if request.method == 'GET':
        return render(request, 'login.html')
    if request.method == 'POST':
        result = {'status':False,'msg':None}
        name = request.POST.get('username')
        pwd = request.POST.get('password')
        ck_code = request.POST.get('check_code') # 用户输入的验证码
        ss_code = request.session['check_code']  # 页面生成的验证码
        user_obj = UserInfo.objects.filter(username=name).first()
        blog_obj = Blog.objects.filter(user_id=user_obj.id).first()
        # print('****************',theme_id.theme)
        if user_obj.username == name and user_obj.pwd == md5(pwd):
            if ck_code.lower() == ss_code.lower():
                request.session['user_id'] = user_obj.id
                request.session['username'] = name
                request.session['is_login']=True
                request.session['theme_id']=blog_obj.theme
                request.session['suffix']=blog_obj.suffix
                request.session['title']=blog_obj.title
                if request.POST.get("one-month"):
                    request.session.set_expiry(60*60*24*30)
                print('登陆成功')
                return redirect('http://127.0.0.1:8000')
            else:
                result['status'] = True
                result['msg'] = '验证码错误或过期'
                return render(request, 'login.html',{'result':result})
        else:
            result['msg'] = '用户名或密码错误'
            return render(request, 'login.html', {'result':result})

• login.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    {% load staticfiles %}
    <link rel="stylesheet" href="{% static 'plugins/bootstrap/css/bootstrap.css' %}"/>
    <link rel="stylesheet" href="{% static 'css/account_styles.css' %}"/>
    <style>
        .login-button{
            background-color: #4b90ef;
            margin-top:20px;
        }
    </style>
</head>
<body class="bg-color">
<div class="register-band" style="height: 470px;">
    <div class="content-band">
        <form action="/login.html" method="POST" novalidate>
            {% csrf_token %}
            <div class="login-title">欢迎登陆</div>
            <div class="login-text">用户名</div>
            <div>
                <input type="text" class="form-control" name="username" placeholder="用户名">
            </div>
            <div class="login-text">密码</div>
            <div>
                <input type="password" class="form-control" name="password" placeholder="密码">
            </div>
            <div class="check-code-band">
                <input type="text" class="form-control" name="check_code" placeholder="验证码">
          <!--访问生成验证码的URL，触发check_code函数-->      
　　　　　　　　　 <img src="/check_code.html" onclick="changeImg(this);"/>
            </div>
            <div style="margin-top: 15px;"><input type="checkbox" name="one-month"/>一个月内免登陆</div>
            <div style="margin-top: 15px;color: red;">{{ result.msg }}</div>
            <div style="text-align: center;"><input type="submit" value="登陆" class="register-button login-button"/></div>
        </form>
    </div>
</div>
<script>
    function changeImg(ths) {
        // src里面加上？就会再发一次请求，验证码就会刷新一次
        ths.src = ths.src + '?';
    }
</script>
</body>
</html>