sqlalchemy 更新column,如果存在则更新，如果不存在，则添加新记录

Posted 2020-11-25 路痴队长

sqlalchemy 更新column,如果存在则更新，如果不存在，则添加新记录

class Friendship(Base):
    __tablename__ = 'friendship'
    
    id = Column(Integer,primary_key=True)
    me = Column(Integer,ForeignKey('users.id'))
    friend = Column(Integer,ForeignKey('users.id'))
    def __init__(self,me,friend):
            self.me = me 
            self.friend = friend
    
    def check_existing(self):
        existing = session.query(Friendship).filter_by(me=self.me,friend=self.friend).first()
        if not existing:
            friendship = Friendship(self.me,self.friend)
        else:
            friendship = existing
        session.close()
        return friendship

if __name__ == '__main__':
    friendship = Friendship(self.from_user_id,self.friend_id)
    friendship = friendship.check_existing()
    session.add(friendship)
    session.commit()
    session.close()