【python】python postgresql获取插入数据的id

在Python中使用PostgreSQL数据库插入数据后，可以通过RETURNING子句来获取插入数据的id。以下是一个示例：

import psycopg2

# 连接到PostgreSQL数据库
conn = psycopg2.connect(database="your_database", user="your_username", password="your_password", host="your_host", port="your_port")

# 创建一个游标对象
cur = conn.cursor()

# 执行插入数据的SQL语句，并使用RETURNING子句获取插入数据的id
cur.execute("INSERT INTO your_table (column1, column2) VALUES (%s, %s) RETURNING id", ("value1", "value2"))

# 获取插入数据的id
inserted_id = cur.fetchone()[0]
print("插入的数据id为:", inserted_id)

# 提交事务
conn.commit()

# 关闭游标和数据库连接
cur.close()
conn.close()

在上面的示例中，我们首先连接到PostgreSQL数据库，然后创建一个游标对象。接下来，我们执行插入数据的SQL语句，并使用RETURNING id子句来获取插入数据的id。然后，我们使用fetchone()方法获取查询结果的第一行数据，即插入数据的id。最后，我们提交事务并关闭游标和数据库连接。

请注意，上述示例中的参数（数据库名称、用户名、密码、主机和端口）需要根据你的实际情况进行修改。