DB엔지니어가 공부하는 python

파이썬을 이용하여 postgresql DB에 접속하여 SQL 실행 결과를 변수에 저장 하여 활용하기 (1)

sTricky 2023. 2. 22. 16:18

파이썬을 이용하여 postgresql DB에 접속하여 SQL 실행 결과를 변수에 저장 하여 활용하기 (1)

안녕하세요.
우선 파이썬에서 postgresql DB에 접속하기 위한 라이브러리가 있어야겠죠.
바로 psycopg2를 활용 할 수 있습니다.

그리고나서 적당한 변수를 선언하고, DB 접속정보를 넣어주면 바로 SQL을 실행 할 수 있겠죠.
테이블 목록을 가지고 오는 쿼리를 한번 예로 코드를 작성 해보도록 하겠습니다.

import psycopg2

# PostgreSQL 데이터베이스 연결하기
conn = psycopg2.connect(
    host="아이피",
    database="데이터베이스",
    user="유저명",
    password="비밀번호"
)

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

# SQL 실행
cur.execute("SELECT tablename FROM pg_tables")  # SQL 입력하기

# 결과를 변수에 저장
rows = cur.fetchall()

# 연결 종료
cur.close()
conn.close()

이 결과를 가지고 for 문을 실행한다면 이런식으로 할 수 있겠죠.

for i in rows:
    print(i)

매우 간단하게 해결!

 

by.sTricky