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 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 表单
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_template
else:
template = self.create_template
return 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 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 的方式实现已有数据的回填,该方案在编辑中非常有用