pymysql DBHelp insert 插入并获取ID

#获取最后的插入的ID
def getLastId(db):
    db.execute("select last_insert_id();")
    data = db.fetchall();
    return data[0][0]
#插入dict key_value 格式   返回ID       
def insertSql(db,tb,com):
    keys="";
    values="";
    for (key,value) in  com.items(): 
        keys+="`"+key+"`,"
        if(type(value) is int):
             values+=str(value)+","
        else:
           values+="'"+pymysql.escape_string(value)+"',"
    sql="INSERT INTO `touyan`.`"+tb+"` ("+keys[0:-1]+") VALUES ("+values[0:-1]+");"    
    print(sql)
    db.execute(sql)
    d = db.fetchall();
    return getLastId(db)
#created||get 存在返回ID 不存在则插入 然后返回ID
def GetOrInsertStock(db,name):
    db.execute("select id from stockholder where name='"+pymysql.escape_string(name)+"';")
    data = db.fetchall();
    if(len(data)>0):
        return data[0][0]
    else:
         sql="INSERT INTO `touyan`.`stockholder` (`name`) VALUES ('"+pymysql.escape_string(name)+"');"
         db.execute(sql)
         return getLastId(db)

猜你喜欢

转载自blog.csdn.net/lkjasdgfh/article/details/80546669