Blog

Python에서 SQLAlchemy를 활용하면 데이터베이스와의 상호작용이 매우 간단해집니다. SQLAlchemy는 Python의 강력한 ORM(Object-Relational Mapping) 라이브러리로, 데이터베이스와의 연결을 관리하고, SQL 쿼리를 객체 지향적으로 다룰 수 있는 기능을 제공합니다. 이번 포스트에서는 SQLAlchemy를 사용하여 데이터베이스에 연결하는 방법을 자세히 설명하겠습니다.

SQLAlchemy란?

SQLAlchemy는 Python에서 데이터베이스와 효과적으로 상호작용할 수 있도록 설계된 라이브러리입니다. 이 라이브러리를 사용하면 SQL 쿼리를 직접 작성하는 대신, Python 객체를 통해 데이터베이스의 데이터를 쉽게 조작할 수 있습니다. SQLAlchemy는 다양한 데이터베이스 시스템을 지원하여, 이식성이 뛰어나고 여러 프로젝트에서 유용하게 활용될 수 있습니다.

환경 구성 및 설치

SQLAlchemy를 사용하기 위해 먼저 필요한 라이브러리를 설치해야 합니다. 일반적으로 다음과 같은 명령어를 통해 설치할 수 있습니다:

pip install sqlalchemy
pip install pymysql # MySQL을 사용하는 경우

사용할 데이터베이스에 따라 추가 라이브러리를 설치해야 할 수도 있습니다. 예를 들어, PostgreSQL을 사용할 경우 psycopg2가 필요합니다:

pip install psycopg2

데이터베이스 연결하기

SQLAlchemy에서 데이터베이스와 연결하기 위해서는 create_engine 함수를 사용합니다. 이 함수는 연결 문자열을 인자로 받아 데이터베이스 엔진을 생성합니다. 다음은 MySQL 데이터베이스에 연결하는 예시입니다:

from sqlalchemy import create_engine
# 데이터베이스 연결 정보 설정
db_info = {
  "user": "유저명",
  "password": "비밀번호",
  "host": "ip주소",
  "port": "포트번호",
  "database": "데이터베이스명"
}
# 연결 문자열 생성
connection_string = f"mysql+pymysql://{db_info['user']}:{db_info['password']}@{db_info['host']}:{db_info['port']}/{db_info['database']}"
# 데이터베이스 엔진 생성
engine = create_engine(connection_string, echo=True)

트랜잭션 관리

트랜잭션 관리는 데이터베이스에서 데이터의 일관성을 유지하는 데 아주 중요합니다. SQLAlchemy는 명시적으로 트랜잭션을 제어할 수 있는 방법을 제공합니다. 다음은 트랜잭션을 시작하고 관리하는 방법에 대한 예시입니다:

with engine.connect() as connection:
  # 명시적으로 트랜잭션 시작
  transaction = connection.begin()
  try:
    # 데이터베이스에 데이터 삽입 등의 작업 수행
    # ...
    # 트랜잭션 커밋
    transaction.commit()
  except Exception as e:
    # 오류 발생 시 롤백
    transaction.rollback()
    print(f"트랜잭션이 롤백되었습니다: {e}")

데이터프레임을 데이터베이스에 입출력하기

Pandas와 함께 사용하면 SQLAlchemy를 통해 데이터프레임을 간편하게 데이터베이스에 저장하거나 불러올 수 있습니다. 예를 들어, 데이터프레임을 데이터베이스 테이블에 저장하는 방법은 다음과 같습니다:

import pandas as pd
# 예시 데이터프레임 생성
data = {
  "column_1": ["값1", "값2"],
  "column_2": [1, 2]
}
df = pd.DataFrame(data)
# 데이터프레임을 데이터베이스에 저장
df.to_sql("테이블명", con=engine, if_exists="append", index=False)

여기서 if_exists 옵션은 테이블이 존재할 경우의 동작을 정의합니다.

• fail: 테이블이 존재하면 오류 발생
• replace: 기존 테이블을 삭제하고 새로운 테이블 생성
• append: 기존 테이블에 데이터 추가

데이터베이스 테이블 불러오기

데이터베이스에 저장된 테이블을 Pandas 데이터프레임으로 불러오는 방법은 다음과 같습니다:

df = pd.read_sql_table("테이블명", con=engine)

연결 종료

데이터베이스 작업이 모두 끝난 후에는 연결을 종료해야 합니다. 연결 종료는 connection.close()와 engine.dispose() 메서드를 통해 수행할 수 있습니다.

connection.close()
engine.dispose()

이러한 과정을 통해 SQLAlchemy를 활용한 데이터베이스 연결과 조작을 쉽게 수행할 수 있습니다. 데이터베이스와의 연동이 필요할 때 SQLAlchemy는 매우 유용한 도구가 될 것입니다. 이와 같은 절차를 따라가며 프로젝트를 진행해 보시길 바랍니다.

이상으로, SQLAlchemy를 활용하여 데이터베이스와의 연결을 설정하고 트랜잭션을 관리하는 방법에 대해 알아보았습니다. 데이터베이스 작업이 필요할 때마다 이 가이드를 참고하시기 바랍니다.

자주 찾는 질문 Q&A