Python mysql获取主键id

偏执的太偏执、 2021-06-24 15:59 662阅读 0赞

用到的mysql语句主要为：

> author\_id = cursor.lastrowid  
> 相等于 ： author\_id = conn.insert\_id()

有这么一段需求，插入数据后，获取返回的主键id，首次可以获取到，第二次插入失败，就获取不到主键id了。其实我们可以这样实现：

（一）需求：

例子:

我们将name ， plat（平台）为唯一主键，

因为name 在不同platform上可能相同，所以对于不同platform的name，我们都要将name存进去，是同一platform的name 就不再进行第二次存入

（二）Python代码实现：

# 1 先进行查询是否存在
    # 2 再进行更新操作（其实这一步可以不用的）
    # 3 对数据进行插入
    
    author_id = None
    ## （一）insert into mysql
    try:
        # 查重处理
        self.cursor.execute("""select author_id from tb_author where name = %s and platform = %s """,
                       [item['name'], item['platform']])
        # 是否有重复数据
        repetition = self.cursor.fetchone()
        # 重复
        if repetition:
            if len(repetition) > 0:
                author_id = repetition[0]
        else:
            # 更新
            result_code = self.cursor.execute(
                "UPDATE tb_author set avatar=%s,type=%s,fans_count=%s,scan_count=%s,collect_count=%s,is_delete=%s where tb_author.name = %s and tb_author.platform = %s",
                [item['avatar'], item['type'], item['fans_count'], item['scan_count'], item['collect_count'],
                 item['is_delete'], item['name'], item['platform']])
            # 第一次插入
            if not result_code:
                self.cursor.execute(
                    "insert into tb_author (name,platform,avatar,type,fans_count,scan_count,collect_count,is_delete,gmt_create,gmt_modified) values(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)",
                    (item['name'], item['platform'], item['avatar'], item['type'], item['fans_count'],
                     item['scan_count'], item['collect_count'], item['is_delete'], item['gmt_create'],
                     item['gmt_modified']))
                author_id = self.cursor.lastrowid
        print('author_id ==== > ',author_id)
    except Exception as e:
        print('inser mysql error ---> ',e)
    self.conn.commit()
    self.cursor.close()
    self.conn.close()