[python] 파이썬에서 Psycopg2를 사용하여 데이터베이스 테이블의 샘플 데이터를 생성하는 방법은?

22 Nov 2023

python

데이터베이스 테이블에 샘플 데이터를 생성하는 것은 개발자들에게 중요한 작업입니다. Psycopg2라는 Python 모듈을 활용하면 PostgreSQL 데이터베이스에 접속하여 테이블에 데이터를 추가할 수 있습니다. 아래는 Psycopg2를 사용하여 데이터베이스 테이블에 샘플 데이터를 생성하는 간단한 예제입니다.

Psycopg2 설치하기

먼저, Psycopg2 모듈을 설치해야 합니다. 아래의 명령어를 사용하여 pip를 통해 Psycopg2를 설치할 수 있습니다.

pip install psycopg2

데이터베이스 연결 설정하기

다음으로, Psycopg2를 사용하여 데이터베이스에 연결합니다. 아래의 예제는 로컬 PostgreSQL 데이터베이스에 연결하는 방법입니다.

import psycopg2

# 데이터베이스 연결 설정
conn = psycopg2.connect(
    host="localhost",
    database="mydatabase",
    user="myuser",
    password="mypassword"
)

SQL 쿼리 실행하기

이제 데이터베이스에 연결한 후, SQL 쿼리를 실행하여 데이터를 추가할 수 있습니다. 아래의 예제는 “users”라는 테이블에 샘플 데이터를 추가하는 방법입니다.

# 커서 생성
cur = conn.cursor()

# SQL 쿼리 실행
sql = "INSERT INTO users (name, age) VALUES (%s, %s)"
data = [("John", 25), ("Alice", 30), ("Bob", 35)]

# 다중 행 삽입
cur.executemany(sql, data)

# 데이터베이스에 변경 사항 반영
conn.commit()

# 커서와 데이터베이스 연결 종료
cur.close()
conn.close()

위의 예제에서는 executemany 메서드를 사용하여 다중 행 삽입을 수행하였습니다. 데이터를 추가한 후에는 반드시 commit 메서드를 호출하여 데이터베이스에 변경 사항을 반영해야 합니다.

Psycopg2를 사용하여 파이썬에서 데이터베이스 테이블에 샘플 데이터를 생성하는 방법에 대해 살펴보았습니다. 자세한 내용은 Psycopg2의 공식 문서를 참고하시기 바랍니다.

Psycopg2 공식 문서: https://www.psycopg.org/docs/