Python - MySQL 테이블 생성, 삭제

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

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

1. pymysql 설치

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

$ python3 -m pip install PyMySQL

2. 테이블 생성

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 = '''CREATE TABLE user (
        id int(11) NOT NULL AUTO_INCREMENT PRIMARY KEY,
        name varchar(255),
        email varchar(255)
        )
        '''

with conn:
    with conn.cursor() as cur:
        cur.execute(sql)
        conn.commit()

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

mysql> show tables from testdb;
+------------------+
| Tables_in_testdb |
+------------------+
| user             |
+------------------+

다음 명령어로 테이블의 columns들도 확인할 수 있습니다.

mysql> show columns from user from testdb;
+-------+--------------+------+-----+---------+----------------+
| Field | Type         | Null | Key | Default | Extra          |
+-------+--------------+------+-----+---------+----------------+
| id    | int          | NO   | PRI | NULL    | auto_increment |
| name  | varchar(255) | YES  |     | NULL    |                |
| email | varchar(255) | YES  |     | NULL    |                |
+-------+--------------+------+-----+---------+----------------+

3. 테이블 생성 확인

다음과 같이 파이썬에서 SQL을 실행하여 생성된 테이블을 확인할 수 있습니다.

import pymysql

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

sql = 'SHOW TABLES'

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

Output:

('user',)

4. 테이블 삭제

다음과 같이 테이블을 삭제할 수 있습니다. SQL 구문에서 IF EXISTS는 테이블이 존재하는 경우만 삭제를 시도한다는 의미입니다.

SQL에서 IF EXISTS를 제거하면 SQL이 수행될 때 테이블 삭제를 시도하며, 존재하지 않을 때는 에러가 발생합니다.

import pymysql

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

sql = "DROP TABLE IF EXISTS user"

with conn:
    with conn.cursor() as cur:
        cur.execute(sql)
    conn.commit()
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha