web-dev-qa-db-ja.com

Python / postgres / psycopg2：挿入された行のIDを取得する

私はPythonとpsycopg2を使用してpostgresに接続しています。

行を挿入すると...

sql_string = "INSERT INTO hundred (name,name_slug,status) VALUES ("
sql_string += hundred_name + ", '" + hundred_slug + "', " + status + ");"
cursor.execute(sql_string)

...挿入した行のIDを取得するにはどうすればよいですか？しようとしています：

hundred = cursor.fetchall()

RETURNING idの使用中にエラーを返します：

sql_string = "INSERT INTO domes_hundred (name,name_slug,status) VALUES ("
sql_string += hundred_name + ", '" + hundred_slug + "', " + status + ") RETURNING id;"
hundred = cursor.execute(sql_string)

単にNoneを返します。

更新：currvalも同様です（このコマンドを直接postgresで使用しても機能します）：

sql_string = "SELECT currval(pg_get_serial_sequence('hundred', 'id'));"
hundred_id = cursor.execute(sql_string)

誰でもアドバイスできますか？

ありがとう！

pythonpostgresqlpsycopg2
81
AP257
cursor.execute("INSERT INTO .... RETURNING id")
id_of_new_row = cursor.fetchone()[0]

また、値を含むSQL文字列を手動で作成しないでください。値を個別に渡すことができます（そうすべきです！）エスケープする必要がなく、SQLインジェクションが不可能です：

sql_string = "INSERT INTO domes_hundred (name,name_slug,status) VALUES (%s,%s,%s) RETURNING id;"
cursor.execute(sql_string, (hundred_name, hundred_slug, status))
hundred = cursor.fetchone()[0]

詳細については、psycopgのドキュメントを参照してください。 http://initd.org/psycopg/docs/usage.html#passing-parameters-to-sql-queries

165
ThiefMaster

私は同様の問題を抱えていたのでここで終わりましたが、まだRETURNING ID節をサポートしていないPostgres-XCを使用しています。その場合、次を使用できます。

cursor.execute（ 'INSERT INTO ........'）
 cursor.execute（ 'SELECT LASTVAL（）'）
 lastid = cursor.fetchone（）['lastval']

万が一に役立つ場合に備えて！

11
Jamie Brown

RETURNING句を検討してください http://www.postgresql.org/docs/8.3/static/sql-insert.html

2
Mike Graham