Flask, SQLAlchemy를 사용한 다대다 관계 데이터 삽입

2024-07-27

Flask, SQLAlchemy를 사용한 다대다 관계 데이터 삽입

Flask는 웹 애플리케이션 개발을 위한 Python 프레임워크입니다. SQLAlchemy는 Python용 객체 관계 매핑(ORM) 라이브러리로, 데이터베이스와 상호 작용을 쉽게 합니다.

다대다 관계는 하나의 엔티티가 다른 엔티티의 여러 인스턴스와 연결될 수 있는 관계입니다. 예를 들어, 게시글과 태그 간의 관계는 다대다 관계입니다. 하나의 게시글에는 여러 태그가 있을 수 있고, 하나의 태그도 여러 게시글에 속할 수 있습니다.

이 문서에서는 Flask와 SQLAlchemy를 사용하여 다대다 관계 데이터를 삽입하는 방법을 설명합니다.

단계

다음 단계를 수행하여 Flask, SQLAlchemy를 사용하여 다대다 관계 데이터를 삽입합니다.

모델 정의

먼저, 사용할 데이터베이스 테이블을 나타내는 모델을 정의해야 합니다. 다음은 게시글과 태그를 나타내는 모델의 예입니다.

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(255))
    body = db.Column(db.Text)

    tags = db.relationship('Tag', secondary='post_tags', backref='posts')

class Tag(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255))

post_tags = db.Table('post_tags',
    db.Column('post_id', db.Integer, db.ForeignKey('post.id')),
    db.Column('tag_id', db.Integer, db.ForeignKey('tag.id'))
)

위 코드에서 Post 모델은 id, title, body 속성을 갖습니다. tags 속성은 Tag 모델과의 다대다 관계를 나타냅니다. Tag 모델은 id와 name 속성을 갖습니다.

post_tags 테이블은 다대다 관계를 위한 연결 테이블입니다. 이 테이블에는 post_id와 tag_id 열이 포함되어 각 게시글과 태그를 연결합니다.

데이터 삽입

다음은 게시글과 태그를 만들고 다대다 관계를 설정하는 방법의 예입니다.

post = Post(title="My First Post", body="This is my first blog post.")

tag1 = Tag(name="Python")
tag2 = Tag(name="Flask")
tag3 = Tag(name="SQLAlchemy")

post.tags.append(tag1)
post.tags.append(tag2)
post.tags.append(tag3)

db.session.add(post)
db.session.commit()

위 코드에서 먼저 Post 모델의 새 인스턴스를 만듭니다. 그런 다음 title과 body 속성을 설정합니다.

다음으로 Tag 모델의 세 개의 새 인스턴스를 만듭니다. 각 인스턴스에는 name 속성을 설정합니다.

마지막으로 append() 메서드를 사용하여 게시글 인스턴스의 tags 목록에 태그 인스턴스를 추가합니다.

add() 메서드를 사용하여 데이터베이스 세션에 게시글 인스턴스를 추가하고 commit() 메서드를 사용하여 변경 사항을 커밋합니다.

관계 조회

다음은 게시글과 관련된 모든 태그를 조회하는 방법의 예입니다.

post = Post.query.get(1)

for tag in post.tags:
    print(tag.name)

위 코드에서 먼저 query.get() 메서드를 사용하여 ID가 1인 게시글 인스턴스를 검색합니다.

그런 다음 tags 속성을 사용하여 게시글과 관련된 모든 태그를 반복합니다. 마지막으로 각 태그의 name 속성을 출력합니다.

관계 삭제

post = Post.query.get(1)

tag = post.tags.pop()

Flask, SQLAlchemy를 사용한 다대다 관계 데이터 삽입 예제

from flask import Flask, render_template
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

# 데이터베이스 설정
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///example.db'
db = SQLAlchemy(app)

# 모델 정의
class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(255))
    body = db.Column(db.Text)

    tags = db.relationship('Tag', secondary='post_tags', backref='posts')

class Tag(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255))

post_tags = db.Table('post_tags',
    db.Column('post_id', db.Integer, db.ForeignKey('post.id')),
    db.Column('tag_id', db.Integer, db.ForeignKey('tag.id'))
)

# 데이터 삽입
@app.route('/')
def index():
    # 게시글과 태그 만들기
    post = Post(title="첫 번째 게시글", body="안녕하세요! 이것은 제 첫 번째 Flask 블로그 게시글입니다.")

    tag1 = Tag(name="Python")
    tag2 = Tag(name="Flask")
    tag3 = Tag(name="SQLAlchemy")

    # 게시글에 태그 추가
    post.tags.append(tag1)
    post.tags.append(tag2)
    post.tags.append(tag3)

    # 데이터베이스에 저장
    db.session.add(post)
    db.session.commit()

    # 모든 게시글 조회
    posts = Post.query.all()
    return render_template('index.html', posts=posts)

if __name__ == '__main__':
    app.run(debug=True)

설명

위 코드는 다음과 같은 작업을 수행합니다.

Flask 애플리케이션 만들기: Flask 클래스를 사용하여 Flask 애플리케이션 인스턴스를 만듭니다.
데이터베이스 설정: SQLALCHEMY_DATABASE_URI 설정 변수를 사용하여 SQLite 데이터베이스의 위치를 설정합니다.
SQLAlchemy 초기화: SQLAlchemy 객체를 만들고 Flask 애플리케이션에 연결합니다.
모델 정의: Post 및 Tag 모델 클래스를 정의합니다. 이러한 클래스는 데이터베이스 테이블과 해당 속성을 나타냅니다.
다대다 관계 정의: post_tags 연결 테이블을 사용하여 Post 및 Tag 모델 간의 다대다 관계를 정의합니다.
데이터 삽입: index 함수에서 Post 및 Tag 모델의 인스턴스를 만들고 데이터베이스에 저장합니다.
모든 게시글 조회: index 함수에서 Post.query.all() 메서드를 사용하여 모든 게시글을 조회하고 index.html 템플릿에 렌더링합니다.

사용 방법

이 코드를 사용하려면 다음 단계를 수행하십시오.

코드를 파일에 저장합니다 (예: example.py).
다음 명령을 사용하여 가상 환경을 활성화합니다.

source venv/bin/activate

다음 명령을 사용하여 Flask 애플리케이션을 실행합니다.

flask run

웹 브라우저에서 http://localhost:5000/ 주소를 엽니다.

브라우저에 "첫 번째 게시글"이라는 제목의 게시글이 표시됩니다. 게시글에는 "Python", "Flask", "SQLAlchemy" 태그가 있습니다.

Flask, SQLAlchemy를 사용한 다대다 관계 데이터 삽입: 대체 방법

tags 속성 직접 사용

다음은 tags 속성을 직접 사용하여 게시글에 태그를 추가하는 방법입니다.

post = Post(title="첫 번째 게시글", body="안녕하세요! 이것은 제 첫 번째 Flask 블로그 게시글입니다.")

tag1 = Tag(name="Python")
tag2 = Tag(name="Flask")
tag3 = Tag(name="SQLAlchemy")

post.tags = [tag1, tag2, tag3]

db.session.add(post)
db.session.commit()

위 코드는 append() 메서드를 사용하는 것보다 간결합니다.

session.add_all() 사용

다음은 session.add_all() 메서드를 사용하여 게시글과 태그를 동시에 데이터베이스에 추가하는 방법입니다.

post = Post(title="첫 번째 게시글", body="안녕하세요! 이것은 제 첫 번째 Flask 블로그 게시글입니다.")

tag1 = Tag(name="Python")
tag2 = Tag(name="Flask")
tag3 = Tag(name="SQLAlchemy")

db.session.add_all([post, tag1, tag2, tag3])
db.session.commit()

위 코드는 add() 메서드를 여러 번 호출하는 것보다 효율적일 수 있습니다.

flush= 옵션 사용

다음은 flush= 옵션을 사용하여 add() 메서드를 호출할 때 데이터베이스에 즉시 변경 사항을 저장하는 방법입니다.

post = Post(title="첫 번째 게시글", body="안녕하세요! 이것은 제 첫 번째 Flask 블로그 게시글입니다.")

tag1 = Tag(name="Python")
tag2 = Tag(name="Flask")
tag3 = Tag(name="SQLAlchemy")

post.tags.append(tag1)
post.tags.append(tag2)
post.tags.append(tag3)

db.session.add(post, flush=True)

위 코드는 commit() 메서드를 호출하지 않아도 데이터베이스에 변경 사항을 저장합니다.

insert() 메서드 사용

다음은 insert() 메서드를 사용하여 post_tags 연결 테이블에 직접 레코드를 삽입하는 방법입니다.

post = Post.query.get(1)
tag1 = Tag.query.get(1)
tag2 = Tag.query.get(2)
tag3 = Tag.query.get(3)

db.session.execute(
    post_tags.insert().values(post_id=post.id, tag_id=tag1.id),
    post_tags.insert().values(post_id=post.id, tag_id=tag2.id),
    post_tags.insert().values(post_id=post.id, tag_id=tag3.id)
)
db.session.commit()

위 코드는 append() 메서드를 사용하는 것보다 더 명시적이지만 더 복잡할 수 있습니다.

주의 사항

위에 제시된 대체 방법은 모두 동일한 결과를 생성합니다. 그러나 특정 상황에 따라 한 가지 방법이 다른 방법보다 더 적합할 수 있습니다. 예를 들어, 데이터를 자주 삽입하는 경우 session.add_all() 메서드를 사용하는 것이 더 효율적일 수 있습니다.

python flask sqlalchemy