我从一个星期前开始使用flask,现在已经遇到了一个小问题,我不知道flask是否可以将其作为框架来处理,或者是否需要以“旧”方式解决。
这是情况。
我有三个数据库 Platten , Label 和 Artist 。要填写表格,我有一个表格可以输入LP的信息(艺术家,标签,发行日期等)。
通过提交表格,我要检查Artist db,如果Artist已经是否存在。如果是exitend,请获取ID并将其保存为LP。如果艺术家不在数据库中,则将其写入并保存新的ID。
标签也一样。
这是我的models.py文件: 从应用程序导入数据库 从werkzeug.security导入generate_password_hash,check_password_hash 从flask_login导入UserMixin 从应用程序导入login_manager
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(64), index=True, unique=True)
email = db.Column(db.String(120), index=True, unique=True)
password_hash = db.Column(db.String(128))
def __repr__(self):
return '<User {}>'.format(self.username)
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password_hash, password)
class Label(db.Model):
__tablename__ = 'labels'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
def __repr__(self):
return '<Label {}>'.format(self.name)
class Artist(db.Model):
__tablename__ = 'artists'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
def __repr__(self):
return '<Artist {}>'.format(self.name)
class Platte(db.Model):
__tablename__ = 'platten'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64), index=True)
label_id = db.Column(db.Integer, db.ForeignKey('labels.id'))
label = db.relationship('Label', backref=db.backref('platten'), lazy=True)
release_date = db.Column(db.Date())
artist_id = db.Column(db.Integer, db.ForeignKey('artists.id'))
artist = db.relationship('Artist', backref=db.backref('platten'), lazy=True)
def __repr__(self):
return '<Platte {}>'.format(self.title)
这是我的route.py文件:
from flask import render_template, flash, redirect, url_for, request
from app import app, db
from app.forms import LoginForm, RegistrationForm, PlattenForm
from flask_login import current_user, login_user, logout_user, login_required
from app.models import User, Platte, Artist, Label
from werkzeug.urls import url_parse
@app.route('/')
@app.route('/index')
@login_required
def index():
platten = Platte.query.all()
return render_template('index.html', title='Home', platten=platten)
@app.route('/add', methods=['GET', 'POST'])
@login_required
def platten_add():
form = PlattenForm()
if form.validate_on_submit():
artist = Artist(name=form.artist.data)
label = Label(name=form.label.data)
# artist_check = Artist.query.filter_by(name=artist).first()
# if artist_check is None:
# db.session.add(artist)
# db.session.commit()
# artist_id = Artist.query.filter_by(name=artist).first()
# artist_id = artist_id.id
# else:
# artist_id = artist_check.id
platte = Platte(title=form.title.data, label_id=label.id, label=label, artist_id=artist.id, artist=artist, release_date=form.release_date.data)
db.session.add(platte)
db.session.commit()
flash('Platte ' + platte.title + ', hinzugefügt!')
return redirect(url_for('platten_add'))
return render_template('platten_add.html', title='Add', form=form)
routes.py中的注释部分将是“旧的”老式方式,即我将如何进行数据库插入。
有没有办法在flask中执行此操作,还是必须像代码的注释部分一样手动执行此操作? 感谢您的帮助。
答案 0 :(得分:0)
否,这在flask_sqlalchemy
中是不可能的。为了访问artist.id
之类的对象ID,必须已将其保存。
也可以是一对多,这是首选的方式:
class Label(db.Model):
__tablename__ = 'labels'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
platten = db.relationship('Platte', backref='labels')
def __repr__(self):
return '<Label {}>'.format(self.name)
class Artist(db.Model):
__tablename__ = 'artists'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
platten = db.relationship('Platte', backref='artists')
def __repr__(self):
return '<Artist {}>'.format(self.name)
class Platte(db.Model):
__tablename__ = 'platten'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64), index=True)
label_id = db.Column(db.Integer, db.ForeignKey('labels.id'))
release_date = db.Column(db.Date())
artist_id = db.Column(db.Integer, db.ForeignKey('artists.id'))
def __repr__(self):
return '<Platte {}>'.format(self.title)
这又意味着您现在可以执行以下操作:
@app.route('/add', methods=['GET', 'POST'])
@login_required
def platten_add():
form = PlattenForm()
if form.validate_on_submit():
artist = Artist.query.filter_by(name=form.artist.data).first()
if not artist:
artist = Artist(name=form.artist.data)
db.session.add(artist)
label = Label.query.filter_by(name=form.label.data).first()
if not label:
label = Label(name=form.label.data)
db.session.add(label)
platte = Platte(title=form.title.data, release_date=form.release_date.data)
label.platten.append(platte)
artist.platten.append(platte)
db.session.commit()
flash('Platte ' + platte.title + ', hinzugefügt!')
return redirect(url_for('platten_add'))
return render_template('platten_add.html', title='Add', form=form)