[파이썬] PostgreSQL DB 연동 - DB 연결 및 조회, psycopg2

PostgreSQL DB 연동

---

PostgreSQL DB 연동 - DB 연결 및 조회, psycopg2

---

들어가며

---

PostgreSQL은 오픈 소스 데이터베이스 시스템 중 하나로 전 세계적으로 널리 사용되는 데이터베이스입니다. 이 글에서는 파이썬을 활용하여 PostgreSQL DB에 접속하고, SELECT 문으로 조회하는 방법을 소개합니다.

---

• psycopg2 패키지 설치
• DB 연결
• DB 조회

---

psycopg2 패키지 설치

---

먼저 필요한 패키지를 설치합니다. PostgreSQL을 사용하기 위한 파이썬 패키지는 psycopg2 입니다.

pip install psycopg2

---

DB 연결

---

위에서 설치한 패키지를 import하고 DB 연결 정보를 입력합니다. DB에 연결하기 위해 필요한 정보는 host 주소, port 번호, database 이름, user 이름, password 입니다. 다음과 같이 연결이 필요한 DB의 정보를 입력합니다.

import psycopg2

connect = psycopg2.connect(host='HOST',
                           port=5432,
                           database='DATABASE',
                           user='USER',
                           password='PASSWORD')

cursor = connect.cursor()

---

DB 조회

---

DB를 연결했다면, SELECT 문을 활용하여 DB를 조회하겠습니다. pandas의 read_sql과 위의 DB 정보를 con 옵션으로 데이터를 조회합니다.

import pandas as pd

df = pd.read_sql('''SELECT * FROM TEST_DB''', con=connect)

---

DB 연결을 종료할 때는 connection과 cursor를 모두 close 명령어를 사용해 닫아야 합니다.

cursor.close()
connect.close()

---

전체 코드

---

# --------------------------------------------------------------------------------- */

## psycopg2 패키지 설치

# ! pip install psycopg2


# --------------------------------------------------------------------------------- */

## DB 연결

import psycopg2

connect = psycopg2.connect(host='HOST',
                           port=5432,
                           database='DATABASE',
                           user='USER',
                           password='PASSWORD')

cursor = connect.cursor()


# --------------------------------------------------------------------------------- */

## DB 조회

import pandas as pd

df = pd.read_sql('''SELECT * FROM TEST_DB''', con=connect)

# DB 연결 종료
cursor.close()
connect.close()