写入三个数据库,其中一个带有烧瓶的提交

时间:2018-06-25 14:52:50

标签: python flask foreign-keys flask-sqlalchemy table-relationships


我从一个星期前开始使用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中执行此操作,还是必须像代码的注释部分一样手动执行此操作? 感谢您的帮助。

1 个答案:

答案 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)