wtforms+flask_sqlalchemy在flask-admin视图下实现日期的修改与更新

背景：

在flask-admin 的modelview视图下实现自定义视图的表单修改/编辑是件不太那么容易的事情，特别是想不自定义前端view的情况下。

材料：

wtforms+flask_sqlalchemy

制作：

上代码

1、模型代码

python 复制代码

from .exts import db
from flask_login import UserMixin
from datetime import datetime


class AiConfig(db.Model, UserMixin):
    __tablename__ = 'ai_config'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    created_at = db.Column(db.TIMESTAMP, default=datetime.utcnow)
    updated_at = db.Column(db.TIMESTAMP, default=datetime.utcnow)
    status = db.Column(db.Integer)
    ai_name = db.Column(db.String(128))

注：默认日期函数可以采用 default=datetime.utcnow 来实现

2、视图测代码

a、AiConfigForm 表单

python 复制代码

class AiConfigForm(form.Form):

    def get_countries():
        return Organ.query.order_by('updated_at').all()

    ai_name = fields.StringField('AI名称', validators=[validators.DataRequired()])
    ai_code = fields.StringField('AI编号', validators=[validators.DataRequired()])
    ai_extends = fields.TextAreaField('配置扩展', widget=TextArea())
    root_url = fields.URLField('根URL', validators=[validators.DataRequired()], widget=URLInput())
    organ = QuerySelectField('机构名称', validators=[validators.DataRequired()], query_factory=get_countries, get_label='org_name')  
    description = fields.TextAreaField('描述', widget=TextArea())
    status = fields.SelectField('状态', choices=[(1, '有效'), (0, '无效')], default=0)

b、modeview视图下实现表单新增

python 复制代码

   @expose('/new/', methods=('GET', 'POST'))
    def create_view(self):
        bean = AiConfig()
        bean.updated_at = datetime.today()
        form = AiConfigForm(request.form, obj=bean)
        if self.validate_form(form):
            model = self.create_model(form)
            if model:
                flash(gettext('Record was successfully created.'), 'success')
                return redirect(self.get_save_return_url(model, is_created=True))

        if self.create_modal and request.args.get('modal'):
            template = self.create_modal_template
        else:
            template = self.create_template
        return self.render(template, form=form)

注：上面代码片段中这样修改update_at 是不生效的，目前尚未找到在这个代码交互中修改表单AiConfigForm 中并未定义字段update_at 属性的方法（因为属性赋值是在其self.create_model()内置函数中实现的）

c、modelview 视图下实现表单修改

python 复制代码

    @expose('/edit/', methods=('GET', 'POST'))
    def edit_view(self):
        id = request.args['id']
        return_url = request.values.get('url') or self.get_url('.index_view')
        model = self.get_one(id)
        if model is None:
            flash(gettext('Record does not exist.'), 'error')
            return redirect(return_url)
        form = AiConfigForm(request.form, obj=model)

        if self.validate_form(form):
            model.updated_at = datetime.today()
            if self.update_model(form, model):
                flash(gettext('Record was successfully saved.'), 'success')
                # save button
                return redirect(self.get_save_return_url(model, is_created=False))   

        if request.method == 'GET' or form.errors:
            self.on_form_prefill(form, id)

        if self.edit_modal and request.args.get('modal'):
            template = self.edit_modal_template
        else:
            template = self.edit_template
        return self.render(template, form=form)

注：上述代码片段中 model.updated_at = datetime.today() 是有效的，因为在self.update_model(form,model) 内置函数中只是将form 中表单数据通过 form.populate_obj(model) 值cp 的方式简单覆盖来实现，我们只需要在调用该方法前将model 进行修改便可生效。

效果

1、编辑/新增 view中不存在日期2

2、数据库中存在日期

3、列表中该记录存在日期

注：这就充分说明通过上述逻辑实现了新增日期和修改日期的后台修改

疑问与经验

疑问

1、为啥不在AiConfigForm 表单中通过影藏字段来实现日期呢？

该方案有尝试，但是不行，因为数据库属性是日期类型的，实体中该字段也是日期类型，如果只是在表单中采用影藏字段，新增时非常好用，只需要通过form.updated_at.data=datetime.today()实现赋值，但编辑时就会因为模型字段与表单中针对updated_at 字段的类型定义不一致导致无法通过populate_obj 函数进行模型值赋值导致报异常，同时由于此刻表单中并没有updated_at这样的日期类型字段，有的只是影藏字段，故无法form.updated_at.data=datetime.today() 实现赋值。当然如果不怕麻烦可以写2个form 视图就可以友好结果上述问题。

2、为啥不直接自定义view模板呢？

当然可以，只是因为懒。

3、为啥不建2个AiConfigForm 视图，一个用于编辑一个用于新增？

当然可以，而且可以做到很优雅，只是懒

经验

1、在ModelView 的子视图中重写部分方法，如@expose('/new/', methods=('GET', 'POST'))和@expose('/edit/', methods=('GET', 'POST'))，中如果采用wtforms 自定义表单，但又不想自定义表单视图，那么只需要简单的通过form = AiConfigForm(request.form, obj=bean) 方法在上面新增、编辑函数中使用wtforms 表单。但这里要说明的是AiConfigForm(request.form, obj=bean) 中的 obj 对象接收的是模型，通过把模型赋值给obj 的方式实现已有数据的回填，该方案在编辑中非常有用