파이썬과 JWT를 사용한 간편한 웹 API 보안 구현하기

22 Sep 2023

소개

웹 어플리케이션을 개발할 때 보안은 매우 중요한 요소입니다. 특히 웹 API를 사용하는 경우, 사용자 인증과 권한 부여를 효과적으로 처리해야 합니다. 이를 위해 파이썬과 JWT(JSON Web Token)을 사용하여 간편한 웹 API 보안을 구현해보겠습니다.

JWT란?

JWT는 웹 애플리케이션 간에 안전하게 정보를 교환하기 위한 토큰 기반의 인증 방식입니다. JWT는 세 부분으로 이루어져 있으며, Header, Payload, Signature로 구성됩니다.

Header: 토큰의 유형과 암호화 알고리즘 등의 정보를 포함합니다.
Payload: 토큰에 담을 추가적인 정보를 포함합니다. 예를 들어 사용자 식별자, 권한 등을 포함할 수 있습니다.
Signature: Header와 Payload를 인코딩한 후 비밀 키로 서명한 값입니다. 토큰이 변조되지 않았는지 검증하는 역할을 합니다.

파이썬으로 JWT 구현하기

패키지 설치 먼저, pyjwt 패키지를 설치해야 합니다. 아래 명령어를 사용하여 설치할 수 있습니다.

pip install pyjwt

토큰 생성 JWT를 사용하기 위해서는 비밀 키가 필요합니다. 비밀 키는 서버에서 관리하고 사용됩니다. 아래는 토큰을 생성하는 코드입니다.

import jwt

# 비밀 키 설정
SECRET_KEY = 'your-secret-key'

# 토큰 생성
def generate_token(user_id):
    payload = {"user_id": user_id}
    token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')
    return token

토큰 검증 API 요청이 들어왔을 때, 토큰을 검증하여 사용자 인증 및 권한 부여를 처리해야 합니다. 아래는 토큰을 검증하는 코드입니다.

def verify_token(token):
    try:
        payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])
        user_id = payload.get('user_id')
        # 사용자 인증 및 권한 처리 로직
    except jwt.ExpiredSignatureError:
        # 토큰 유효 기간이 만료된 경우
        # 예외 처리 로직
    except jwt.InvalidTokenError:
        # 잘못된 토큰인 경우
       # 예외 처리 로직

마무리

파이썬과 JWT를 사용하여 웹 API 보안을 구현하는 방법을 알아보았습니다. JWT를 통해 간단하면서도 효과적으로 사용자 인증과 권한 부여를 처리할 수 있습니다. 보안이 중요한 웹 어플리케이션 개발 시, JWT를 사용하여 안전한 인증 방식을 구현해보세요.

#웹API #보안