[Python] pymysql 사용하기

[Python] pymysql 사용하기

pymysql을 db와 연동 후 select 하는 과정을 알아보자.

pymysql 연동

먼저 pymysql 라이브러리를 import 해준다.

import pymysql

DB와 Python 연동을 위해 pymysql.connect() 함수에 DB 정보를 입력 후, 연결한다.

conn = pymysql.connect(
        host = db_host,
        user = db_user,
        password= db_passwd,
        port = 3306,
        db = 'test_db',
        charset='utf8'
)

DB 정보를 코드에 직접 사용하지말고, 분리해서 넣는게 좋다.

[Python] 민감 정보 분리해서 사용하기 (config 파일 분리)

 

cursor 생성

cursor 클래스는 실제적으로 DB의 sql 구문을 실행시키고, 조회된 결과를 가져온다.

그렇기 때문에, 연동된 DB에 조회를 하기 위해선 cursor를 먼저 생성해준다.

cur = conn.cursor()
with conn.cursor() as cur

 

DB 조회

execute()를 이용해 쿼리를 실행하고, 실행한 결과값을 fetchone()을 이용해 result 변수에 저장한다.

fetchone()을 사용하는 경우, 데이터 1개(하나의 row)를 반환한다.

sql = 'SELECT id FROM member'
cur.execute(sql)
result =cursor.fetchone()

execute()	sql 쿼리를 실행시키기 위해 호출
fetchone()	조회된 결과로부터 데이터 1개를 반환
fetchall()	모든 데이터를 한번에 클라이언트로 가져와 사용
fetchmany()	여러개의 데이터를 반환

 

DB 연결 닫기

원하는 결과값을 얻었으면, Connaction 객체의 close() 메서드를 이용하여 DB 연결을 닫아준다. 

conn.close()

 

파라미터를 받아서 DB에 전달하고 싶은 경우는 다음 포스팅을 참고

[Pymysql] 파라미터 전달