PostgreSQL データベースを使用して Python にデータを挿入した後、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 を取得します。最後に、トランザクションをコミットし、カーソルとデータベース接続を閉じます。
上記の例のパラメーター (データベース名、ユーザー名、パスワード、ホスト、ポート) は、実際の状況に応じて変更する必要があることに注意してください。