最近公司網站，需要在注冊模塊添加驗證碼，防止其他人頻繁的惡意注冊，我們后端使用的是python進行開發，所以研究了下python圖片驗證碼的方法。
最后確定使用python里面PIL庫，通過Image, ImageDraw, ImageFont, ImageFilter的模塊生成圖片驗證碼
設計思路(這里就不畫圖了)：
(1)用戶填寫用戶名（必須先填）
(2)客戶端點擊獲取驗證碼，請求里帶用戶名參數
(3)調用生成驗證碼接口
(4)按一定規則將用戶名生成鍵值，以生成的鍵值為鍵，驗證碼為值，保存到緩存中
(5)將驗證碼的字節流返回客戶端并顯示
(6)輸入驗證碼，點擊注冊
(7)后臺根據用戶名和驗證碼驗證
生成驗證碼函數
#!/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" # 數字
init_chars = ''.join((_letter_cases, _upper_cases, _numbers)) # 生成允許的字符集合
default_font = "./DejaVuSans.ttf" # 驗證碼字體
# 生成驗證碼接口
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):
"""
生成驗證碼圖片
:param size: 圖片的大小，格式（寬，高），默認為(120, 30)
:param chars: 允許的字符集合，格式字符串
:param img_type: 圖片保存的格式，默認為GIF，可選的為GIF，JPEG，TIFF，PNG
:param mode: 圖片模式，默認為RGB
:param bg_color: 背景顏色，默認為白色
:param fg_color: 前景色，驗證碼字符顏色，默認為藍色#0000FF
:param font_size: 驗證碼字體大小
:param font_type: 驗證碼字體，默認為 DejaVuSans.ttf
:param length: 驗證碼字符個數
:param draw_lines: 是否劃干擾線
:param n_line: 干擾線的條數范圍，格式元組，默認為(1, 2)，只有draw_lines為True時有效
:param draw_points: 是否畫干擾點
:param point_chance: 干擾點出現的概率，大小范圍[0, 100]
:param save_img: 是否保存為圖片
:return: [0]: 驗證碼字節流, [1]: 驗證碼圖片中的字符串
"""
width, height = size # 寬， 高
img = Image.new(mode, size, bg_color) # 創建圖形
draw = ImageDraw.Draw(img) # 創建畫筆
def get_chars():
"""生成給定長度的字符串，返回列表格式"""
return random.sample(chars, length)
def create_lines():
"""繪制干擾線"""
line_num = random.randint(*n_line) # 干擾線條數
for i in range(line_num):
# 起始點
begin = (random.randint(0, size[0]), random.randint(0, size[1]))
# 結束點
end = (random.randint(0, size[0]), random.randint(0, size[1]))
draw.line([begin, end], fill=(0, 0, 0))
def create_points():
"""繪制干擾點"""
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():
"""繪制驗證碼字符"""
c_chars = get_chars()
strs = ' %s ' % ' '.join(c_chars) # 每個字符前后以空格隔開
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()
# 圖形扭曲參數
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) # 創建扭曲
img = img.filter(ImageFilter.EDGE_ENHANCE_MORE) # 濾鏡，邊界加強（閾值更大）
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
注意：返回的流要進行轉換，在返回前端
self.write(simplejson.dumps({'code': 0, 'img': stream.getvalue().encode('base64')}))
#這里是將stream的值進行了一次base64的編碼
前端js設置圖片src代碼
$("#verify_code_img").attr("src", "data:image/gif;base64," + data.img);
效果
是不是很簡單，下次分享前端生成圖片驗證碼的方法
