mirror of https://github.com/snachodog/mybuddy.git
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
from os import getenv
|
|
from time import time
|
|
|
|
import pytz
|
|
|
|
from django.conf import settings
|
|
from django.utils import timezone, translation
|
|
from django.contrib.auth.middleware import RemoteUserMiddleware
|
|
|
|
|
|
class UserLanguageMiddleware:
|
|
"""
|
|
Customizes settings based on user language setting.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
user = request.user
|
|
if hasattr(user, "settings") and user.settings.language:
|
|
language = user.settings.language
|
|
elif request.LANGUAGE_CODE:
|
|
language = request.LANGUAGE_CODE
|
|
else:
|
|
language = settings.LANGUAGE_CODE
|
|
|
|
if language:
|
|
# Set the language before generating the response.
|
|
translation.activate(language)
|
|
|
|
response = self.get_response(request)
|
|
|
|
# Deactivate the translation before the response is sent so it not
|
|
# reused in other threads.
|
|
translation.deactivate()
|
|
|
|
return response
|
|
|
|
|
|
class UserTimezoneMiddleware:
|
|
"""
|
|
Sets the timezone based on a user specific setting. This middleware must run after
|
|
`django.contrib.auth.middleware.AuthenticationMiddleware` because it uses the
|
|
request.user object.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
user = request.user
|
|
if hasattr(user, "settings") and user.settings.timezone:
|
|
try:
|
|
timezone.activate(pytz.timezone(user.settings.timezone))
|
|
except pytz.UnknownTimeZoneError:
|
|
pass
|
|
return self.get_response(request)
|
|
|
|
|
|
class RollingSessionMiddleware:
|
|
"""
|
|
Periodically resets the session expiry for existing sessions.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
if request.session.keys():
|
|
session_refresh = request.session.get("session_refresh")
|
|
if session_refresh:
|
|
try:
|
|
delta = int(time()) - session_refresh
|
|
except (ValueError, TypeError):
|
|
delta = settings.ROLLING_SESSION_REFRESH + 1
|
|
if delta > settings.ROLLING_SESSION_REFRESH:
|
|
request.session["session_refresh"] = int(time())
|
|
request.session.set_expiry(settings.SESSION_COOKIE_AGE)
|
|
else:
|
|
request.session["session_refresh"] = int(time())
|
|
return self.get_response(request)
|
|
|
|
|
|
class CustomRemoteUser(RemoteUserMiddleware):
|
|
"""
|
|
Middleware used for remote authentication when `REVERSE_PROXY_AUTH` is True.
|
|
"""
|
|
|
|
header = getenv("PROXY_HEADER", "HTTP_REMOTE_USER")
|