python 生成圖形驗證碼

Faremax 發(fā)布于2019-07-31 11:18 / 3492人閱讀

摘要：文章鏈接日常在網(wǎng)站使用過程中經(jīng)常遇到圖形驗證，今天準(zhǔn)備自己做個圖形驗證碼，這算是個簡單的功能，也適合新手練習(xí)的，便于自己學(xué)習(xí)。生成隨機(jī)顏色，返回的是三色。隨機(jī)畫線，在圖片寬高范圍內(nèi)隨機(jī)生成個坐標(biāo)點，并通過隨機(jī)顏色產(chǎn)生線條。

文章鏈接：https://mp.weixin.qq.com/s/LYUBRNallHcjnhJb1R3ZBg

日常在網(wǎng)站使用過程中經(jīng)常遇到圖形驗證，今天準(zhǔn)備自己做個圖形驗證碼，這算是個簡單的功能，也適合新手練習(xí)的，便于自己學(xué)習(xí)。
主要用到的庫--PIL圖像處理庫，簡單的思路，我們需要隨機(jī)的顏色，隨機(jī)的數(shù)字或字母，隨機(jī)的線條、點作為干擾元素拼湊成一張圖片。

生成隨機(jī)顏色，返回的是rgb三色。

def getRandomColor():
    r = random.randint(0, 255)
    g = random.randint(0, 255)
    b = random.randint(0, 255)
    return (r, g, b)

從數(shù)字、大小寫字母里生成隨機(jī)字符。

def getRandomChar():
    random_num = str(random.randint(0, 9))
    random_lower = chr(random.randint(97, 122))  # 小寫字母a~z
    random_upper = chr(random.randint(65, 90))  # 大寫字母A~Z
    random_char = random.choice([random_num, random_lower, random_upper])
    return random_char

圖片操作，生成一張隨機(jī)背景色的圖片，隨機(jī)生成5種字符+5種顏色，在圖片上描繪字，由于默認(rèn)的字體很小，還需要對字進(jìn)行處理，不同系統(tǒng)下的字體文件存放位置不一樣，這里我是把window下的 arial.ttf 字體復(fù)制到了當(dāng)前文件夾下直接使用的。

# 圖片寬高
width = 160
height = 50

def createImg():
    bg_color = getRandomColor()
    # 創(chuàng)建一張隨機(jī)背景色的圖片
    img = Image.new(mode="RGB", size=(width, height), color=bg_color)
    # 獲取圖片畫筆，用于描繪字
    draw = ImageDraw.Draw(img)
    # 修改字體
    font = ImageFont.truetype(font="arial.ttf", size=36)
    for i in range(5):
        # 隨機(jī)生成5種字符+5種顏色
        random_txt = getRandomChar()
        txt_color = getRandomColor()
        # 避免文字顏色和背景色一致重合
        while txt_color == bg_color:
            txt_color = getRandomColor()
        # 根據(jù)坐標(biāo)填充文字
        draw.text((10 + 30 * i, 3), text=random_txt, fill=txt_color, font=font)
    # 打開圖片操作，并保存在當(dāng)前文件夾下
    with open("test.png", "wb") as f:
        img.save(f, format="png")

這個時候可以看到文件夾下面的圖片

這里是張很清晰的圖片，為了有干擾元素，這里還需要在圖片加入些線條、點作為干擾點。
隨機(jī)畫線，在圖片寬高范圍內(nèi)隨機(jī)生成2個坐標(biāo)點，并通過隨機(jī)顏色產(chǎn)生線條。

def drawLine(draw):
    for i in range(5):
        x1 = random.randint(0, width)
        x2 = random.randint(0, width)
        y1 = random.randint(0, height)
        y2 = random.randint(0, height)
        draw.line((x1, y1, x2, y2), fill=getRandomColor())

隨機(jī)畫點，隨機(jī)生成橫縱坐標(biāo)點。

def drawPoint(draw):
    for i in range(50):
        x = random.randint(0, width)
        y = random.randint(0, height)
        draw.point((x,y), fill=getRandomColor())

生成方法

def createImg():
    bg_color = getRandomColor()
    # 創(chuàng)建一張隨機(jī)背景色的圖片
    img = Image.new(mode="RGB", size=(width, height), color=bg_color)
    # 獲取圖片畫筆，用于描繪字
    draw = ImageDraw.Draw(img)
    # 修改字體
    font = ImageFont.truetype(font="arial.ttf", size=36)
    for i in range(5):
        # 隨機(jī)生成5種字符+5種顏色
        random_txt = getRandomChar()
        txt_color = getRandomColor()
        # 避免文字顏色和背景色一致重合
        while txt_color == bg_color:
            txt_color = getRandomColor()
        # 根據(jù)坐標(biāo)填充文字
        draw.text((10 + 30 * i, 3), text=random_txt, fill=txt_color, font=font)
    # 畫干擾線點
    drawLine(draw)
    drawPoint(draw)
    # 打開圖片操作，并保存在當(dāng)前文件夾下
    with open("test.png", "wb") as f:
        img.save(f, format="png")

最終生成的圖片

這里介紹的是圖片生成的方法，可以將圖片直接顯示在前端，也可以使用接口返回url。這里我簡單的把圖片做成鏈接顯示在網(wǎng)頁上，https://www.manjiexiang.cn/blog/validate 用Django做的，需要注意的是圖片保存的路徑。

歡迎關(guān)注我的個人博客：https://www.manjiexiang.cn/

更多精彩歡迎關(guān)注微信號：春風(fēng)十里不如認(rèn)識你
一起學(xué)習(xí)，一起進(jìn)步，歡迎上車，有問題隨時聯(lián)系，一起解決?。?！