[python] 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)누구든지 조언 할 수 있습니까?
감사!
답변
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
답변
비슷한 문제가 있었기 때문에 여기에 왔지만 아직 RETURNING ID 절을 지원하지 않는 Postgres-XC를 사용하고 있습니다. 이 경우 다음을 사용할 수 있습니다.
cursor.execute ( 'INSERT INTO ........') cursor.execute ( 'SELECT LASTVAL ()') lastid = cursor.fetchone () [ 'lastval']
누구에게나 유용했던 경우를 대비하여!
