[파이썬] Flask의 ORM 사용 및 데이터베이스 연동

01 Sep 2023

python

Flask는 파이썬 기반의 경량 웹 프레임워크로, 데이터베이스 연동을 위해 ORM(Object Relational Mapping)을 사용할 수 있습니다. ORM은 객체와 데이터베이스를 매핑해주는 기술로, 객체 지향 프로그래밍과 관계형 데이터베이스의 데이터를 연결해줍니다. 이를 통해 데이터베이스 작업을 보다 쉽게 처리할 수 있습니다.

SQLAlchemy를 통한 ORM 사용

Flask에서는 SQLAlchemy를 사용하여 ORM을 구현할 수 있습니다. SQLAlchemy는 파이썬의 대표적인 ORM 라이브러리로, 다양한 데이터베이스와 호환되는 기능을 제공합니다.

먼저, Flask 애플리케이션에서 SQLAlchemy를 사용하기 위해 아래와 같이 필요한 모듈을 import 합니다.

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

다음으로, Flask 애플리케이션을 생성하고 데이터베이스 연결 정보를 설정합니다.

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = '<database_uri>'
db = SQLAlchemy(app)

위 코드에서 <database_uri>는 사용하고자 하는 데이터베이스의 연결 정보입니다. 실제 데이터베이스에 맞게 설정해야 합니다.

이제 SQLAlchemy ORM을 사용하여 데이터베이스와 상호작용할 수 있습니다. 예를 들어, 사용자 정보를 저장하는 User 모델을 정의하고, 데이터베이스에 쿼리를 실행해보겠습니다.

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))
    email = db.Column(db.String(50))

@app.route('/users')
def get_users():
    users = User.query.all()
    return jsonify(users)

위 코드에서 User 클래스는 데이터베이스의 users 테이블과 매핑됩니다. User.query.all() 메서드를 사용하여 모든 사용자를 조회한 다음, JSON 형식으로 반환합니다.

데이터베이스 마이그레이션

데이터베이스 스키마의 변경 사항을 적용하기 위해서는 마이그레이션 작업이 필요합니다. 마이그레이션은 데이터베이스 스키마를 업데이트하는 동작입니다. Flask에서는 flask-migrate 패키지를 사용하여 데이터베이스 마이그레이션을 수행할 수 있습니다.

먼저, flask-migrate를 설치합니다.

pip install flask-migrate

설치가 완료되면, 아래와 같이 필요한 모듈을 import 합니다.

from flask_migrate import Migrate

그리고, Migrate 객체를 생성하고 애플리케이션과 데이터베이스를 연결합니다.

migrate = Migrate(app, db)

이제 마이그레이션을 수행하기 위해 몇 가지 명령을 실행할 수 있습니다.

마이그레이션 스크립트 생성: flask db init
변경된 모델을 기반으로 마이그레이션 스크립트 생성: flask db migrate
스키마 변경 사항을 데이터베이스에 적용: flask db upgrade

위 명령들을 사용하여 스키마 변경 사항을 데이터베이스에 반영할 수 있습니다.

결론

Flask의 ORM과 데이터베이스 연동을 통해 데이터베이스 작업을 간편하게 처리할 수 있습니다. SQLAlchemy를 사용하여 ORM을 사용하고, flask-migrate를 통해 데이터베이스 마이그레이션을 수행함으로써 데이터베이스와의 상호작용을 원활하게 할 수 있습니다. Flask를 이용하여 웹 애플리케이션을 개발할 때 ORM과 데이터베이스 연동을 적절히 활용해보세요.