如何對 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')]
    ), 
)