root: cleanup session keys to use common format (#3003)
cleanup session keys to use common format Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
		@ -23,7 +23,10 @@ from requests import RequestException
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik import __version__
 | 
			
		||||
from authentik.core.middleware import SESSION_IMPERSONATE_ORIGINAL_USER, SESSION_IMPERSONATE_USER
 | 
			
		||||
from authentik.core.middleware import (
 | 
			
		||||
    SESSION_KEY_IMPERSONATE_ORIGINAL_USER,
 | 
			
		||||
    SESSION_KEY_IMPERSONATE_USER,
 | 
			
		||||
)
 | 
			
		||||
from authentik.core.models import ExpiringModel, Group, PropertyMapping, User
 | 
			
		||||
from authentik.events.geo import GEOIP_READER
 | 
			
		||||
from authentik.events.utils import cleanse_dict, get_user, model_to_dict, sanitize_dict
 | 
			
		||||
@ -233,15 +236,15 @@ class Event(ExpiringModel):
 | 
			
		||||
        if hasattr(request, "user"):
 | 
			
		||||
            original_user = None
 | 
			
		||||
            if hasattr(request, "session"):
 | 
			
		||||
                original_user = request.session.get(SESSION_IMPERSONATE_ORIGINAL_USER, None)
 | 
			
		||||
                original_user = request.session.get(SESSION_KEY_IMPERSONATE_ORIGINAL_USER, None)
 | 
			
		||||
            self.user = get_user(request.user, original_user)
 | 
			
		||||
        if user:
 | 
			
		||||
            self.user = get_user(user)
 | 
			
		||||
        # Check if we're currently impersonating, and add that user
 | 
			
		||||
        if hasattr(request, "session"):
 | 
			
		||||
            if SESSION_IMPERSONATE_ORIGINAL_USER in request.session:
 | 
			
		||||
                self.user = get_user(request.session[SESSION_IMPERSONATE_ORIGINAL_USER])
 | 
			
		||||
                self.user["on_behalf_of"] = get_user(request.session[SESSION_IMPERSONATE_USER])
 | 
			
		||||
            if SESSION_KEY_IMPERSONATE_ORIGINAL_USER in request.session:
 | 
			
		||||
                self.user = get_user(request.session[SESSION_KEY_IMPERSONATE_ORIGINAL_USER])
 | 
			
		||||
                self.user["on_behalf_of"] = get_user(request.session[SESSION_KEY_IMPERSONATE_USER])
 | 
			
		||||
        # User 255.255.255.255 as fallback if IP cannot be determined
 | 
			
		||||
        self.client_ip = get_client_ip(request)
 | 
			
		||||
        # Apply GeoIP Data, when enabled
 | 
			
		||||
 | 
			
		||||
		Reference in New Issue
	
	Block a user