当前位置: 首页>后端>正文

flask_restx 参数验证自定义返回信息 flask 数据验证

Flask-WTF

它是简化了WTForms操作的一个第三方库。WTForms表单的两个主要功能是验证用户提交数据的合法性以及渲染模板。还包括一些其他的功能:CSRF保护,文件上传等。安装Flask-WTF默认也会安装WTForms,因此使用一下命令来安装Flask-WTF。
pip install html" class="superseo">flask-wtf

表单验证

安装完flask-wtf后,第一个功能就是用表单来做数据验证,现在又一个forms.py文件,然后再里面创建一个RegisForm的注册验证表单。
建立一个文件夹,在下面再建立一个templates,里面创建一个regist.htnl。
再文件夹下创建一个表单文件forms.py和主文件demo.py。
主文件:

from flask import Flask, request, render_template
from forms import RegistForm

app = Flask(__name__)

@app.route('/')
def index():
    return "首页"

# 接收GET请求
@app.route('/regist/', methods=['GET','POST'])
def regist():
    if request.method == 'GET':
        return render_template('regist.html')
    else:
        form = RegistForm(request.form)
        if form.validate():
            return "success"
        else:
            # 验证的错误信息
            print(form.errors)
            return "fail"
        # username = request.form.get('username')
        # password = request.form.get('password')
        # password_repate = request.form.get('password_repate')
        # if len(username) <3 or len(username) >10:
        #     return "用户名长度不正确"
        # if password != password_repate:
        #     return "两次密码不一致"
        # if len(password) <3 or len(password) >10:
        #     return "密码长度不正确"



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

forms文件:

from wtforms import Form,StringField, validators
from wtforms.validators import Length, Regexp, EqualTo

class RegistForm(Form):
    # 再后面加上message="xxxx" 出现错误就可以显示中文 在小括号里
    username = StringField(validators=[Length(min=3, max=10)])
    password = StringField(validators=[Length(min=3, max=10)])
    password_repate = StringField(validators=[Length(min=3, max=10), EqualTo("password")])

html文件:

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<form action="" method="post">
<!--    表单提交到flask的名字-->
    用户名: <input type="text" name="username"><br>
    密码: <input type="text" name="password"><br>
    确认密码: <input type="text" name="password_repate"><br>
    <input type="submit" value="注册">
</form>
</body>
</html>

其他验证

在主文件处demo.py文件定义一个登录

from forms import RegistForm, LoginForm

@app.route('/login/', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        form = LoginForm(request.form)
        if form.validate():  # 与forms文件建立联系
            return "success"
        else:
            # 验证的错误信息
            print(form.errors)
            return "fail"

在templates文件夹下新建一个login文件:

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<form action="" method="post">
    <table>
<!--        <tr>-->
<!--            <td>邮箱:</td>-->
<!--            <td><input type="text" name="email"></td>-->
<!--        </tr>-->
<!--         <tr>-->
<!--            <td>年龄:</td>-->
<!--            <td><input type="text" name="age"></td>-->
<!--        </tr>-->
<!--           <tr>-->
<!--            <td>用户名:</td>-->
<!--            <td><input type="text" name="username"></td>-->
<!--        </tr>-->
<!--           <tr>-->
<!--            <td>手机号:</td>-->
<!--            <td><input type="text" name="phone"></td>-->
<!--        </tr>-->
<!--          <tr>-->
<!--              <td>个人中心:</td>-->
<!--              <td><input type="text" name="info"></td>-->
<!--          </tr>-->
        <tr>
              <td>验证码:</td>
              <td><input type="text" name="captcha"></td>
          </tr>
        <tr>
            <td><input type="submit" value="提交"></td>
        </tr>
    </table>

</form>
</body>
</html>

在forms文件里定义验证:

from wtforms import Form,StringField, validators,IntegerField
from wtforms.validators import Length, Regexp, EqualTo, Email, NumberRange, InputRequired
from wtforms.validators import URL, ValidationError

class LoginForm(Form):
    # email = StringField(validators=[Email()])
    # age = IntegerField(validators=[NumberRange(1, 120, message='年龄范围错误')])
    # username = StringField(validators=[InputRequired(message='用户名必须要添加')])
    # phone = StringField(validators=[Regexp(r'1[385]\d{9}')])
    # info = StringField(validators=[URL()])
    # 验证码
    captcha = StringField(validators=[Length(min=4, max=4)])

    # 内容?
    def validate_captcha(self, field):
        print(field)
        if field.data != '5213':
            raise ValidationError('验证码错误')

RegistForm传递的是request.form进去进行初始化,并且判断form.validate会返回用户提交的数据是否满足表单的验证。

渲染模板

form还可以渲染模板,可以少写一点点的代码,比如重写以上例子,RegistForm表单代码如下:

class RegistForm(Form):
	name = StringField('用户名', validators=[Length(min=3, max=10)])
	emial = StringField('邮箱', validators=[email()])
	password = StringField('密码', validators=[DataRequired(), Length(min =6, max=10),EqualTo('confinEqualTo("password")])
	confirm = StringField('确认密码')

以上增加了第一个位置参数,用来在html文件中,做标签提示作用。
在app中的视图函数中, 修改为如下:

@app.route('/regist/', methods=['GET', 'POST']def
def regist():
	form = RegistForm(request.form)
	if request.method == 'POST' and form.validate():
		user = User(name=form.name.data,email=form.email.data,password=form.password.data)
		db.session.add(user)
		db.session.commit()
		return '注册成功'
	return render_tamplate('regist.html', form=form)

以上唯一不同的是在渲染模板的时候传入了form表单参数进去,这样在模板中就可以使用表单form变量了。
regist.html文件:诸如此类的形式添加

<tr>
		<td>{{ form.name.label }}</td>
	 	<td>{{ form.name() }}</td>
 	</tr>



https://www.xamrdz.com/backend/3z31942308.html

相关文章: