如何在python3中使用pillow模塊實現一個驗證碼功能

這期內容當中小編將會給大家帶來有關如何在python3中使用pillow模塊實現一個驗證碼功能，文章內容豐富且以專業的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。

# -*- coding: utf-8 -*-
# __author__: Pad0y

from PIL import Image, ImageDraw, ImageFont
from random import choice, randint, randrange
import string

# 候選字符集,大小寫字母+數字
chrs = string.ascii_letters + string.digits


def selected_chrs(length):
  """
  返回length個隨機字符串
  :param length:
  :return:
  """
  result = ''.join(choice(chrs) for _ in range(length))
  return result


def get_color():
  """
  設置隨機顏色
  :return:
  """
  r = randint(0, 255)
  g = randint(0, 255)
  b = randint(0, 255)
  return (r, g, b)


def main(size=(200, 100), chrNumber=6, bgcolor=(255, 255, 255)):
  """
  定義圖片大小，驗證碼長度，背景顏色
  :param size:
  :param chrNumber:
  :param bgcolor:
  :return:
  """
  # 創建空白圖像和繪圖對象
  image_tmp = Image.new('RGB', size, bgcolor)
  draw = ImageDraw.Draw(image_tmp)

  # 生成并計算隨機字符的寬度和高度
  text = selected_chrs(chrNumber)
  font = ImageFont.truetype('c:\\windows\\fonts\\Roboto-Regular.ttf', 48) # 選定一款系統字體
  width, height = draw.textsize(text, font)
  if width + 2*chrNumber > size[0] or height > size[1]:
    print('Size Error!')
    return

  # 繪制字符串
  startX = 0
  width_eachchr = width // chrNumber # 計算每個字符寬度
  for i in range(chrNumber):
    startX += width_eachchr + 1
    position = (startX, (size[1]-height)//2+randint(-10, 10)) # 字符坐標, Y坐標上下浮動
    draw.text(xy=position, text=text[i], font=font, fill=get_color()) # 繪制函數

  # 對像素位置進行微調，實現驗證碼扭曲效果
  img_final = Image.new('RGB', size, bgcolor)
  pixels_final = img_final.load()
  pixels_tmp = image_tmp.load()
  for y in range(size[1]):
    offset = randint(-1, 0) # randint()相當于閉區間[x,y]
    for x in range(size[0]):
      newx = x + offset # 像素微調
      if newx >= size[0]:
        newx = size[0] - 1
      elif newx < 0:
        newx = 0
      pixels_final[newx, y] = pixels_tmp[x, y]

  # 繪制隨機顏色隨機位置的干擾像素
  draw = ImageDraw.Draw(img_final)
  for i in range(int(size[0]*size[1]*0.07)): # 7%密度的干擾像素
    draw.point((randrange(size[0]), randrange(size[1])), fill=get_color()) # randrange取值范圍是左開右閉

  # 繪制隨機干擾線，這里設置為8條
  for i in range(8):
    start = (0, randrange(size[1]))
    end = (size[0], randrange(size[1]))
    draw.line([start, end], fill=get_color(), width=1)

  # 繪制隨機弧線
  for i in range(8):
    start = (-50, -50) # 起始位置在外邊看起來才會像弧線
    end = (size[0]+10, randint(0, size[1]+10))
    draw.arc(start+end, 0, 360, fill=get_color())

  # 保存圖片
  img_final.save('Veri_code.jpg')
  img_final.show()


if __name__ == '__main__':
  main((200, 100), 6, (255, 255, 255))

效果圖如下

上述就是小編為大家分享的如何在python3中使用pillow模塊實現一個驗證碼功能了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關知識，歡迎關注億速云行業資訊頻道。