[python] SQLAlchemy-Utils를 사용하여 컬럼 암호화하기

18 Dec 2023

python

보안은 현대 소프트웨어 개발의 필수 요건 중 하나로, 데이터베이스의 중요 정보를 암호화하는 것은 매우 중요합니다. SQLAlchemy-Utils는 SQLAlchemy를 확장하여 강력한 데이터베이스 처리 기능을 제공하는 패키지인데, 여기에 포함된 암호화 기능을 사용하여 컬럼을 쉽게 암호화할 수 있습니다.

SQLAlchemy-Utils 암호화 컬럼 생성하기

우선, SQLAlchemy-Utils를 설치합니다.

pip install sqlalchemy-utils

다음으로, 암호화할 컬럼을 정의합니다. 예를 들어, 사용자의 비밀번호를 암호화하려면 다음과 같이 할 수 있습니다.

from sqlalchemy_utils import EncryptedType
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

# 암호화 키 설정
encryption_key = 'my_encryption_key'

# 베이스 클래스 선언
Base = declarative_base()

# 사용자 모델 정의
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    username = Column(String)
    # 비밀번호를 암호화된 문자열로 저장
    password = Column(EncryptedType(String, encryption_key))

위의 코드에서 EncryptedType을 사용하여 password 컬럼을 암호화했습니다.

SQLAlchemy-Utils를 사용하여 데이터베이스 연동하기

이제 SQLAlchemy를 사용하여 암호화된 컬럼을 가진 데이터베이스와 연결할 수 있습니다.

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# SQLite 메모리 데이터베이스 엔진 생성
engine = create_engine('sqlite:///:memory:')

# 데이터베이스 스키마 생성
Base.metadata.create_all(engine)

# 데이터베이스 세션 생성
Session = sessionmaker(bind=engine)
session = Session()

이제 SQLAlchemy-Utils의 EncryptedType을 사용하여 데이터베이스에 암호화된 컬럼을 추가하고 데이터를 저장할 수 있습니다.

결론

SQLAlchemy-Utils는 데이터베이스의 중요 정보를 안전하게 저장하기 위해 효과적으로 암호화할 수 있는 강력한 도구입니다. 위의 예제를 따라하면, 손쉽게 데이터베이스 컬럼을 암호화할 수 있으며, 보안 수준을 높일 수 있습니다. SQLAlchemy-Utils의 다양한 기능을 활용하여 프로젝트의 보안 요구 사항을 충족시킬 수 있습니다.

더 많은 정보를 원하시면 SQLAlchemy-Utils 공식 문서를 참조해보세요.