[파이썬] SQLAlchemy 커스텀 쿼리 구성

SQLAlchemy는 파이썬에서 데이터베이스와 상호작용하는 데 사용되는 강력하고 유연한 ORM(Object-Relational Mapping) 라이브러리입니다. 기본 CRUD(Create, Read, Update, Delete) 작업뿐만 아니라 고급 쿼리를 수행하는 데도 사용할 수 있습니다. 이번 블로그 포스트에서는 SQLAlchemy를 사용하여 커스텀 쿼리를 구성하는 방법에 대해 알아보겠습니다.

1. SQLAlchemy ORM을 사용한 기본 쿼리

SQLAlchemy ORM을 사용하면 테이블과 객체 간의 매핑을 통해 데이터베이스 작업을 수행할 수 있습니다. 기본적인 CRUD 작업은 다음과 같이 간단하게 수행할 수 있습니다.

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# 데이터베이스 연결 설정
engine = create_engine('sqlite:///mydatabase.db')
Session = sessionmaker(bind=engine)
session = Session()

# 새로운 레코드 생성
new_record = MyTable(column1='value1', column2='value2')
session.add(new_record)
session.commit()

# 레코드 조회
result = session.query(MyTable).filter_by(column1='value1').first()
print(result.column2)

# 레코드 업데이트
result.column1 = 'new_value'
session.commit()

# 레코드 삭제
session.delete(result)
session.commit()

위의 코드에서 MyTable은 테이블을 나타내는 클래스입니다. column1column2는 테이블의 열 이름입니다. 이러한 기본 쿼리 작업을 SQLAlchemy ORM을 통해 간단하게 수행할 수 있습니다.

2. SQLAlchemy 커스텀 쿼리 작성

SQLAlchemy에서는 ORM을 사용하지 않고 직접 SQL 쿼리를 작성하여 고급 데이터베이스 작업을 수행할 수도 있습니다. session 객체의 execute() 메서드를 사용하여 커스텀 쿼리를 실행할 수 있습니다.

from sqlalchemy import text

# 커스텀 쿼리 실행
query = text("SELECT * FROM mytable WHERE column1=:value")
result = session.execute(query, {'value': 'some_value'})

# 결과 처리
for row in result:
    print(row)

위의 코드에서는 text() 함수를 사용하여 커스텀 쿼리를 작성하고, session.execute() 메서드를 통해 실행합니다. 실행 결과는 result 객체로 반환되며, 이를 순회하면서 결과를 처리할 수 있습니다.

3. SQLAlchemy 표현식 사용하기

SQLAlchemy는 SQL을 구성하는 데 사용할 수 있는 표현식(Expression)을 제공합니다. 이러한 표현식을 사용하면 SQL 코드를 더 직관적으로 작성할 수 있습니다. 다음은 SQLAlchemy 표현식을 사용하여 커스텀 쿼리를 작성하는 예입니다.

from sqlalchemy import select

# 표현식 사용
query = select([MyTable.column1, MyTable.column2]).where(MyTable.column1 == 'value1')
result = session.execute(query)

# 결과 처리
for row in result:
    print(row)

위의 코드에서는 select() 함수를 사용하여 커스텀 쿼리를 작성하고, 표현식을 사용하여 원하는 열을 선택하고 필터링합니다. 이를 통해 더 유연하고 읽기 쉬운 쿼리를 작성할 수 있습니다.

결론

SQLAlchemy를 사용하면 ORM을 통해 기본적인 CRUD 작업을 간편하게 수행할 수 있습니다. 그러나 필요한 경우 커스텀 쿼리를 작성하여 고급 데이터베이스 작업을 수행할 수도 있습니다. SQLAlchemy의 표현식을 잘 활용하면 SQL 코드를 더 직관적으로 작성할 수 있으며, 데이터베이스 상호작용을 더욱 효율적으로 수행할 수 있습니다.