파이썬으로 JWT 토큰을 이용한 다중 계정 인증 시스템 구현 방법

22 Sep 2023

들어가기 전에

다중 계정 인증 시스템은 하나의 애플리케이션에서 여러 계정을 사용하여 로그인 및 인증을 처리하는 기능을 말합니다. 이번 포스트에서는 파이썬을 사용하여 JWT(JSON Web Token)을 통해 다중 계정 인증 시스템을 구현하는 방법에 대해 알아보겠습니다.

JWT란?

JWT(JSON Web Token)은 웹 애플리케이션 간 정보를 안전하게 전달하기 위해 사용되는 인증 방식 중 하나입니다. JWT는 Base64로 인코딩된 헤더, 페이로드, 서명 세 부분으로 구성되어 있습니다.

헤더(Header): 토큰의 타입과 사용된 해시 알고리즘 정보를 담고 있는 부분입니다.
페이로드(Payload): 클레임(claim) 정보를 포함한 부분입니다. 클레임은 토큰에 대한 추가적인 정보를 담고 있습니다.
서명(Signature): 토큰의 유효성 검증을 위한 서명입니다. 서명은 헤더, 페이로드, 비밀키를 사용하여 생성됩니다.

필요한 라이브러리 설치

먼저, 파이썬에서 JWT를 다루기 위해 PyJWT라이브러리를 설치해야 합니다. 아래 명령어를 사용하여 설치할 수 있습니다.

pip install PyJWT

다중 계정 인증 시스템 구현

이제 다중 계정 인증 시스템을 구현해보겠습니다. 다음은 간단한 예제 코드입니다.

import jwt

# 사용자 정보를 담은 딕셔너리
users = {
    'user1': 'password1',
    'user2': 'password2',
    # 추가적인 사용자 정보
    # ...
}

# 로그인 함수
def login(username, password):
    if username in users and users[username] == password:
        # JWT 토큰 생성
        token = jwt.encode({'username': username}, 'secret_key', algorithm='HS256')
        return token
    else:
        return None

# API 엔드포인트 함수
def api_endpoint(token):
    try:
        # 토큰 유효성 검증
        payload = jwt.decode(token, 'secret_key', algorithms=['HS256'])
        username = payload['username']
        
        # 사용자 정보에 따른 처리 로직
        # ...
        
        return "Success"
    except jwt.ExpiredSignatureError:
        return "Token expired"
    except jwt.InvalidTokenError:
        return "Invalid token"

# 로그인 및 API 엔드포인트 사용 예시
token = login('user1', 'password1')
result = api_endpoint(token)
print(result)

위 코드에서 users 딕셔너리는 사용자 정보를 담고 있습니다. 로그인 함수 login()은 지정된 사용자 이름과 비밀번호를 확인하고, 유효한 경우 JWT 토큰을 생성하여 반환합니다. API 엔드포인트 함수 api_endpoint()는 전달된 토큰의 유효성을 검증하고, 토큰에 포함된 사용자 정보에 따라 처리 로직을 수행합니다.

마무리

이번 포스트에서는 파이썬으로 JWT 토큰을 이용한 다중 계정 인증 시스템을 구현하는 방법에 대해 알아보았습니다. JWT 토큰은 인증에 추가적인 보안성을 제공하며, 다중 계정 인증 시스템 구현에 유용하게 활용될 수 있습니다.