--- a/group.py +++ b/group.py @@ -18,11 +18,9 @@ created_by = db.Column(db.Integer, db.ForeignKey('users.id')) creator = db.relationship('User', backref='groups', lazy='dynamic') - memberships = db.relationship('Membership', backref='groups', lazy='dynamic') def __init__(self, name): self.name = name - def persist(self): # add record --- a/membership.py +++ b/membership.py @@ -14,6 +14,9 @@ gid = db.Column(db.Integer, db.ForeignKey('groups.id')) created = db.Column(db.DateTime, default=datetime.utcnow) + user = db.relationship('User', backref='memberships', lazy='dynamic') + group = db.relationship('Group', backref='memberships', lazy='dynamic') + def __init__(self, uid, gid): self.uid = uid self.gid = gid --- a/user.py +++ b/user.py @@ -14,7 +14,6 @@ id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(80)) email = db.Column(db.String(120), unique=True) - memberships = db.relationship('Membership', backref='user', lazy='dynamic') created = db.Column(db.DateTime, default=datetime.utcnow) last_active = db.Column(db.DateTime, default=datetime.utcnow) @@ -35,17 +34,14 @@ def getByName(username): #Query record by username tofind = User.query.filter_by(username=username).first() return tofind - - + def persist(self): #To create record db.session.add(self) db.session.commit() - def delete(self): # To delete a record db.session.delete(self) db.session.commit() - def __repr__(self):