■ Cursor 클래스의 rowcount 속성을 사용해 데이터 처리 건수를 구하는 방법을 보여준다.
▶ 테스트 테이블 (SQL)
|
1 2 3 4 5 6 7 8 |
CREATE TABLE test ( ID INTEGER PRIMARY KEY, NAME VARCHAR, AGE INTEGER ); |
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import psycopg2 from psycopg2 import Error try: connection = psycopg2.connect(host = "192.168.29.118", port = "5432", database = "testdb", user = "testuser1", password = "test1234") cursor = connection.cursor() cursor.execute("DELETE FROM test") print(f"삭제 : {cursor.rowcount}건") cursor.execute("INSERT INTO test (id, name, age) VALUES (1, '홍길동', 20)") print(f"추가 : {cursor.rowcount}건") cursor.execute("INSERT INTO test (id, name, age) VALUES (2, '김철수', 30)") print(f"추가 : {cursor.rowcount}건") cursor.execute("INSERT INTO test (id, name, age) VALUES (3, '이영희', 25)") print(f"추가 : {cursor.rowcount}건") connection.commit() print("트랜잭션이 완료되었습니다.") except (Exception, Error) as error: connection.rollback() print("트랜잭션이 취소되었습니다.") print(f"PostgreSQL 연결시 에러가 발생했습니다 : {error}") finally: if connection: cursor.close() connection.close() print("PostgreSQL 연결이 종료되었습니다.") |
▶ requirements.txt
|
1 2 3 |
psycopg2-binary==2.9.9 |