[파이썬] 파이썬으로 DB(PostgreSQL) 연결해서 사용하기

파이썬으로 DB작업 가능한지 궁금하신 분들 계실텐데요, 파이썬으로 DB를 조회하고 사용할 수 있습니다.

 

1. 먼저 psycopg2를 import 해줍니다. 없는 분들은 설치해주면 됩니다.

#pip install psycopg2
import psycopg2

2. connect함수를 사용해서 postgresql과 연결합니다. host주소, db이름, 계정,비밀번호, 포트 정보가 필요합니다.

연결할 DB에 맞게 정보를 입력합니다.

db = psycopg2.connect(host='111.111.1.111', dbname='db',user='user',password='password',port=5432)

postgresql 디폴트 포트는 5432입니다.

 

3. 다음은 데이터 조작을 위한 객체를 생성합니다.

cursor=.db.cursor()

 

4. 마지막으로 execute() 함수를 통해서 실행하면 됩니다.

 

조회(select)

    cursor.execute(f"select rsrc_id from table where title like '20230111';")
    result=cursor.fetchone() # 하나만 return
    result=cursor.fetchmany(3) # 숫자만큼 return
    result=cursor.fetchall() # 모두 return

조회를 원하는 개수에 따라 선택해서 사용하시면 됩니다.

 

생성(create), 입력(insert), 수정(update), 삭제(delete)

cursor.execute("delete from table where value ='abc';")
db.commit()

변경되는 경우에는 commit을 해줘야 합니다.

 

5. 작업을 완료되면 db를 해제합니다.

cursor.close()
db.close()

 

 

참고사항

db에 변경사항이 생겼을경우에 commit을 안하시면면 에러가 발생할 수 있습니다.

에러나시면 db 다시 연결해서 사용하면 됩니다.