파이썬과 JWT를 이용한 API 토큰 기반의 접근 제어 방법

22 Sep 2023

API는 현대 웹 애플리케이션에서 필수적인 요소 중 하나이며, 보안 측면에서 접근 제어는 매우 중요합니다. JWT( JSON Web Token)는 API 인증과 인가를 위한 표준 방법 중 하나로, 이를 파이썬과 함께 사용하면 강력한 접근 제어 메커니즘을 구축할 수 있습니다.

JWT란 무엇인가요?

JWT는 웹 표준으로 정의된 토큰 기반의 인증과 인가 방식입니다. 이는 사용자의 신원을 증명하고 권한을 부여하여 안전한 API 접근을 가능하게 합니다. JWT는 세 부분으로 구성되어 있으며, 각 부분은 점으로 구분됩니다.

헤더(Header): 토큰의 유형과 해싱 알고리즘 정보가 포함됩니다.
내용(Payload): 토큰에 포함할 클레임(Claim) 정보, 사용자 아이디, 만료 시간 등이 포함됩니다.
서명(Signature): 헤더와 내용을 인코딩하여 생성된 서명 부분입니다.

파이썬에서 JWT 사용하기

파이썬에서 JWT를 사용하기 위해서는 먼저 JWT 라이브러리를 설치해야 합니다. 다음과 같이 pip를 사용하여 설치할 수 있습니다.

pip install PyJWT

이제 파이썬에서 JWT를 사용하여 API 토큰 기반의 접근 제어를 구현해보겠습니다.

import jwt

# 비밀 키 설정
secret_key = "my_secret_key"

# 토큰 생성
def generate_token(user_id):
    payload = {"user_id": user_id}
    token = jwt.encode(payload, secret_key)

    return token

# 토큰 검증
def verify_token(token):
    try:
        payload = jwt.decode(token, secret_key)
        return payload["user_id"]
    except jwt.DecodeError:
        return None

위 예제에서는 비밀 키를 my_secret_key로 설정한 후, generate_token 함수를 통해 토큰을 생성합니다. verify_token 함수를 사용하여 토큰을 검증하고, 유효한 경우 사용자 아이디를 반환합니다.

접근 제어 방법

API에 접근 제어를 적용하기 위해 클라이언트는 먼저 토큰을 발급받아야 합니다. 클라이언트는 발급받은 토큰을 API 호출 시 HTTP 헤더에 포함시켜 전송합니다. 서버는 헤더에서 토큰을 추출하고, 이를 검증하여 접근 권한을 확인합니다.

from flask import Flask, request

app = Flask(__name__)

@app.route("/api/data")
def get_data():
    token = request.headers.get("Authorization")

    if token:
        user_id = verify_token(token)

        if user_id:
            # 유저 아이디에 따라 데이터를 반환하거나 작업 수행
            return "Success"

    return "Unauthorized", 401

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

위 예제는 Flask를 사용하여 API 엔드포인트(/api/data)를 구현하는 방법을 보여줍니다. 클라이언트는 API 호출 시 Authorization 헤더에 발급받은 토큰을 넣어 전송하며, 서버는 verify_token 함수를 사용하여 토큰을 검증합니다. 토큰이 유효하면 요청을 처리하고, 그렇지 않으면 인가되지 않은 상태로 응답합니다.

결론

이렇게 파이썬과 JWT를 이용하여 API 토큰 기반의 접근 제어 방법을 구현할 수 있습니다. JWT를 사용하면 사용자의 신원을 증명하고 권한을 부여하여 API 보안을 강화할 수 있으며, 파이썬과의 통합은 매우 간단하고 효과적입니다. 앞으로 API 개발 시 JWT를 활용하여 보안을 강화해보세요!

#API #보안