파이썬을 이용해 mysql에 dummy 데이터 추가하기

부하테스트 목적으로 dummy 데이터를 생성해야할 일이 종종 생긴다. 하나하나 수작업으로 insert 쿼리를 생성하는 것은 컴퓨터공학도 답지 못한 생각이다. 파이썬으로 쿼리에 필요한 랜덤 데이터를 생성하고 mysql과 연동한 후 쿼리를 전송하면된다.

import pymysql
import uuid
import secrets

# MySQL Connection 연결
con = pymysql.connect(host='', user='', password='', database='')

# Connection으로부터 Cursor 생성
cur = con.cursor()

# 랜덤 데이터 생성 및 INSERT 쿼리 실행
for i in range(1000):
    login_id = str(uuid.uuid4())[:10]
    password = secrets.token_urlsafe(16)
    sql = "INSERT INTO member (login_id, password) VALUES (%s, %s)"
    cur.execute(sql, (login_id, password))

# DB 연결 종료
con.commit()
con.close()

1000개의 쿼리를 날려주면된다. 주의할 점으로는 connect할 때 host에는 port 번호 없이 ip 주소만 작성해주면된다. 또 너무 많은 양의 쿼리를 한번에 날릴 경우 오류가 발생할 수 있으니 적절한 숫자를 조절해가며 dummy 데이터를 추가하면된다.