'''生成验证码图形'''

from flask import Flask , render_template , session  ,make_response

import random

import string

from PIL import Image, ImageFont, ImageDraw

from io import BytesIO

 

app = Flask(__name__)     # 实例化Flask类

app.secret_key = "mrsoft" # 设置secret_key

 

def rndColor():

    '''随机颜色'''

    return (random.randint(32, 127), random.randint(32, 127), random.randint(32, 127))

 

def gene_text():

    '''生成4位验证码'''

    return ''.join(random.sample(string.ascii_letters+string.digits, 4))

 

def draw_lines(draw, num, width, height):

    '''干扰线'''

    for num in range(num):

        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)

 

def get_verify_code():

    '''生成验证码图形'''

    code = gene_text()

    # 图片大小120×50

    width, height = 120, 50

    # 新图片对象

    im = Image.new('RGB',(width, height),'white')

    # 字体

    # font = ImageFont.truetype('app/static/fonts/arial.ttf', 40)

    font = ImageFont.truetype(r'C:\Windows\Fonts\SIMLI.TTF', 50)  # 设置验证码字体为隶书

    # draw对象

    draw = ImageDraw.Draw(im)

    draw_lines(draw,num=5,width=200,height=50)

    # 绘制字符串

    for item in range(4):

        draw.text((5+random.randint(-3,3)+23*item, 5+random.randint(-3,3)),

                  text=code[item], fill=rndColor(),font=font )

    return im, code

 

 

@app.route('/login')

def login():

    '''登录'''

    return render_template("login.html") # 渲染页面

 

@app.route('/code')

def get_code():

    image, code = get_verify_code()

    # 图片以二进制形式写入

    buf = BytesIO()

    image.save(buf, 'jpeg')

    buf_str = buf.getvalue()

    # 把buf_str作为response返回前端，并设置首部字段

    response = make_response(buf_str)

    response.headers['Content-Type'] = 'image/gif'

    # 将验证码字符串储存在session中

    session['image'] = code

    return response