[python] 데이터베이스에 데이터 저장하기
이번 포스트에서는 Python에서 데이터베이스에 데이터를 저장하는 방법에 대해 다루어보겠습니다.
SQLite 데이터베이스에 데이터 저장하기
SQLite는 경량 데이터베이스 엔진으로서 Python에서 내장된 데이터베이스 모듈을 통해 손쉽게 사용할 수 있습니다. SQLite 데이터베이스에 데이터를 저장하는 간단한 예제 코드는 다음과 같습니다.
import sqlite3
# 데이터베이스 연결
conn = sqlite3.connect('example.db')
# 커서 생성
cursor = conn.cursor()
# 테이블 생성
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
username TEXT,
email TEXT
)
''')
# 데이터 추가
cursor.execute('INSERT INTO users (username, email) VALUES (?, ?)', ('john_doe', 'john@example.com'))
# 변경사항 저장
conn.commit()
# 데이터베이스 연결 종료
conn.close()
이 예제 코드는 example.db SQLite 데이터베이스 파일을 연결하고, users 테이블을 생성한 후, 사용자 정보를 추가하는 과정을 보여줍니다.
MySQL 데이터베이스에 데이터 저장하기
MySQL 데이터베이스를 사용하기 위해서는 mysql-connector-python 라이브러리를 설치해야 합니다. 아래는 MySQL 데이터베이스에 데이터를 저장하는 간단한 예제 코드입니다.
import mysql.connector
# 데이터베이스 연결
conn = mysql.connector.connect(
host="localhost",
user="username",
password="password",
database="mydatabase"
)
# 커서 생성
cursor = conn.cursor()
# 테이블 생성
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INT AUTO_INCREMENT PRIMARY KEY,
username VARCHAR(255),
email VARCHAR(255)
)
''')
# 데이터 추가
sql = "INSERT INTO users (username, email) VALUES (%s, %s)"
val = ("john_doe", "john@example.com")
cursor.execute(sql, val)
# 변경사항 저장
conn.commit()
# 데이터베이스 연결 종료
conn.close()
이 예제 코드는 MySQL 데이터베이스에 연결하고, users 테이블을 생성한 후, 사용자 정보를 추가하는 과정을 보여줍니다.
Python을 사용하여 데이터베이스에 데이터를 저장하는 방법에 대해 간단히 살펴보았습니다. 데이터베이스의 유형에 따라 서로 다른 연결 및 데이터 저장 방법이 있으니 사용하려는 데이터베이스에 맞는 라이브러리와 문서를 참고하시기 바랍니다.