Files
django-mfa2/mfa/middleware.py
Mohamed ElKalioby 393a0f3ece Added Pyre
2024-01-23 14:08:46 +03:00

24 lines
666 B
Python

import time
from django.http import HttpResponseRedirect
try:
from django.core.urlresolvers import reverse # pyre-ignore[21]
except:
from django.urls import reverse
from django.conf import settings
def process(request):
next_check = request.session.get("mfa", {}).get("next_check", False)
if not next_check:
return None
now = int(time.time())
if now >= next_check:
method = request.session["mfa"]["method"]
path = request.META["PATH_INFO"]
return HttpResponseRedirect(
reverse(method + "_auth")
+ "?next=%s" % (settings.BASE_URL + path).replace("//", "/")
)
return None