最近公司網(wǎng)站，需要在注冊(cè)模塊添加驗(yàn)證碼，防止其他人頻繁的惡意注冊(cè)，我們后端使用的是python進(jìn)行開(kāi)發(fā)，所以研究了下python圖片驗(yàn)證碼的方法。

最后確定使用python里面PIL庫(kù)，通過(guò)Image, ImageDraw, ImageFont, ImageFilter的模塊生成圖片驗(yàn)證碼

設(shè)計(jì)思路(這里就不畫(huà)圖了)：

(1)用戶填寫(xiě)用戶名（必須先填）
(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"                        # 小寫(xiě)字母
_upper_cases = "ABCDEFGHJKLMNPQRSTUVWXY"                        # 大寫(xiě)字母
_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為T(mén)rue時(shí)有效
    :param draw_points: 是否畫(huà)干擾點(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)建畫(huà)筆

    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è)字符前后以空格隔開(kāi)

        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)證碼的方法