（二十八）Flask之wtforms库【上手使用篇】

孤寒者2024-04-21 19:59

目录：

每篇前言：

|-----------------------------------------------------------------------|
| 🏆🏆作者介绍：【孤寒者】---CSDN全栈领域优质创作者、HDZ核心组成员、华为云享专家Python全栈领域博主、CSDN原力计划作者 |

WTForms 是一个用于处理Web表单的Python库。它设计简单，易于使用，广泛用于Web应用程序的表单处理，特别是与Flask等框架一起使用。

wtforms依照功能类别来说有以下几个类别：

  • Forms: 主要用于表单验证、字段定义、HTML生成，并把各种验证流程聚集在一起进行验证。
  • Fields: 主要负责渲染(生成HTML)和数据转换。
  • Validator:主要用于验证用户输入的数据的合法性。比如Length验证器可以用于验证输入数据的长度。
  • Widgets：html插件，允许使用者在字段中通过该字典自定义html小部件。
  • Meta：用于使用者自定义wtforms功能，例如csrf功能开启。
  • Extensions：丰富的扩展库，可以与其他框架结合使用，例如django。
bash 复制代码 
pip install wtforms==2.1

官方文档：

用户登录验证：

  • 当用户登录时，需要对用户提交的用户名和密码进行多种格式校验，如：

    用户名不能为空；长度必须大于6；

    密码不能为空；长度必须大于12；密码必须包含字母、数字、特殊字符等（通过正则自定义）...

python 复制代码 
from flask import Flask, render_template, request, redirect
from wtforms import Form
from wtforms.fields import core, simple
from wtforms import validators
from wtforms import widgets

app = Flask(__name__, template_folder='templates')
app.debug = True


class LoginForm(Form):
    user = simple.StringField(
        label='用户名',
        validators=[
            validators.DataRequired(message='用户名不能为空'),
            validators.Length(min=6, max=18, message=f'用户名长度必须大于{min}且小于{max}')
        ],
        widget=widgets.TextInput(),
        render_kw={'class': 'form-control'}    # 设置生成的html标签的属性
    )

    pwd = simple.PasswordField(
        label='密码',
        validators=[
            validators.DataRequired(message='密码不能为空'),
            validators.Length(min=8, message=f'密码必须大于{min}'),
            # validators.Regexp(regex=r'^(?=.*[A-Z])'  # 至少一个大写字母
            #                         r'(?=.*[a-z])'  # 至少一个小写字母
            #                         r'(?=.*\d)'  # 至少一个数字
            #                         r'(?=.*[@$!%*?&])'  # 至少一个特殊字符
            #                         r'[A-Za-z\d@$!%*?&]{8,}$',  # 总长度至少8个字符
            #                   message='密码至少8个字符，至少一个大写字母，一个小写字母，一个数字和一个特殊字符')
        ],
        widget=widgets.PasswordInput(),
        render_kw={'class': 'form-control'}
    )


@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        form = LoginForm()
        return render_template('login.html', form=form)

    form = LoginForm(formdata=request.form)
    if not form.validate():
        return render_template('login.html', form=form)
	# 对用户提交数据进行校验，form.data是校验完成后的数据字典
    if form.data['user'] == '1234567' and form.data['pwd'] == '123456789':
        print('用户提交的数据通过格式验证，提交的值为：', form.data)
        return 'Login OK~'
    else:
        return render_template('login.html',msg='用户名或密码错误', form=form)


if __name__ == '__main__':
    app.run()

login.html：

html 复制代码 
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>用户登录</title>
</head>
<body>
<form action="" method="post">
    {{form.user}} {{form.user.errors[0]}}
    {{form.pwd}} {{form.pwd.errors[0]}}
    <input type="submit" value="提交">{{msg}}
</form>
</body>
</html>

用户注册验证：

  • 注册页面需要让用户输入：用户名、密码、确认密码、性别、爱好...

来一个实战例子，底下会拆分详讲：

python 复制代码 
from flask import Flask, render_template, request, redirect
from wtforms import Form
from wtforms.fields import core, simple, html5
from wtforms import validators
from wtforms import widgets


app = Flask(__name__, template_folder='templates')
app.debug = True


class RegisterForm(Form):
    name = simple.StringField(
        label='用户名',
        validators=[
            validators.DataRequired()
        ],
        widget=widgets.TextInput(),
        render_kw={'class': 'form-control'},
        default='GuHanZhe'                         # 页面输入框默认值
    )

    pwd = simple.PasswordField(
        label='密码',
        validators=[
            validators.DataRequired(message='密码不能为空')
        ],
        widget=widgets.PasswordInput(),
        render_kw={'class': 'form-control'}
    )

    pwd_confirm = simple.PasswordField(
        label='确认密码',
        validators=[
            validators.DataRequired(message='确认密码不能为空'),
            validators.EqualTo('pwd', message='两次密码输入不一致')    # EqualTo作用是比较当前字段和指定字段名的字段值是否相等
        ],
        widget=widgets.PasswordInput(),
        render_kw={'class': 'form-control'}
    )

    email = html5.EmailField(
        label='邮箱',
        validators=[
            validators.DataRequired(message='邮箱不能为空'),
            validators.Email(message='邮箱格式有误')
        ],
        widget=widgets.TextInput(input_type='email'),
        render_kw={'class': 'form-control'}
    )

    gender = core.RadioField(
        label='性别',
        choices=(
            (1, '男'),
            (2, '女'),
        ),
        coerce=int
    )

    city = core.SelectField(
        label='城市',
        choices=(
            ('bj', '北京'),
            ('sh', '上海')
        )
    )

    hobby = core.SelectMultipleField(
        label='爱好',
        choices=(
            (1, '篮球'),
            (2, '足球')
        ),
        coerce=int
    )

    favor = core.SelectMultipleField(
        label='爱好',
        choices=(
            (1, '篮球'),
            (2, '足球')
        ),
        widget=widgets.ListWidget(prefix_label=False),
        option_widget=widgets.CheckboxInput(),
        coerce=int,
        default=[1, 2]
    )

    def __int__(self, *args, **kwargs):
        super(RegisterForm, self).__init__(*args, **kwargs)
        self.favor.choices = ((1, '篮球'), (2, '足球'), (3, '羽毛球'))

    def validate_pwd_confirm(self, field):
        """
        自定义pwd_confirm字段规则，例：与pwd字段是否一致
        """
        # 最开始初始化时，self.data中已有所有值
        if field.data != self.data['pwd']:
            # raise validators.ValidationError('密码不一致')   # 继续后续字段的验证
            raise validators.StopValidation('密码不一致')      # 不再继续后续字段的验证


@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'GET':
        form = RegisterForm(data={'gender': 1})
        return render_template('register.html', form=form)

    form = RegisterForm(formdata=request.form)
    if form.validate():
        print('用户提交数据通过格式验证，提交的值为：', form.data)
    else:
        print(form.errors)
    return render_template('register.html', form=form)


if __name__ == '__main__':
    app.run()

  1. 有关于上述代码中coerce=int的作用：

    在wtforms库中，coerce参数是用来强制转换字段值的参数。coerce=int的作用是将选项中的值强制转换为整数类型。

    在RadioField中，choices参数定义了可选的值，它是一个元组，其中包含了每个选项的值和标签。在上图中，每个选项的值是1和2，而标签是'男'和'女'。由于HTTP表单提交的数据通常是字符串形式，使用coerce=int告诉wtforms将用户提交的值强制转换为整数类型。

    这对于确保表单数据的类型与后端处理代码的期望类型一致非常有用。在这个例子中，gender字段的值将被强制转换为整数，而不是保持为字符串。这样，在处理表单数据时就可以直接使用整数类型，而不需要手动进行类型转换。

  2. 上述我定义了这么多的字段，难道在写前端register.html代码的时候要一个个敲吗？？？

    肯定不是的！

    wtforms支持我们使用for循环~

    回想一下：一个类的实例如何才能支持for循环？

    在《Python全栈系列教程》专栏里讲过，只要一个类内部实现了iter魔法方法，且这个方法返回了一个迭代器，那么这个类的实例就支持for循环。

    所以来看下wtforms源码，确认一下：

    进Form ---> 进BaseForm：

使用示例：

python 复制代码 
from flask import Flask, render_template, request, redirect
from wtforms import Form
from wtforms.fields import core, simple, html5
from wtforms import validators
from wtforms import widgets


app = Flask(__name__, template_folder='templates')
app.debug = True


class RegisterForm(Form):
    name = simple.StringField(
        label='用户名',
        validators=[
            validators.DataRequired()
        ],
        widget=widgets.TextInput(),
        render_kw={'class': 'form-control'},
        default='GuHanZhe'                         # 页面输入框默认值
    )

    pwd = simple.PasswordField(
        label='密码',
        validators=[
            validators.DataRequired(message='密码不能为空')
        ],
        widget=widgets.PasswordInput(),
        render_kw={'class': 'form-control'}
    )

    pwd_confirm = simple.PasswordField(
        label='确认密码',
        validators=[
            validators.DataRequired(message='确认密码不能为空'),
            validators.EqualTo('pwd', message='两次密码输入不一致')    # EqualTo作用是比较当前字段和指定字段名的字段值是否相等
        ],
        widget=widgets.PasswordInput(),
        render_kw={'class': 'form-control'}
    )

    email = html5.EmailField(
        label='邮箱',
        validators=[
            validators.DataRequired(message='邮箱不能为空'),
            validators.Email(message='邮箱格式有误')
        ],
        widget=widgets.TextInput(input_type='email'),
        render_kw={'class': 'form-control'}
    )

    gender = core.RadioField(
        label='性别',
        choices=(
            (1, '男'),
            (2, '女'),
        ),
        coerce=int
    )

    city = core.SelectField(
        label='城市',
        choices=(
            ('bj', '北京'),
            ('sh', '上海')
        )
    )

    hobby = core.SelectMultipleField(
        label='爱好',
        choices=(
            (1, '篮球'),
            (2, '足球')
        ),
        coerce=int
    )

    favor = core.SelectMultipleField(
        label='爱好',
        choices=(
            (1, '篮球'),
            (2, '足球')
        ),
        widget=widgets.ListWidget(prefix_label=False),
        option_widget=widgets.CheckboxInput(),
        coerce=int,
        default=[1, 2]
    )


@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'GET':
        form = RegisterForm()
        return render_template('register.html', form=form)

    form = RegisterForm(formdata=request.form)
    if form.validate():
        print('用户提交数据通过格式验证，提交的值为：', form.data)
    else:
        print(form.errors)
    return '登录成功~'


if __name__ == '__main__':
    app.run()

register.html：

html 复制代码 
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>用户注册</title>
</head>
<body>
<form action="" method="post">
    {% for item in form %}
    <p>{{ item.label }}: {{item}} {{item.errors[0]}}</p>
    {% endfor %}
    <input type="submit" value="提交">
</form>
</body>
</html>

  • 问题引入：

    实际生产中，可能有些下拉框的值是从数据库中取出展示的。此处以city这个为例：

    python 复制代码 
        city = core.SelectField(
        label='城市',
        choices=SQLHelper.fetch_all('select id, name from city_info', {})
    )

    如果直接运行访问，这个下拉框是没有任何问题的。

    但是实际生产中可能会遇到的一个问题是：Flask服务没关，但是往数据库这张表加了几条数据，那么，不管怎样刷新页面，这个下拉框都不会出现这些新加的数据。

    但是将Flask服务重启一下就OK 了。

    原因很简单------因为在RegisterForm类中这些字段都是静态字段，运行的时候只执行一次！

  • 解决方法就是：

    重写RegisterForm类的构造方法：让每次实例化这个类的时候都执行一次sql查询语句并更新对应字段值：

    python 复制代码 
        def __int__(self, *args, **kwargs):
        super(RegisterForm, self).__init__(*args, **kwargs)
        self.city.choices = SQLHelper.fetch_all('select id, name from city_info', {})

如何自定义校验规则：

【方法名以validate_开头，后面是对应需要校验的字段名】

python 复制代码 
    def validate_pwd_confirm(self, field):
        """
        自定义pwd_confirm字段规则，例：与pwd字段是否一致
        """
        # 最开始初始化时，self.data中已有所有值
		# field.data就是当前字段的值
        if field.data != self.data['pwd']:
            # raise validators.ValidationError('密码不一致')   # 继续后续字段的验证
            raise validators.StopValidation('密码不一致')      # 不再继续后续字段的验证

抽象解读使用wtforms编写的类：

简单谈一嘴：

比如LoginForm类中的user字段，很容易知道user是一个实例（可以点进去StringField发现它是一个类）：

在视图函数中，实例化form后将其传给了前端：

而前端就相当于执行了print(form.user)

那么这就执行对应类StringField的str魔法方法，这个玩意返回什么，页面就看到什么~

开始抽象：

bash 复制代码 
class LoginForm(Form):
	user = 类（正则， 插件）
	字段 = 类（正则， 插件）
	字段 = 类（正则， 插件）
	字段 = 类（正则， 插件）


form = LoginForm(Form)
# 生成html标签
print(form.user)   ------>   类.__str__   ------>   插件.xx方法

# 验证
form = LoginForm(formdata=request.form)
if form.validate():
	# 内部找到所有的字段：
	#                  比如：user + 用户发过来的对应的数据   ------>    正则校验
相关推荐
程序员的开发手册8 分钟前
新手教学系列——慎用Flask-SQLAlchemy慢日志记录
数据库·python·flask·sqlalchemy
Java4ye13 分钟前
Netty 是如何解析 Redis RESP 协议的——请求篇
后端
木觞清2 小时前
Django学习第三天
python·学习·django
电饭叔2 小时前
《python程序语言设计》2018版第5章第52题利用turtle绘制sin函数
开发语言·python
YCCX_XFF213 小时前
ImportError: DLL load failed while importing _imaging: 操作系统无法运行 %1
开发语言·python
FutureUniant5 小时前
GitHub每日最火火火项目（7.7）
python·计算机视觉·ai·github·视频
杰哥在此5 小时前
Java面试题：讨论持续集成/持续部署的重要性，并描述如何在项目中实施CI/CD流程
java·开发语言·python·面试·编程
PY1786 小时前
Python的上下文管理器
数据库·python·oracle
Struggle to dream7 小时前
Python编译器的选择
开发语言·python
刘铸纬7 小时前
Golang中defer和return顺序
开发语言·后端·golang
热门推荐
012024年江苏省研究生数学建模科研创新实践大赛C题气象数据高精度融合技术研究论文和代码分析02Coze扣子平台完整体验和实践（附国内和国际版对比）03组基轨迹建模 GBTM的介绍与实现（Stata 或 R）042024亚太赛（中文赛）数学建模竞赛选题建议+初步分析05yolov8实战第五天——yolov8+ffmpg实时视频流检测并进行实时推流——（推流，保姆教学）06【经验分享】Ubuntu22.04安装微信(linux官方版)072024 年第十四届 APMCM 亚太地区大学生数学建模竞赛B题超详细解题思路+数据预处理问题一代码分享082025张宇考研数学，百度网盘视频课+36讲PDF讲义+真题09【立创EDA-PCB设计基础】6.布线铺铜实战及细节详解10通用大模型VS垂直大模型，你更青睐哪一方？