使用Flask-SQLAlchemy创建基本表

时间:2014-07-16 13:07:05

标签: python database sqlalchemy flask-sqlalchemy

我对如何使用Flask-SQLAlchemy创建表格感到困惑 - 我在其他地方发现的示例似乎使用了与下面不同的初始化(例如,使用Base)。

db = SQLAlchemy(app)
migrate = Migrate(app, db)

# Define models
roles_users = db.Table('roles_users',
        db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
        db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))

class Role(db.Model, RoleMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))

class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(255))
    active = db.Column(db.Boolean())
    confirmed_at = db.Column(db.DateTime())
    nickname = db.Column(db.String(255))
    favcolor = db.Column(db.String(255)) #FAVS
    favshape = db.Column(db.String(255)) #FAVS
    favflower = db.Column(db.String(255)) #FAVS
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))

    def __str__(self):
            return '<User id=%s email=%s>' % (self.id, self.email)

基本上,我们想要将所有fav变量(favcolorfavshapefavflower)放在一个单独的表格中它自己的,名为favorites

如何确保我可以使用primary_key访问这些收藏夹?

似乎很基本!希望有一个简单的方法来做到这一点......

1 个答案:

答案 0 :(得分:0)

创建&#39; UserFavorites&#39;与用户具有一对一关系的模型。

class UserFavorites(db.Model):
    user_id = db.Column(db.Integer, db.ForeignKey(User.id), primary_key=True)
    color = db.Column(db.String, nullable=False, default='')
    shape = db.Column(db.String, nullable=False, default='')
    flower = db.Column(db.String, nullable=False, default='')

    user = db.relationship(User, backref=db.backref('favorites', uselist=False))

在创建用户实例时创建UserFavorites实例。

user = User(email='example@example.com', favorites=UserFavorites(color='green'))
db.session.add(user)
db.session.commit()

user = User.query.filter(
    User.email == 'example@example.com'
).options(
    db.joinedload(User.favorites)
).one()
print user.favorites.color

您可以加入此关系,向用户查询其收藏夹。

likes_green = User.query.join(User.favorites).filter(
    UserFavorites.color == 'green'
).all()
print [u.email for u in likes_green]