python, postgres - psycopg2 사용법

테이블 대규모 업데이트를 해야하는 상황이였는데 아직 순수 쿼리로만 업데이트를 하기엔 힘들어서

파이썬을 이용해 sql업데이트를 했습니다. 그때 사용한 파이썬 라이브러리가 psycopg2였습니다. 라이브러리 사용법과 구문에 대해서 간략하게 기록해 놓았습니다.

 

목차

1. 데이터베이스 연결하기

2. table 생성

3. 회원정보 입력

4. 회원정보 조회

5. connection close

 

* jupyter notebook을 기준으로 작성했습니다.

 

psycopg2 설치

!(느낌표)를 사용하면 cmd창 사용이 가능합니다.

 

import후 ip,dbname, user,password, port입력

1. 데이터베이스 연결하기

 

- psycopg2.connect : 데이터베이스명, 사용자 ID, PW를 입력하여 DB연결

- conn.cursor() : 데이터베이스와 통신할 수 있는 cursor생성

 

2. table 생성

 

회원 테이블 생성

회원테이블을 생성합니다. 실제로 회원테이블에서는 email, id 등 여러컬럼을 생성해야되지만 예제이기 때문에 간략하게 생성했습니다.

 

commit() 메서드를 호출해 실제로 데이터베이스에 값을 반영시킵니다.

 

3. 회원정보 입력

회원 정보 입력

4. 회원정보 조회

결과 조회

명령어	설명
fetchall()	전체 값 출력
fetchmany(size=)	지정된 사이즈(ex. size=4) 갯수만큼 출력
fetchone()	한개의 결과 값만 출력

fetch로 return한 값은 cur객체에서 값이 사라집니다. (중복된 결과를 가져오지 못함)

 

5. connection close

 

connection close

사용이 완료된 인스턴스들 해제 해주는 것을 권고합니다.

 

https://www.psycopg.org/docs/usage.html

Basic module usage — Psycopg 2.9.1 documentation

- psycopy2 document에서 참고