from captcha.image import ImageCaptcha
import random
class Captcha_Get():
  def __init__(self,
         CHAR_SET = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'],
         CAPTCHA_LEN = 4):
    self.CHAR_SET = CHAR_SET
    self.CAPTCHA_LEN = CAPTCHA_LEN
  def get_captcha(self):
    captcha_list = []
    for i in range(self.CAPTCHA_LEN):
      random_choice = random.choice(self.CHAR_SET)
      captcha_list.append(random_choice)
    return captcha_list
  def get_captcha_image(self):
    image = ImageCaptcha()
    captcha_list = self.get_captcha()#返回一個(gè)列表
    captcha_str = ''.join(captcha_list)#將列表的所有內(nèi)容整合成一個(gè)字符串
    captcha_image = image.generate(captcha_str)
    #captcha_image返回<_io.BytesIO object at 0x000001C8758C8728>，它是一個(gè)<class '_io.BytesIO'>
    return captcha_str,captcha_image#因?yàn)橐蚫jango登陸相結(jié)合所以驗(yàn)證碼的內(nèi)容也要返回

from django.http import JsonResponse,HttpResponse
from .captcha_image import *
def auth_code_port(request):
  """
  生成驗(yàn)證碼的接口
  :param request:
  :return: 圖片的對(duì)象
  """
  captcha_str, image_64 = Captcha_Get().get_captcha_image()
  #request.session['captcha_str'] = captcha_str #將驗(yàn)證碼內(nèi)容放入session以便于后邊判斷，但必須遷移數(shù)據(jù)庫，否則報(bào)錯(cuò)
  #image = 'data:image/png;base64,'+image_64
  resp = HttpResponse(image_64, content_type='image/png')
  return resp