[python] SQLAlchemy-Utils를 사용하여 데이터베이스 권한 설정하기

18 Dec 2023

python

데이터베이스는 보안상 중요한 역할을 합니다. 애플리케이션을 개발하고 운영할 때, 데이터베이스에 액세스하는 권한을 세밀하게 관리하는 것이 중요합니다. Python의 SQLAlchemy-Utils 라이브러리는 데이터베이스의 권한을 설정하고 관리하는 데 도움이 되는 다양한 기능을 제공합니다.

SQLAlchemy-Utils 라이브러리란?

SQLAlchemy-Utils는 SQLAlchemy의 확장 라이브러리로, 데이터베이스와 관련된 여러 유틸리티 기능을 제공합니다. 이를 사용하면 데이터베이스 모델을 보다 쉽게 정의하고 관리할 수 있습니다. 또한 권한 관리, 유효성 검사, 암호화 등의 고급 기능을 제공하여 데이터베이스를 보다 안전하게 관리할 수 있습니다.

데이터베이스 권한 설정하기

SQLAlchemy-Utils를 사용하여 데이터베이스 권한을 설정하려면, 다음의 단계를 따릅니다.

1. Role 모델 정의하기

from sqlalchemy_utils import ChoiceType, PasswordType
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String

Base = declarative_base()

class Role(Base):
    __tablename__ = 'roles'

    id = Column(Integer, primary_key=True)
    name = Column(ChoiceType({
        ('admin', 'Administrator'),
        ('editor', 'Editor'),
        ('viewer', 'Viewer')
    }))
    password = Column(PasswordType(
        schemes=[
            'pbkdf2_sha512',
            'md5_crypt'
        ]
    ))

위 코드에서는 Role 모델을 정의하고, 각 역할에 대한 이름과 암호를 저장하기 위한 칼럼을 정의합니다.

2. 사용자 모델에 권한 항목 추가

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    username = Column(String(50), unique=True)
    role_id = Column(Integer, ForeignKey('roles.id'))
    password = Column(PasswordType(
        schemes=[
            'pbkdf2_sha512',
            'md5_crypt'
        ]
    ))

사용자 모델에는 권한을 나타내는 role_id를 추가합니다.

3. 데이터베이스에 권한 저장

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///example.db')
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()

admin_role = Role(name='admin', password='admin_pass')
editor_role = Role(name='editor', password='editor_pass')
viewer_role = Role(name='viewer', password='viewer_pass')

session.add_all([admin_role, editor_role, viewer_role])
session.commit()

위 코드에서는 각 역할에 대한 데이터를 데이터베이스에 저장합니다.

4. 데이터베이스 권한 확인

user = session.query(User).filter_by(username='user1').first()
print(user.role.name)

사용자의 권한을 확인하기 위해서는 위와 같이 코드를 작성할 수 있습니다.

마치며

SQLAlchemy-Utils를 사용하여 데이터베이스의 권한을 설정하는 방법에 대해 알아보았습니다. 이를 통해 데이터베이스 보안을 강화하고 권한을 세밀하게 관리하는 데 도움이 될 것입니다.

더 많은 내용을 알고 싶다면 SQLAlchemy-Utils 공식 문서를 참고하세요.