【Python】生成图片验证码

Description

• Yixiaohan/show-me-the-code 第0010题

• 0010 ：使用 Python 生成类似于下图中的字母验证码图片

  

Notes

    这个小项目中涉及随机字符串的形成，以及PIL模块的使用。本项目中涉及到的PIL模块的一些基础操作如下：
   
1. 用Image.new()建立一个新的图片，可以自行设置颜色编码方式、图片大小以及填充颜色：
    bgImg = Image.new('RGB', (185, 90), bg_color)
    以上语句将建立一个新的RGB方式编码的图片，大小是185×90，填充颜色bg_color是通过randomColor()方式随机生成的。
   
2. 用ImageDraw.Draw()传入一个Image类对象，以其为底创建“画布“，在其上作画：
    canvas = ImageDraw.Draw(bgImg)
   
3. 如果要在图片上添加文字的话，要通过ImageFont.truetype()来设置字体：
    font = ImageFont.truetype(r'C:\Windows\Fonts\Jokerman.ttf', 60)
   
4. 通过“画布”的text()来设置文字的位置、内容、颜色，以及第三点中设置的font：
    canvas.text((0, 0), ans, text_color, font)

    除此之外还有一点需要稍加注意：因为验证码图片的背景颜色和字体颜色都是随机产生的，如果不巧颜色太过相近，则难以辨认验证码的内容。所以设置了colorDifference()函数，以判断二者的接近程度，若过于接近，则重新生成字体颜色。

My Code

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

"""
* 0010
    使用 Python 生成类似于下图中的字母验证码图片
    2017/1/28
"""

import random
from PIL import Image, ImageDraw, ImageFont

def randomColor():
    r = random.randint(0, 256)
    g = random.randint(0, 256)
    b = random.randint(0, 256)
    return (r, g, b)

def colorDifference(bg_color, text_color):
    d = 0
    for i in range(0, 3):
        d += (text_color[i] - bg_color[i]) ^ 2
    return d

root = ""
a = ord('a')
A = ord('A')
for i in range(0, 26):
    root += chr(a + i)
    root += chr(A + i)

# bgImg = Image.open("background.png")
# bgImg.show()

font = ImageFont.truetype(r'C:\Windows\Fonts\Jokerman.ttf', 60) # 创建字体对象给ImageDraw中的text函数使用

for j in range(0, 10):
    ans = ""
    for i in range(0, 4):
        ans += random.choice(root)
    # print(ans)

    bg_color = randomColor()
    bgImg = Image.new('RGB', (185, 90), bg_color) # 新建一个图片对象, 背景颜色随机
    # bgImg.show()
    canvas = ImageDraw.Draw(bgImg)

    text_color = randomColor()
    while(colorDifference(bg_color, text_color) < 100): # 让字体颜色和背景颜色反差大一些，以防看不清
        text_color = randomColor()

    canvas.text((0, 0),ans, text_color, font)

    name = ans + '.jpg'
    bgImg.save(name, 'jpeg')
    rstImg = Image.open(name)
    rstImg.show()

    inp = input('Please type in the characters in the image:')
    while inp != ans:
        inp = input('Incorrect input. Please try again:')

Result

    运行程序，可以看到文件夹里出现了花花绿绿的图片验证码。虽然随机的配色很是有些辣眼睛，不过好歹完成任务了~题中给的实例图片似乎加了特效，或许可以尝试用Image模块中的其他函数更改一下效果。