python flask 用pillow實現登錄驗證碼驗證

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>