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