[python] Flask-RESTful에서 맞춤형 인증 방법

30 Nov 2023

python

Flask-RESTful은 Python으로 작성된 RESTful API를 쉽게 구축할 수 있는 강력한 프레임워크입니다. 기본적으로 Flask-RESTful은 인증과 관련한 많은 기능을 제공하지만, 때로는 맞춤형 인증 방법을 사용해야 할 수도 있습니다. 이번 블로그에서는 Flask-RESTful에서 맞춤형 인증 방법을 어떻게 구현할 수 있는지 알아보겠습니다.

1. 사용자 정의 인증 데코레이터 만들기

Flask-RESTful에서 맞춤형 인증을 구현하기 위해 가장 먼저 해야 할 일은 사용자 정의 인증 데코레이터를 만드는 것입니다. 이를 통해 API 엔드포인트에 접근하기 전에 인증을 수행할 수 있습니다.

예를 들어, 다음과 같이 사용자 정의 인증 데코레이터를 만들 수 있습니다.

from flask import request
from flask_restful import Resource

def authenticate(func):
    def wrapper(*args, **kwargs):
        # 인증 로직 구현
        token = request.headers.get('Authorization')
        if token == 'my_secret_token':
            return func(*args, **kwargs)
        else:
            return {'message': 'Unauthorized'}, 401
    return wrapper

class MyResource(Resource):
    @authenticate
    def get(self):
        return {'message': 'Authenticated'}

위의 예제에서는 authenticate 데코레이터를 사용하여 인증 로직을 구현하였습니다. 클라이언트 요청의 헤더에서 Authorization 값을 가져와서 토큰을 확인하고, 인증에 성공하면 해당 API 엔드포인트가 실행됩니다. 인증에 실패한 경우 적절한 오류 응답을 반환합니다.

2. Blueprint를 사용하여 인증 적용하기

Flask-RESTful에서는 Blueprint를 사용하여 여러 개의 API 엔드포인트를 그룹화할 수 있습니다. 이를 활용하여 맞춤형 인증 방법을 적용할 수 있습니다.

예를 들어, 다음과 같이 Blueprint를 사용하여 인증을 적용할 수 있습니다.

from flask import Flask
from flask_restful import Api, Resource
from flask.blueprints import Blueprint

app = Flask(__name__)
api_bp = Blueprint('api', __name__)
api = Api(api_bp)

def authenticate(func):
    def wrapper(*args, **kwargs):
        # 인증 로직 구현
        token = request.headers.get('Authorization')
        if token == 'my_secret_token':
            return func(*args, **kwargs)
        else:
            return {'message': 'Unauthorized'}, 401
    return wrapper

class MyResource(Resource):
    @authenticate
    def get(self):
        return {'message': 'Authenticated'}

api.add_resource(MyResource, '/my-resource')

app.register_blueprint(api_bp, url_prefix='/api')

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

위의 예제에서는 Blueprint를 사용하여 API 엔드포인트를 그룹화하고, 해당 Blueprint에 맞춤형 인증 데코레이터를 적용하였습니다. 이제 /api/my-resource 엔드포인트에 접근할 때 맞춤형 인증이 수행됩니다.

결론

Flask-RESTful에서 맞춤형 인증을 구현하는 방법을 알아보았습니다. 사용자 정의 인증 데코레이터를 만들고, Blueprint를 사용하여 인증을 적용하는 방법을 살펴보았습니다. 이를 통해 Flask-RESTful을 사용하여 보다 강력하고 안전한 API를 개발할 수 있습니다.

더 자세한 내용은 아래 참고 자료를 확인해주세요.