[AWS] postgresql python 연동하기

python에 postgresql을 연동하기 위해

psycopg2 라이브러리를 설치해 줍니다.

pip install psycopg2 # 윈도우

$ pip install psycopg2-binary # 리눅스

 

 

 

 

 

 

import psycopg2 as pg2

conn=pg2.connect(database="test_db",	# DB이름
                 host="IP",		# IP
                 user="testuser",	# 사용자 이름
                 password="0000",	# 비밀번호
                 port="5432")		# posgresql 포트번호(=5432)
        
cur = conn.cursor() # cursor 생성

설치한 psycopg2를 import 해줍니다.

 

DB 정보를 입력하여 연결합니다.

 

 

 

 

 

 

 

 

테이블을 만들고 (create), insert, select 해보며

잘 연동되고 실행되는지 확인해 보겠습니다.

 

cur.execute("CREATE TABLE Fruits (id INTEGER, date DATE, name VARCHAR(80))")
conn.commit()

 

Fruits라는 테이블과

id, date, name이라는 칼럼 3개를 만들었습니다.

 

DBeaver로 확인한 결과

 

 

 

 

 

 

 

 

 

cur.execute("INSERT INTO Fruits VALUES(1, '2022-08-10', 'orange')")
cur.execute("INSERT INTO Fruits VALUES(1, '2022-08-13', 'banana')")
cur.execute("INSERT INTO Fruits VALUES(1, '2022-08-15', 'cherry')")
conn.commit()

 

orange, banana, cherry라는 데이터 3개를 insert 해주었습니다.

 

DBeaver로 확인한 결과

 

 

 

 

 

 

 

 

import pandas as pd
cur.execute("SELECT * FROM Fruits;")
pd.DataFrame(cur.fetchall())

 

select한 데이터를 보기 위해 pandas를 불러왔습니다.

 

 

 

 

 

 

 

cur.close()
conn.close()

DB 연결을 종료해줍니다.

 

 

 

 

 

 

 

 

class로 만들기

참고 : Python Postgresql 연동하기