본문 바로가기

DB엔지니어가 공부하는 python

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

파이썬을 이용하여 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