使用Python生成隨機圖片驗證碼的代碼詳解

更新時間：2023年07月28日 08:35:07 作者：小L～～～

當我們在寫一個Web項目的時候一般要寫登錄操作，而為了安全起見，現(xiàn)在的登錄功能都會加上輸入圖片驗證碼這一功能，所以本文就給大家介紹一下如何使用Python生成隨機圖片驗證碼,需要的朋友可以參考下

Python 生成隨機圖片驗證碼

在寫一個Web項目的時候一般要寫登錄操作，而為了安全起見，現(xiàn)在的登錄功能都會加上輸入圖片驗證碼這一功能，在利用Django開發(fā)Web項目的過程中，可以使用 Python 生成一個如下所示的圖片驗證碼：

pillow詳細內(nèi)容參考官方文檔：https://pillow.readthedocs.io/en/stable/?badge=latest

pillow的GitHub地址：https://github.com/python-pillow/Pillow

安裝pillow包

pip install pillow

pillow包生成圖片基本用法

創(chuàng)建圖片

from PIL import Image
img = Image.new(mode='RGB', size=(120, 30), color=(255, 255, 255))
# 在圖片查看器中查看
img.show()
# 保存到本地
with open('code.png', 'wb') as f:
    img.save(f, format='png')

創(chuàng)建一支畫筆，用于在圖片上畫任意內(nèi)容

img = Image.new(mode='RGB', size=(120, 30), color=(255, 255, 255))
draw = ImageDraw.Draw(img, mode='RGB')

畫點

# 第一個參數(shù)表示坐標，第二個參數(shù)表示顏色
draw.point([100, 100], fill="red")
draw.point([300, 300], fill=(255, 255, 255))

畫線

# 第一個參數(shù)表示起始坐標和結(jié)束坐標，第二個參數(shù)表示顏色
draw.line((100, 100, 100, 100), fill="red")
draw.line((100, 100, 300, 100), fill=(255, 255, 255))

畫圓

# 第一個參數(shù)：表示起始坐標和結(jié)束坐標（圓要畫在其中間）
# 第二個參數(shù)：表示開始角度
# 第三個參數(shù)：表示結(jié)束角度
# 第四個參數(shù)：表示顏色
draw.arc((100,100,300,300),0,90,fill="red")

寫文本

# 第一個參數(shù)：表示起始坐標
# 第二個參數(shù)：表示寫入內(nèi)容
# 第三個參數(shù)：表示顏色
draw.text([0,0],'python',"red")

特殊字體文字

如下所示，將下載下來的ttf字體文件導入就可以生成帶有對應(yīng)字體的圖片

分享一個免費下載字體網(wǎng)站：http://www.webpagepublicity.com/free-fonts.html

找到自己下載的字體文件保存好后按照如下示例進行寫代碼即可

# 第一個參數(shù)：表示字體文件路徑
# 第二個參數(shù)：表示字體大小
font = ImageFont.truetype("kumo.ttf", 28)
# 第一個參數(shù)：表示起始坐標
# 第二個參數(shù)：表示寫入內(nèi)容
# 第三個參數(shù)：表示顏色
# 第四個參數(shù)：表示顏色
draw.text([0, 0], 'python', "red", font=font)

生成圖片驗證碼

import random
from PIL import Image, ImageDraw, ImageFont, ImageFilter
# 生成默認含4個字符驗證碼的圖片
def check_code(width=120, height=30, char_length=4, font_file='Monaco.ttf', font_size=28):
    code = []
    img = Image.new(mode='RGB', size=(width, height), color=(255, 255, 255))
    draw = ImageDraw.Draw(img, mode='RGB')
    def rndChar():
        """
        生成隨機大小寫字母
        :return:
        """
        up_or_low = random.randint(0, 1)
        if up_or_low == 0:
            return chr(random.randint(97, 122))
        return chr(random.randint(65, 90))
    def rndColor():
        """
        生成隨機顏色
        :return:
        """
        return (random.randint(0, 255), random.randint(10, 255), random.randint(64, 255))
    # 寫文字
    font = ImageFont.truetype(font_file, font_size)
    for i in range(char_length):
        char = rndChar()
        code.append(char)
        h = random.randint(0, 4)
        draw.text([i * width / char_length, h], char, font=font, fill=rndColor())
    # 寫干擾點
    for i in range(40):
        draw.point([random.randint(0, width), random.randint(0, height)], fill=rndColor())
    # 寫干擾圓圈
    for i in range(40):
        draw.point([random.randint(0, width), random.randint(0, height)], fill=rndColor())
        x = random.randint(0, width)
        y = random.randint(0, height)
        draw.arc((x, y, x + 4, y + 4), 0, 90, fill=rndColor())
    # 畫干擾線
    for i in range(5):
        x1 = random.randint(0, width)
        y1 = random.randint(0, height)
        x2 = random.randint(0, width)
        y2 = random.randint(0, height)
        draw.line((x1, y1, x2, y2), fill=rndColor())
    img = img.filter(ImageFilter.EDGE_ENHANCE_MORE)
    return img, ''.join(code)
if __name__ == '__main__':
    img, code_str = check_code()
    print(code_str)
    with open('code.png', 'wb') as f:
        img.save(f, format='png')
    # 1. 直接打開
    # img,code = check_code()
    # img.show()
    # 2. 寫入文件
    # img,code = check_code()
    # with open('code.png','wb') as f:
    #     img.save(f,format='png')
    # 3. 寫入內(nèi)存(Python3) **在web項目開發(fā)中一般將生成的圖片寫入內(nèi)存而不是寫入文件保存起來**
    # from io import BytesIO
    # stream = BytesIO()
    # img.save(stream, 'png')
    # stream.getvalue()
    # 4. 寫入內(nèi)存（Python2）
    # import StringIO
    # stream = StringIO.StringIO()
    # img.save(stream, 'png')
    # stream.getvalue()

運行結(jié)果如下所示：