如何对 Flask-admin 字段执行验证

Question

我想对 Flask Admin 执行一些验证。 Flask 应用程序连接到后端 postgreSQL 数据库并使用 SQLAlchemy ORM。

我希望能够对 ServerView(flask.ext.admin.contrib.sqla.modelView) 中的两个字段（lan_nics、wan_nics）执行验证检查。 基本上我想确保两个字段的总和不大于某个数字。 这是表格：

服务器视图

这是 test_app 代码。 我试图按照flask-admin 的文档进行操作。 我添加了一个 def，然后在验证器字典的 form_args 中引用了该 def ......但它似乎不起作用。 目前我只是想确保两个字段的值都小于 5。

from flask import Flask
from flask.ext.admin import Admin
from flask.ext.admin.contrib.sqla import ModelView
from flask.ext.sqlalchemy import SQLAlchemy
from wtforms.validators import ValidationError
import psycopg2


# Flask and Flask-SQLAlchemy initialization here

app = Flask(__name__)
db = SQLAlchemy(app)
app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://postgres:postgres@127.0.0.1/testdb'
app.secret_key = 'MySecret'
conn = psycopg2.connect("dbname='testdb' user='postgres' host='127.0.0.1' password='postgres'")

class Server(db.Model):
    __tablename__ = "server"
    id = db.Column('server_id', db.Integer, primary_key=True)
    model = db.Column('model', db.Unicode)
    lan_nics = db.Column('lan_nics', db.Integer)
    wan_nics = db.Column('wan_nics', db.Integer)

    def __str__(self):
        return self.model

class Hardware(db.Model):
    __tablename__ = "hardware"
    id = db.Column('hardware_id', db.Integer, primary_key=True)
    model = db.Column('model', db.Unicode)
    max_nics = db.Column('max_nics', db.Integer)

    def __str__(self):
        return self.model


class ServerView(ModelView):

    # Disable model creation
    can_create = True

    # Override displayed fields
    column_list = ('model', 'lan_nics', 'wan_nics')

    column_display_all_relations = True
    column_searchable_list = ('model')
    column_labels = dict(model='Model', lan_nics='LAN Nics',  wan_nics='WAN NICs')
    form_columns = [ 'model', 'lan_nics', 'wan_nics' ]


    def max_allowed(form, field):
        if field.data > 5:
            raise ValidationError('Max number of interfaces exceeded')

    form_args = dict(
        lan_nics=dict(validators=[max_allowed]), 
        wan_nics=dict(validators=[max_allowed])
    )


class HardwareView(ModelView):
    # Disable model creation
    can_create = True

    # Override displayed fields
    column_list = ('model', 'max_nics')

    column_display_all_relations = True
    column_searchable_list = ('model')
    column_labels = dict(model='Model', max_nics='Max Nics')
    form_columns = ['model', 'max_nics' ]

    def __str__(self):
        return self.model


admin = Admin(app, name="MyApp1")
# Add administrative views here
admin.add_view(ModelView(Server, db.session))
admin.add_view(ModelView(Hardware, db.session))

app.run(port=8000, host="0.0.0.0")

Answer 1

而不是form_args ，我做了on_model_change ：

class ServerView(ModelView):
    ...
    ...
    ...

    def on_model_change(self, form, model, is_created):
        hardware = "hardware"
        query = "SELECT nics FROM hardware WHERE model = '{}'".format(hardware)
        with conn.cursor() as cur:
            cur.execute(query)
            max_nics = int(cur.fetchone()[0])
        if (int(form.lan.data) + int(form.wan.data) > max_nics):
            raise ValidationError('Max number of interfaces exceeded!')
        else:
            return model

Answer 2

实际上，您的第一种方法有效。 你有一个小错字：而不是if field.data > 5:你应该检查if len(field.data) > 5: 。

flask-admin 的文档并没有真正涵盖验证器，但在幕后，它只是将工作委托给 WTForms。 他们的文档非常有帮助： https : //wtforms.readthedocs.io/en/3.0.x/validators/它还显示了现有的验证器，例如Length ，您可以改写：

from wtforms.validators import Length

form_args = dict(
    lan_nics=dict(
        validators=[Length(min=5, message='Max number of interfaces exceeded')]
    ), 
)