python验证码生成

使用captcha将验证码转化为图片

from captcha.image import ImageCaptcha
import numpy as np
from PIL import Image
import random


number = ["0","1","2","3","4","5","6","7","8","9"]


def random_captcha_text(char_set=number, captcha_size=4):
    # 验证码列表
    captcha_text = []
    for i in range(captcha_size):
        # 随机选择
        c = random.choice(char_set)
        # 加入验证码列表
        captcha_text.append(c)
    return captcha_text


# 生成字符对应的验证码
def gen_captcha_text_and_images():
    image = ImageCaptcha()
    # 获得随机生成的验证码
    captcha_text = random_captcha_text()
    # 把列表转换为字符串
    captcha_text = ''.join(captcha_text)
    # 生成验证码
    captcha = image.generate(captcha_text)
    # 写入文件
    image.write(captcha_text, 'captcha/images/' + captcha_text + ".jpg")

num = 10000
if __name__ == '__main__':
    for i in range(num):
        gen_captcha_text_and_images()
        if i%1000 == 0:
            print("(%d/%d)" %(i,num))
    print("生成完毕")