【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模块中的其他函数更改一下效果。