이것저것 코드/코드기록
# Flask Decorator - 플라스크 로그인 여부 판단 데코레이터
octo54
2022. 1. 21. 14:32
반응형
Decorator Example Code
Overview
- 백엔드 서버에서 특정 URL 에 접속을 할 때 잘못된 경로 접속 또는 로그인 유무, Not Found Page 등 중간에서 처리를 해야할 경우가 있다. 이때 파이썬에서 제공하는 Deacorator를 사용하여 구현할 수 있으며, 간단한 코드로 정리를 했다.
참고 사이트 : Flask Dacorator
Login Require Decorator
def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if session.get('유저 아이디', None) is None:
return redirect(url_for('리다이렉트 URL 또는 라우팅 함수 이름'))
return f(*args, **kwargs)
return decorated_function
- 데코레이터로 쓰일 함수를 정의한다.
login_required
라는 함수를 정의하고 매개변수로는 데코레이터가 쓰이는 함수를 받아 준다. login_required
함수는 매개변수로 받은 함수f
를 실행하기 전에 처리하는 함수decorated_function
을 정의하고 이 함수는 , 함수f
를return
한다.- 마지막으로
login_required
는decorated_function
을return
한다.