生成图片验证码

genVerificationCode.py

from PIL import Image, ImageDraw, ImageFont, ImageFilter
import random

try:
    import cStringIO as StringIO
except ImportError:
    from io import StringIO
    from io import BytesIO


question = ['1+1', '7-4', '2+3', '6-5', '7+2']
answer = [2, 3, 5, 1, 9]

_letter_cases = "abcdefghijkmnpqrstuvwxy"
_upper_cases = "ABCDEFGHIJKLMNPQRSTUVWXY"
_numbers = "1234567890"
init_chars = ''.join((_letter_cases, _upper_cases, _numbers))
default_font = "arial.ttf"  # 字体地址需要根据不同场景进行切换


def generate_verify_image(size=(120, 50),
                          chars=init_chars,
                          img_type="GIF",
                          mode="RGB",
                          bg_color=(255, 255, 255),
                          fg_color=(0, 0, 255),
                          font_size=18,
                          font_type=default_font,
                          length=4,
                          draw_lines=True,
                          n_line=(1, 2),
                          draw_points=True,
                          point_chance=2,
                          save_img=False):
    width = 120
    height = 60
    img = Image.new(mode, size, bg_color)
    draw = ImageDraw.Draw(img)

    def get_chars():

        return random.sample(chars, length)

    def create_lines():

        line_num = random.randint(*n_line)

        for i in range(line_num):
            begin = (random.randint(0, size[0]), random.randint(0, size[1]))

            end = (random.randint(0, size[0]), random.randint(0, size[1]))
            draw.line([begin, end], fill=(0, 0, 0))

    def create_points():
        chance = min(100, max(0, int(point_chance)))

        for w in range(width):
            for h in range(height):
                tmp = random.randint(0, 100)
                if tmp > 100 - chance:
                    draw.point((w, h), fill=(0, 0, 0))

    def create_strs():
        c_chars = get_chars()
        strs = ' %s ' % ' '.join(c_chars)
        font = ImageFont.truetype(font_type, font_size)
        font_width, font_height = font.getsize(strs)

        draw.text(((width - font_width) / 3, (height - font_height) / 3),
                  strs, font=font, fill=fg_color)

        return ''.join(c_chars)

    if draw_lines:
        create_lines()
    if draw_points:
        create_points()
    strs = create_strs()

    params = [1 - float(random.randint(1, 2)) / 100,
              0,
              0,
              0,
              1 - float(random.randint(1, 10)) / 100,
              float(random.randint(1, 2)) / 500,
              0.001,
              float(random.randint(1, 2)) / 500
              ]
    img = img.transform(size, Image.PERSPECTIVE, params)

    img = img.filter(ImageFilter.EDGE_ENHANCE_MORE)

    mstream = StringIO()
    # img.save(mstream, img_type)

    if save_img:
        img.save("validate.gif", img_type)

    return mstream, strs


if __name__ == '__main__':
    res = generate_verify_image(save_img=True)[1]
    print(res)