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

背景：

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

材料：

wtforms+flask_sqlalchemy

制作：

上代码

1、模型代码

from .exts import db
from flask_login import UserMixin
from datetime import datetimeclass 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 表单

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视图下实现表单新增

   @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_templateelse:template = self.create_templatereturn self.render(template, form=form)

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

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

    @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 buttonreturn 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_templateelse:template = self.edit_templatereturn 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 的方式实现已有数据的回填，该方案在编辑中非常有用