Flask-WTF 和图形验证码

最新推荐文章于 2024-04-21 13:10:30 发布

一包五毛钱辣条

最新推荐文章于 2024-04-21 13:10:30 发布

阅读量695

点赞数

分类专栏： Python 文章标签： python

本文链接：https://blog.csdn.net/myworld_cxc/article/details/104763445

版权

Python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

Flask-WTF

表单处理的扩展库，提供了CSRF、字段校验等功能，使⽤⾮常⽅便

安装： pip install flask-wtf

注册表单类

from flask_wtf import FlaskForm  # 导⼊表单基类
from wtforms import StringField,PasswordField,SubmitField
from wtforms.validators import DataRequired,Length,EqualTo,Email

# 自定义表单注册类
class Register(FlaskForm):
    username = StringField()  # 用户名
    password = PasswordField()  # 密码
    pwdagain = PasswordField()  # 重复密码
    email = EmailField(validators=[Email("邮箱格式错误")])
    # wtforms.validators中的Regexp类可以使用正则表达式
    phonenum = StringField(validators=[Regexp(r'^(13\d|14[5|7]|15\d|166|17[3|6|7]|18\d)\d{8}$')])

添加视图函数，创建表单对象，并渲染模板⽂件

from form import Register  # 导入自定义的表单注册类

@app.route('/register/', methods=['GET', 'POST'])
def register():
    form = Register()   # 实例化表单类
    if request.method == 'POST':
    	# 这个⽅法是实现表单校验功能的 csrf，数据正确性 都通过了 则为真 否则为假    
        if form.validate_on_submit():
            # print(request.form)
			 print(form.username) #拿到username的整个标签
			 print(form.username.data) #取出username⾥⾯的value值
			 return '数据提交成功'
    return render_template(('register.html', **locals())

原生渲染表单

<!DOCTYPE html>
<html lang="en"> <head>
 <meta charset="UTF-8">
 <title>Title</title>
</head> 
<body> 
<h2>flask-wtf的表单类</h2> 
<form action="{{ url_for('register') }}" method="post">
 {# csrf #}
 {{ form.csrf_token }}
 {{ form.username.label }}
 {{form.username(style="color:red;",class='myself',placeholder='请输⼊⽤户名...') }} 
 #给当前的标签添加属性和值 关键字参数
 #循环迭代取出验证失败的错误信息（也就是你在验证器⾥的属性message的值）
 {% for error in form.username.errors %}
 <span style="color:red;">{{ error }}</span>
 {% endfor %}
 <br>
 {{ form.userpass.label }}
 {{ form.userpass }}
 {% for error in form.password.errors %}
 <span style="color:red;">{{ error }}</span>
 {% endfor %}
 <br> 
 {{ form.confirm.label }}
 {{ form.confirm() }}
 {% for error in form.pwdagain.errors %}
 <span style="color:red;">{{ error }}</span>
 {% endfor %}
 <br>
 {{ form.email.label }}
 {{ form.email() }}
 {% for error in form.email.errors %}
 <span style="color:red;">{{ error }}</span>
 {% endfor %}
 <br>
 {{ form.submit() }}
</form>
</body>
</html>

字段类型	说明
StringField	普通⽂本字段
Submit	Submit 提交按钮
PasswordField	密⽂字段
HiddenField	隐藏字段
RadioField	单选框
BooleanField	复选框
FileField	⽂件上传
SelectField	下拉框
TextAreaField	⽂本域
IntegerField	⽂本字段，值为整数
FloatField	⽂本字段，值为浮点数
DateField	datetime.date类型
DateTimeField	datetime.datetime类型

图形验证码

PIL：Python Imaging Library，已经是Python平台事实上的图像处理标准库了。PIL功能⾮常强⼤，但API却⾮常简单易⽤。

由于PIL仅⽀持到Python 2.7，因为长时间没有维护，于是⼀群志愿者在PIL的基础上创建了兼容的版本，名字叫Pillow，⽀持最新Python 3.x，⼜加⼊了许多新特性。
安装Pillow库
安装： pip install pillow

# Vcode.py
import os
from io import BytesIO
from random import randint
from PIL import Image, ImageFont, ImageDraw


class vcode:
    def __init__(self,width=100, height=34, size=4):
        self.width = width
        self.height = height
        self.size = size
        self.__code = ''  # 验证码字符串
        self.pen = None  # 画笔

    @property  # 装饰器访问私有化属性
    def code(self):
        return self.__code

    def generate(self):
        # 1.创建画布，颜色随机
        im = Image.new("RGB", (self.width, self.height),self.__rand_color(150))
        self.pen = ImageDraw.Draw(im)  # 在画布上创建画笔
        # 2.生成验证码字符串
        self.__rand_string()
        # 3.画验证码
        self.__draw_code()
        # 4.画干扰点
        self.__draw_point()
        # 5.画干扰线
        self.__rand_line()
        # 6.返回验证码图片
        buf = BytesIO()  # 缓冲区
        im.save(buf, 'png')  # 以pug格式将图片保存在缓冲区
        res = buf.getvalue()  # 获取图片的二进制
        return res

    # 创建个随机颜色的函数，数字越接近0颜色越浓，越接近255颜色越淡
    def __rand_color(self, min=0, max=255):
        return randint(min, max), randint(min, max), randint(min, max)

    # 生成验证码字符串
    def __rand_string(self):
        self.__code = ""  # 每次生成字符串前将之前缓存字符串重置为空
        for i in range(self.size):
            self.__code += str(randint(0, 9))

    # 画验证码
    def __draw_code(self):
        # 加载字体
        path = os.path.join(os.getcwd(),'static/fonts/SIMLI.TTF')
        # 设置字体大小和编码格式
        font1 = ImageFont.truetype(path, size=20, encoding="utf-8")
        # 计算字符串宽度
        width = (self.width - 20) // self.size
        # 逐个画字
        for i in range(len(self.__code)):
            x = 13 + width * i  # 计算字符的x坐标
            # 写入字，并随机颜色
            self.pen.text((x, 7), self.__code[i], font=font1, fill=self.__rand_color(0, 80))

    # 画点
    def __draw_point(self):
        for i in range(100):
            self.pen.point((randint(1, self.width-1), randint(1, self.height-1)), self.__rand_color(30, 100))

    # 画线
    def __rand_line(self):
        for i in range(5):
            # 逐个画线，随机起始点和结束点，并随机颜色
            self.pen.line([(randint(1, self.width-1), randint(1, self.height-1)),
            (randint(1, self.width-1), randint(1, self.height-1))], fill=self.__rand_color(50,150), width=2)


vc = vcode()  # 单例属性

# views.py
from Vcode import vc
# 验证码
@app.route("/verify/")
def verify_code():
    result = vc.generate()
    # 把验证码字符串保存到session
    session['code'] = vc.code
    # 创建响应对象
    response = make_response(result)
    response.headers["Content-Type"] = "image/png"
    return response

这样就可以通过/verify/路由，获取图形码。

一包五毛钱辣条

关注

0
点赞
踩
4

收藏

觉得还不错? 一键收藏
0
评论
Flask-WTF 和图形验证码

Flask-WTF表单处理的扩展库，提供了CSRF、字段校验等功能，使⽤⾮常⽅便安装： pip install flask-wtf注册表单类from flask_wtf import FlaskForm # 导⼊表单基类from wtforms import StringField,PasswordField,SubmitFieldfrom wtforms.validators im...
复制链接

扫一扫