You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
flyapps/fir_ser/api/utils/auth/captcha.py

52 lines
1.8 KiB

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# author: liuyu
# date: 2022/2/7
import logging
from captcha.helpers import captcha_image_url
from captcha.models import CaptchaStore
from common.core.sysconfig import Config
from common.libs.storage.localApi import LocalStorage
logger = logging.getLogger(__name__)
class CaptchaAuth(object):
def __init__(self, captcha_key='', g_type=1):
self.g_type = g_type
self.captcha_key = captcha_key
self.random_char_fun = self.__generator()
def __generator(self):
if self.g_type == 1:
# 随机字符串
return 'captcha.helpers.random_char_challenge'
elif self.g_type == 2:
# 数学运算
return 'captcha.helpers.math_challenge'
return None
def __get_captcha_obj(self):
return CaptchaStore.objects.filter(hashkey=self.captcha_key).first()
def generate(self):
self.captcha_key = CaptchaStore.generate_key(self.random_char_fun)
captcha_image = captcha_image_url(self.captcha_key)
captcha_obj = self.__get_captcha_obj()
code_length = 0
if captcha_obj:
code_length = len(captcha_obj.response)
local_storage = LocalStorage(**Config.IOS_PMFILE_DOWNLOAD_DOMAIN)
return {"captcha_image": "/".join([local_storage.get_base_url(), captcha_image.strip("/"), '']),
"captcha_key": self.captcha_key,
"length": code_length}
def valid(self, verify_code):
captcha_obj = self.__get_captcha_obj()
logger.info(f"captcha_key:{self.captcha_key} verify_code:{verify_code} challenge:{captcha_obj}")
if captcha_obj:
if self.captcha_key and verify_code and verify_code.strip(" ").lower() == captcha_obj.response.lower():
return True
return False