[Python] MySQL

PyMySQL 설치

pip install PyMySQL

~/example.py

import pymysql

def connect_mysql():
    return pymysql.connect(
        host='vote-db.example.com',
        user='ubuntu',
        password='123456',
        db='vote_dev',
        charset='utf8'
    )

def select_users():
    conn = connect_mysql()
    try:
        with conn.cursor() as curs:
            sql = "select * from user"
            curs.execute(sql)
            return curs.fetchall()
    finally:
        conn.close()

def insert_user():
    conn = connect_mysql()
    try:
        with conn.cursor() as curs:
            sql = "insert into user (id, password, role) values (%s, %s, %s)"
            curs.execute(sql, ('john_test', '1234', 'USER'))
        conn.commit()
    finally:
        conn.close()

def update_user():
    conn = connect_mysql()
    try:
        with conn.cursor() as curs:
            sql = "update user set password = %s where id = %s"
            curs.execute(sql, ('0000', 'john_test'))
        conn.commit()
    finally:
        conn.close()

def delete_user():
    conn = connect_mysql()
    try:
        with conn.cursor() as curs:
            sql = "delete from user where id = %s"
            curs.execute(sql, 'john_test')
        conn.commit()
    finally:
        conn.close()