sqlalchemy實現添加或更新（如果已經存在）

 測試類

class TableTest(Base):
    __tablename__ = 'table_test'

    id = Column(BIGINT(20),primary_key=True,autoincrement='auto')
    name = Column(String(50,'utf8mb4_unicode_ci'),nullable=False,unique=True)
    age=Column(INTEGER(11),nullable=False)
    height=Column(INTEGER(11),nullable=False)

 

關鍵代碼：

def GetInsertOrUpdateObj(cls,strFilter,**kw):
    '''
    cls:            Model 類名
    strFilter:      filter的參數.eg:"name='name-14'"
    **kw:           【屬性、值】字典,用於構建新實例，或修改存在的記錄

    '''
    #print('strFilter ; ',strFilter)
    existing = session.query(cls).filter(text(strFilter)).first()
    if not existing:
        res=cls()
        for k,v in kw.items():
            if hasattr(res,k):
                setattr(res,k,v)
        return res
    else:
        res = existing
        for k,v in kw.items():
            if hasattr(res,k):
                setattr(res,k,v)

        return res

 

調用示例：

if __name__=='__main__':        
    session.add(GetInsertOrUpdateObj(TableTest,"name='name-14'",name='name-14',age=88,height=4321))
    session.add(GetInsertOrUpdateObj(TableTest,"name='name-114'",age=33114,height=123.14,name='name-114'))

    session.commit()
    session.close()