This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/audit/signals.py

93 lines
2.7 KiB
Python
Raw Normal View History

"""passbook audit signal listener"""
from threading import Thread
from typing import Any, Dict, Optional
from django.contrib.auth.signals import (
user_logged_in,
user_logged_out,
user_login_failed,
)
from django.dispatch import receiver
from django.http import HttpRequest
from passbook.audit.models import Event, EventAction
from passbook.core.models import User
2020-07-07 13:46:57 +00:00
from passbook.stages.invitation.models import Invitation
from passbook.stages.invitation.signals import invitation_created, invitation_used
2020-07-07 13:46:57 +00:00
from passbook.stages.user_write.signals import user_write
class EventNewThread(Thread):
"""Create Event in background thread"""
action: EventAction
request: HttpRequest
kwargs: Dict[str, Any]
2020-07-07 13:46:57 +00:00
user: Optional[User] = None
def __init__(self, action: EventAction, request: HttpRequest, **kwargs):
super().__init__()
self.action = action
self.request = request
self.kwargs = kwargs
def run(self):
Event.new(self.action, **self.kwargs).from_http(self.request, user=self.user)
@receiver(user_logged_in)
# pylint: disable=unused-argument
def on_user_logged_in(sender, request: HttpRequest, user: User, **_):
"""Log successful login"""
thread = EventNewThread(EventAction.LOGIN, request)
thread.user = user
thread.run()
@receiver(user_logged_out)
# pylint: disable=unused-argument
def on_user_logged_out(sender, request: HttpRequest, user: User, **_):
"""Log successfully logout"""
thread = EventNewThread(EventAction.LOGOUT, request)
thread.user = user
thread.run()
2020-07-07 13:46:57 +00:00
@receiver(user_write)
# pylint: disable=unused-argument
def on_user_write(sender, request: HttpRequest, user: User, data: Dict[str, Any], **_):
"""Log User write"""
thread = EventNewThread(EventAction.CUSTOM, request, **data)
thread.user = user
thread.run()
@receiver(user_login_failed)
# pylint: disable=unused-argument
def on_user_login_failed(
sender, credentials: Dict[str, str], request: HttpRequest, **_
):
"""Failed Login"""
thread = EventNewThread(EventAction.LOGIN_FAILED, request, **credentials)
thread.run()
2018-12-10 13:21:42 +00:00
@receiver(invitation_created)
# pylint: disable=unused-argument
2020-07-07 13:46:57 +00:00
def on_invitation_created(sender, request: HttpRequest, invitation: Invitation, **_):
2018-12-10 13:49:15 +00:00
"""Log Invitation creation"""
thread = EventNewThread(
2020-07-07 13:46:57 +00:00
EventAction.INVITE_CREATED, request, invitation_uuid=invitation.invite_uuid.hex
)
thread.run()
2018-12-10 13:21:42 +00:00
@receiver(invitation_used)
# pylint: disable=unused-argument
2020-07-07 13:46:57 +00:00
def on_invitation_used(sender, request: HttpRequest, invitation: Invitation, **_):
2018-12-10 13:49:15 +00:00
"""Log Invitation usage"""
thread = EventNewThread(
2020-07-07 13:46:57 +00:00
EventAction.INVITE_USED, request, invitation_uuid=invitation.invite_uuid.hex
2019-12-31 11:51:16 +00:00
)
thread.run()