生成验证码图片

安装环境

• pip install captcha

代码

import random
import time
from pathlib import Path

from captcha.image import ImageCaptcha


def gen_images(save_dir: str, max_count=100, chars: list[str] = None, size=4):
    chars = chars or list("0123456789abcdefghijklmnopgrstuvwxyz")
    image = ImageCaptcha()

    save_dir = Path(save_dir)
    if not save_dir.exists():
        save_dir.mkdir(parents=True, exist_ok=True)

    for i in range(max_count):
        name = "".join(random.sample(chars, size))
        image_path = save_dir / "{}_{}.png".format(name.upper(), int(time.time() * 1000))
        image.write(name, str(image_path))
        print("第 {} 张生成成功 => {}".format(i + 1, image_path))


if __name__ == '__main__':
    gen_images(save_dir="dataset/code")

效果