jumpserver/apps/authentication/middleware.py

18 lines
533 B
Python
Raw Normal View History

2021-08-26 07:01:43 +00:00
from django.shortcuts import redirect
class MFAMiddleware:
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
response = self.get_response(request)
white_urls = ['login/mfa', 'mfa/select', 'jsi18n/', '/static/']
for url in white_urls:
if request.path.find(url) > -1:
return response
2021-08-26 07:01:43 +00:00
if request.session.get('auth_mfa_required'):
return redirect('authentication:login-mfa')
2021-08-26 07:01:43 +00:00
return response