[PYTHON/PSYCOPG2] execute_values 함수 : BULK INSERT 처리하기

icodebroker PSYCOPG2 2024-06-18

■ execute_values 함수를 사용해 BULK INSERT를 처리하는 방법을 보여준다.

▶ 테스트 테이블 (SQL)


CREATE TABLE test
(
    ID   INTEGER PRIMARY KEY,
    NAME VARCHAR,
    AGE  INTEGER
);

1

2

3

4

5

6

7

8

CREATE TABLE test

(

ID INTEGER PRIMARY KEY,

NAME VARCHAR,

AGE INTEGER

);

▶ 예제 코드 (PY)


import psycopg2
import psycopg2.extras

with psycopg2.connect("host='192.168.29.118' dbname='testdb' user='testuser1' password='test1234'") as connection:
    with connection.cursor() as cursor:

        cursor.execute("DELETE FROM test")

        psycopg2.extras.execute_values(
            cursor, "INSERT INTO test (id, name, age) VALUES %s;",
            [(1, '홍길동', 20), (2, '김철수', 30), (3,'이영희', 25)]
        )

        connection.commit()

        cursor.execute("SELECT * FROM test")

        resultList = cursor.fetchall()

        for result in resultList:
            print(result)

"""
(1, '홍길동', 20)
(2, '김철수', 30)
(3, '이영희', 25)
"""

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

import psycopg2

import psycopg2.extras

with psycopg2.connect("host='192.168.29.118' dbname='testdb' user='testuser1' password='test1234'") as connection:

with connection.cursor() as cursor:

cursor.execute("DELETE FROM test")

psycopg2.extras.execute_values(

cursor, "INSERT INTO test (id, name, age) VALUES %s;",

[(1, '홍길동', 20), (2, '김철수', 30), (3,'이영희', 25)]

)

connection.commit()

cursor.execute("SELECT * FROM test")

resultList = cursor.fetchall()

for result in resultList:

print(result)

"""

(1, '홍길동', 20)

(2, '김철수', 30)

(3, '이영희', 25)

"""

▶ requirements.txt


psycopg2-binary==2.9.9

1

2

3

psycopg2-binary==2.9.9