Python 使用SQLAlchemy向相关表添加数据
我有以下到SQLAlchemy(使用Flask SQLAlchemy)对象的定义:Python 使用SQLAlchemy向相关表添加数据,python,sqlalchemy,Python,Sqlalchemy,我有以下到SQLAlchemy(使用Flask SQLAlchemy)对象的定义: class User(db.Model): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(20), unique=True) password = db.Column(db.String(30)) email = db.Column(db.String(45), unique=
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True)
password = db.Column(db.String(30))
email = db.Column(db.String(45), unique=True)
friends = db.relationship('Friend', backref='user',
lazy='dynamic')
def __init__(self, username, password, email):
self.username = username
self.password = password
self.email = email
def __repr__(self):
return "<User('%s','%s','%s')>" % (self.username, self.email, self.id)
class Friend(db.Model):
id = db.Column(db.Integer, primary_key=True)
userId = db.Column(db.Integer, db.ForeignKey('user.id'))
friendId = db.Column(db.Integer)
created = db.Column(db.DateTime)
def __init__(self, userId, friendId):
self.userId = userId
self.friendId = friendId
self.created = datetime.datetime.now()
def __repr__(self):
return "<Friend(%i,%i)>" % (self.userId, self.friendId)
获得朋友:
MyFriend = bpdata.User.query.filter_by(id=2).first()
现在我想做:
MyUser.Friends.Append(MyFriend)
这是可能的,还是我必须直接将好友ID添加到表friend中?我自己就知道了 我需要做的是:
MyUser.friends.append(Friend(MyUser.id, MyFriend.id))
然后提交更新
更新:
好吧,我找到了做我想做的事的正确方法。首先,我根本不需要朋友桌/班级。完整代码:
association_table = db.Table('association',
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('friend_id', db.Integer, db.ForeignKey('user.id'))
)
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True)
password = db.Column(db.String(30))
email = db.Column(db.String(45), unique=True)
friends = db.relationship("User",
secondary=association_table,
backref='added_by',
primaryjoin=id == association_table.c.user_id,
secondaryjoin=id == association_table.c.friend_id)
有了这些,我现在可以做以下工作:
>>> user1 = User.query.filter_by(id=1).first()
>>> user1.friends
[]
>>> user2 = User.query.filter_by(id=2).first()
>>> user1.friends.append(user2)
>>> user1.friends
[<User('user1','user1@admin.com','2')>]
>>> user1.friends[0].added_by
[<User('admin','admin@admin.com','1')>]
>user1=User.query.filter\u by(id=1.first())
>>>用户1.朋友
[]
>>>user2=User.query.filter_by(id=2.first)()
>>>user1.friends.append(user2)
>>>用户1.朋友
[]
>>>用户1。朋友[0]。已添加用户
[]
我认为你建立了错误的关系。一个用户有很多朋友,而同一个朋友可能是多个用户的朋友,因此您必须使用M2M而不是O2M。
>>> user1 = User.query.filter_by(id=1).first()
>>> user1.friends
[]
>>> user2 = User.query.filter_by(id=2).first()
>>> user1.friends.append(user2)
>>> user1.friends
[<User('user1','user1@admin.com','2')>]
>>> user1.friends[0].added_by
[<User('admin','admin@admin.com','1')>]