最近公司網(wǎng)站合住，需要在注冊(cè)模塊添加驗(yàn)證碼，防止其他人頻繁的惡意注冊(cè)慕购，我們后端使用的是python進(jìn)行開發(fā)聊疲，所以研究了下python圖片驗(yàn)證碼的方法茬底。
最后確定使用python里面PIL庫(kù)沪悲，通過Image, ImageDraw, ImageFont, ImageFilter的模塊生成圖片驗(yàn)證碼
設(shè)計(jì)思路(這里就不畫圖了)：
(1)用戶填寫用戶名（必須先填）
(2)客戶端點(diǎn)擊獲取驗(yàn)證碼，請(qǐng)求里帶用戶名參數(shù)
(3)調(diào)用生成驗(yàn)證碼接口
(4)按一定規(guī)則將用戶名生成鍵值阱表，以生成的鍵值為鍵殿如，驗(yàn)證碼為值贡珊，保存到緩存中
(5)將驗(yàn)證碼的字節(jié)流返回客戶端并顯示
(6)輸入驗(yàn)證碼，點(diǎn)擊注冊(cè)
(7)后臺(tái)根據(jù)用戶名和驗(yàn)證碼驗(yàn)證
生成驗(yàn)證碼函數(shù)
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import random
from PIL import Image, ImageDraw, ImageFont, ImageFilter
try:
import cStringIO as StringIO
except ImportError:
import StringIO
_letter_cases = "abcdefghjkmnpqrstuvwxy" # 小寫字母
_upper_cases = "ABCDEFGHJKLMNPQRSTUVWXY" # 大寫字母
_numbers = "1234567890" # 數(shù)字
init_chars = ''.join((_letter_cases, _upper_cases, _numbers)) # 生成允許的字符集合
default_font = "./DejaVuSans.ttf" # 驗(yàn)證碼字體
# 生成驗(yàn)證碼接口
def generate_verify_image(size=(120, 30),
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):
"""
生成驗(yàn)證碼圖片
:param size: 圖片的大小涉馁，格式（寬门岔，高），默認(rèn)為(120, 30)
:param chars: 允許的字符集合烤送，格式字符串
:param img_type: 圖片保存的格式寒随，默認(rèn)為GIF，可選的為GIF帮坚，JPEG妻往，TIFF，PNG
:param mode: 圖片模式试和，默認(rèn)為RGB
:param bg_color: 背景顏色讯泣，默認(rèn)為白色
:param fg_color: 前景色，驗(yàn)證碼字符顏色阅悍，默認(rèn)為藍(lán)色#0000FF
:param font_size: 驗(yàn)證碼字體大小
:param font_type: 驗(yàn)證碼字體好渠，默認(rèn)為 DejaVuSans.ttf
:param length: 驗(yàn)證碼字符個(gè)數(shù)
:param draw_lines: 是否劃干擾線
:param n_line: 干擾線的條數(shù)范圍，格式元組节视，默認(rèn)為(1, 2)拳锚，只有draw_lines為True時(shí)有效
:param draw_points: 是否畫干擾點(diǎn)
:param point_chance: 干擾點(diǎn)出現(xiàn)的概率，大小范圍[0, 100]
:param save_img: 是否保存為圖片
:return: [0]: 驗(yàn)證碼字節(jié)流, [1]: 驗(yàn)證碼圖片中的字符串
"""
width, height = size # 寬寻行， 高
img = Image.new(mode, size, bg_color) # 創(chuàng)建圖形
draw = ImageDraw.Draw(img) # 創(chuàng)建畫筆
def get_chars():
"""生成給定長(zhǎng)度的字符串晌畅，返回列表格式"""
return random.sample(chars, length)
def create_lines():
"""繪制干擾線"""
line_num = random.randint(*n_line) # 干擾線條數(shù)
for i in range(line_num):
# 起始點(diǎn)
begin = (random.randint(0, size[0]), random.randint(0, size[1]))
# 結(jié)束點(diǎn)
end = (random.randint(0, size[0]), random.randint(0, size[1]))
draw.line([begin, end], fill=(0, 0, 0))
def create_points():
"""繪制干擾點(diǎn)"""
chance = min(100, max(0, int(point_chance))) # 大小限制在[0, 100]
for w in xrange(width):
for h in xrange(height):
tmp = random.randint(0, 100)
if tmp > 100 - chance:
draw.point((w, h), fill=(0, 0, 0))
def create_strs():
"""繪制驗(yàn)證碼字符"""
c_chars = get_chars()
strs = ' %s ' % ' '.join(c_chars) # 每個(gè)字符前后以空格隔開
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()
# 圖形扭曲參數(shù)
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) # 創(chuàng)建扭曲
img = img.filter(ImageFilter.EDGE_ENHANCE_MORE) # 濾鏡，邊界加強(qiáng)（閾值更大）
mstream = StringIO.StringIO()
img.save(mstream, img_type)
if save_img:
img.save("validate.gif", img_type)
return mstream, strs
if __name__ == "__main__":
mstream, strs = generate_verify_image(save_img=True)
print strs
注意：返回的流要進(jìn)行轉(zhuǎn)換寡痰，在返回前端
self.write(simplejson.dumps({'code': 0, 'img': stream.getvalue().encode('base64')}))
#這里是將stream的值進(jìn)行了一次base64的編碼
前端js設(shè)置圖片src代碼
$("#verify_code_img").attr("src", "data:image/gif;base64," + data.img);
效果
是不是很簡(jiǎn)單抗楔，下次分享前端生成圖片驗(yàn)證碼的方法
