我的应用程序使用 Flask、Flask-SQLAlchemy、Flask-WTF 和 Jinja2 的组合。
在当前的版本中,我有一个设置表。该表只有一条记录和一个字段。最初该表包含零条记录。
我想要实现的是:
- 鉴于数据库中不存在任何条目,则显示可供用户输入的空表单
- 假设存在一个条目,则显示该条目,并且
- 如果用户更改了该值,则更新 db 中的 rec。
这是我的代码:
模型.py
class Provider(db.Model):
id = db.Column(db.Integer, primary_key = True)
rssfeed = db.Column(db.String(120), unique = True)
def __init__(self, rssfeed):
self.rssfeed = rssfeed
def __repr__(self):
return '<NZBMatrix feed url %r>' % self.rssfeed
forms.py
class SettingsForm(Form):
rssfeed = TextField('rssfed', validators= [Required()])
views.py
@app.route('/settings', methods=["GET","POST"])
def settings():
""" show settings """
provider = Provider.query.get(1)
form = SettingsForm(obj=provider)
print provider
if request.method == "POST" and form.validate():
if Provider.query.get(1) is None:
provider = Provider(rssfeed=form.rssfeed.data)
form.populate_obj(provider)
db.session.add(provider)
db.session.commit()
flash("Settings added")
return render_template("settings.html", form=form)
就目前情况而言,如果记录不存在,则此代码会创建一条记录,但rssfeed
列为空。
我怎样才能改变这个代码,以便它INSERT
如果记录不存在并且UPDATE
如果有的话?