Python - MySQL DB 생성, 삭제

By JS | Last updated: September 23, 2021

파이썬에서 pymysql을 이용하여 MySQL의 DB 생성 및 삭제하는 방법을 소개합니다.

1. pymysql 설치

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

$ python3 -m pip install PyMySQL

2. DB 생성

다음 코드는 DB를 생성하는 예제입니다.

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

import pymysql

conn = pymysql.connect(host='localhost',
                       user='testuser',
                       password='password!@',
                       charset='utf8')

with conn:
    with conn.cursor() as cur:
        cur.execute('CREATE DATABASE testdb')
        conn.commit()

MySQL 콘솔에서 다음과 같이 DB가 생성되었는지 확인할 수 있습니다.

mysql> show databases;
+--------------------+
| Database           |
+--------------------+
| information_schema |
| mydb               |
| mysql              |
| performance_schema |
| sys                |
| testdb             |
+--------------------+

3. DB 생성 확인

다음과 같이 파이썬에서 생성된 DB들을 확인할 수 있습니다.

import pymysql

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

sql = 'SHOW DATABASES'

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

Output:

('information_schema',)
('mysql',)
('performance_schema',)
('sys',)
('testdb',)

4. DB 삭제

다음과 같이 DB를 삭제할 수 있습니다.

import pymysql

conn = pymysql.connect(host='localhost',
                       user='testuser',
                       password='password!@',
                       charset='utf8')

with conn:
    with conn.cursor() as cur:
        cur.execute('DROP DATABASE testdb')
        conn.commit()

MySQL 콘솔에서 DB가 제거되었는지 확인할 수 있습니다.

mysql> show databases;
+--------------------+
| Database           |
+--------------------+
| information_schema |
| mydb               |
| mysql              |
| performance_schema |
| sys                |
+--------------------+

SHOW DATABASES SQL으로도 DB가 제거된 것을 확인할 수 있습니다.

Related Posts

댓글을 보거나 쓰려면 이 버튼을 눌러주세요.
codechachaCopyright ©2019 codechacha