#!/usr/bin/env python3

# _*_ coding:utf-8 _*_

import string

import random

from captcha.image import ImageCaptcha  # 生成图像验证码

from io import BytesIO

from flask  import Flask, make_response

str = string.ascii_letters +"3456789"

def get_captcha(str_length=4, width=160, height=50):

    # 生成验证码str

    text = "".join([random.choice(str) for  _  in range(str_length)])

    # 生成验证码图片

    image = ImageCaptcha(width, height)

    color = tuple([random.randint(0, 255)for _in range(3)])

    background = tuple([random.randint(0, 255)for _in range(3)])

    img = image.create_captcha_image(text, color=color, background=background)

    # img.show()

    # 将验证码保存到内存中

    file = BytesIO()

    img.save(file, format="JPEG")

    return text, file.getvalue()

##################################################

app = Flask(__name__)

@app.route("/image_codes/<int:code_uuid>")

def get_image_code(code_uuid):

    text, data = get_captcha(str_length=6)

    print(text)

    print(code_uuid)

    print(data)

    resp = make_response(data)

    resp.headers["Content-Type"]  = "image/jpg"

    return resp

if __name__ =='__main__':

    app.run(debug=True)