功能演示
图片验证码
生成图片验证码
需要用到pillow库做图片处理
string选出验证码候选区字符
random随机选择字符生成验证码
def captcha():
# 生成一个图片
image = Image.new('RGB', (120, 30), color=(73, 109, 137))
# 字体
font_path = 'C:\Windows\Fonts\Arial.ttf'
# 创建一个绘画对象,用于在图像上绘制文本和图形
fnt = ImageFont.truetype(font_path, 15)
d = ImageDraw.Draw(image)
# 随机从大写字母和数字里选4个字符做验证码
captcha_text = ''.join(random.choices(string.ascii_uppercase + string.digits, k=4))
# 在图像上绘制验证码文本
d.text((10, 10), captcha_text, font=fnt, fill=(255, 255, 0))
# 添加干扰线条
for _ in range(random.randint(3, 5)):
'''添加3到5条干扰线'''
# 线条起始点和结束点坐标
start = (random.randint(0, image.width), random.randint(0, image.height))
end = (random.randint(0, image.width), random.randint(0, image.height))
d.line([start, end], fill=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
# 添加噪点
for _ in range(100):
'''添加100个噪点'''
# 随机生成噪点坐标
xy = (random.randrange(0, image.width), random.randrange(0, image.height))
# 绘制噪点
d.point((xy, xy), fill=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
首页
验证完用户名,密码和验证码之后,如果登陆成功跳转到首页。这里写的超级简陋。只是为了实验有没有跳转效果。需要更改。
@app.route('/index')
def index():
return jsonify({'data':'首页'})
登录
从前端获取用户名、密码和验证码。这里用户名和验证码是写死的。andmin和123。实际中需要从数据库加载用户信息。
@app.route('/', methods=['GET', 'POST'])
def login():
if request.method == 'GET':
return render_template('index.html')
if request.method == 'POST':
data = request.get_json()
username = data['username']
password = data['password']
code = data['code']
if username == 'admin' and password == '123' and code == session['captcha']:
return jsonify({'status':'success'}), 200
else:
return jsonify({'status':False, 'message':'用户名/密码/验证码错误'})
return jsonify({'status':False, 'message':'请求方法错误'})
另外 验证信息还需要进一步细化。通常是用户名/密码错误和验证码错误是分开的。可以多加一层if实现。