[python] 함수의 데이터베이스 연동 방법

26 Dec 2023

python

Python은 다양한 데이터베이스와의 연동을 지원하는데, 특히 쉽게 사용할 수 있는 라이브러리들이 많이 있습니다. 이 글에서는 Python으로 데이터베이스를 연동하는 방법을 살펴보겠습니다.

SQLite 연동하기

Python에서 SQLite를 사용하기 위해서는 sqlite3 모듈을 사용할 수 있습니다. 다음 예제는 test.db 라는 SQLite 데이터베이스 파일을 만들고 연결하는 예제입니다.

import sqlite3

# 데이터베이스 연결
conn = sqlite3.connect('test.db')

# Connection 객체로부터 Cursor 객체 생성
cursor = conn.cursor()

# 테이블 생성
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)')

# 데이터 삽입 예시
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))

# 변경사항 저장
conn.commit()

# 연결 닫기
conn.close()

MySQL 연동하기

Python에서 MySQL을 사용하기 위해서는 mysql-connector-python 라이브러리를 사용할 수 있습니다. 다음 예제는 MySQL 데이터베이스에 연결하는 예제입니다.

import mysql.connector

# MySQL 연결
conn = mysql.connector.connect(
  host="localhost",
  user="username",
  password="password",
  database="dbname"
)

# Connection 객체로부터 Cursor 객체 생성
cursor = conn.cursor()

# 데이터 삽입 예시
cursor.execute("INSERT INTO users (name, age) VALUES (%s, %s)", ('Bob', 30))

# 변경사항 저장
conn.commit()

# 연결 닫기
conn.close()

MongoDB 연동하기

Python에서 MongoDB를 사용하기 위해서는 pymongo 라이브러리를 사용할 수 있습니다. 다음 예제는 MongoDB 데이터베이스에 연결하고 데이터를 삽입하는 예제입니다.

import pymongo

# MongoDB 연결
client = pymongo.MongoClient('mongodb://localhost:27017/')

# 데이터베이스와 컬렉션 선택
db = client['mydatabase']
col = db['users']

# 데이터 삽입 예시
user = {'name': 'Charlie', 'age': 35}
col.insert_one(user)

위에서 언급한 방법들을 이용하여 Python에서 쉽게 데이터베이스를 다룰 수 있습니다. 각 데이터베이스에 따라 더 다양한 작업들을 할 수 있는데, 관련 문서들을 참고하시면 도움이 될 것입니다.