Python - MySQL 데이터 추가, 삭제, 업데이트

파이썬에서 pymysql을 이용하여 MySQL에 데이터를 추가(INSERT), 업데이트(UPDATE), 삭제(DELETE)하는 방법을 소개합니다.

MySQL의 다른 글들도 참고해주세요.

1. pymysql 설치

pymysql는 파이썬의 MySQL 라이브러리입니다. pip 명령어로 설치할 수 있습니다.

$ python3 -m pip install PyMySQL

2. 데이터 추가(INSERT)

다음 코드는 데이터를 추가하는 예제입니다.

MySQL 서버가 설치되어있고, 계정이 모두 생성된 상태에서 connect()로 서버에 접속합니다. cursor를 통해 SQL을 실행하여 데이터를 추가합니다. with 구문을 사용하기 때문에 cursor, conn의 close()를 호출할 필요가 없습니다. with 구문 종료 시점에 자동으로 호출됩니다.

import pymysql

conn = pymysql.connect(host='localhost',
                       user='testuser',
                       password='Fktm068**',
                       db='testdb',
                       charset='utf8')

sql = "INSERT INTO user (name, email) VALUES (%s, %s)"

with conn:
    with conn.cursor() as cur:
        cur.execute(sql, ('Jaehee', 'jaehee@example.com'))
        cur.execute(sql, ('Jeongeun', 'jeongeun@example.com'))
        conn.commit()

MySQL 콘솔에서 다음과 같이 데이터가 추가되었는지 확인할 수 있습니다.

mysql> select * from testdb.user;
+----+----------+----------------------+
| id | name     | email                |
+----+----------+----------------------+
|  1 | Jaehee   | jaehee@example.com   |
|  2 | Jeongeun | jeongeun@example.com |
+----+----------+----------------------+

3. 데이터 추가 확인

다음과 같이 파이썬에서 SELECT SQL을 실행하여 추가된 데이터를 확인할 수 있습니다. 하기 SQL은 user table에 있는 모든 데이터를 name으로 정렬하여 가져옵니다.

import pymysql

conn = pymysql.connect(host='localhost',
                       user='testuser',
                       password='Fktm068**',
                       db='testdb',
                       charset='utf8')

sql = "SELECT * FROM user ORDER BY name"

with conn:
    with conn.cursor() as cur:
        cur.execute(sql)
        result = cur.fetchall()
        for data in result:
            print(data)

Output:

(1, 'Jaehee', 'jaehee@example.com')
(2, 'Jeongeun', 'jeongeun@example.com')

4. 데이터 업데이트(UPDATE)

다음과 같이 기존에 추가된 항목들의 데이터를 업데이트할 수 있습니다. WHERE를 이용하여 변경하려는 데이터에 대한 조건을 설정합니다. email이 jahee@example.com인 데이터를 찾고, email을 jahee@test.com으로 변경합니다.

import pymysql

conn = pymysql.connect(host='localhost',
                       user='testuser',
                       password='Fktm068**',
                       db='testdb',
                       charset='utf8')

sql = "UPDATE user SET email = %s WHERE email = %s"

with conn:
    with conn.cursor() as cur:
        cur.execute(sql, ('jaehee@test.com', 'jaehee@example.com'))
        conn.commit()

MySQL 콘솔에서 다음과 같이 데이터가 업데이트되었는지 확인할 수 있습니다.

mysql> select * from testdb.user;
+----+----------+----------------------+
| id | name     | email                |
+----+----------+----------------------+
|  1 | Jaehee   | jaehee@test.com      |
|  2 | Jeongeun | jeongeun@example.com |
+----+----------+----------------------+

SELECT SQL으로 데이터가 업데이트되었는지 확인할 수도 있습니다.

5. 데이터 삭제(DELETE)

다음과 같은 SQL로 데이터를 삭제할 수 있습니다. WHERE를 이용하여 조건에 맞는 데이터만 삭제할 수 있습니다. cur.rowcount는 삭제된 항목들의 개수입니다.

import pymysql

conn = pymysql.connect(host='localhost',
                       user='testuser',
                       password='Fktm068**',
                       db='testdb',
                       charset='utf8')

sql = "DELETE FROM user WHERE name = %s"

with conn:
    with conn.cursor() as cur:
        cur.execute(sql, ('Jaehee'))
        conn.commit()
        print(cur.rowcount, "rows deleted")

Output:

1 rows deleted

MySQL 콘솔에서 다음과 같이 데이터가 삭제되었는지 확인할 수 있습니다.

mysql> select * from testdb.user;
+----+----------+----------------------+
| id | name     | email                |
+----+----------+----------------------+
|  2 | Jeongeun | jeongeun@example.com |
+----+----------+----------------------+
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha