From 5905d3bf5b0cbfeeaa05237f28168480e17338ab Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 13 Sep 2022 15:02:07 -0400 Subject: [PATCH] Common: Add register_common_agent_event_serializers() --- monkey/common/event_serializers/__init__.py | 1 + monkey/common/event_serializers/register.py | 9 +++++++++ 2 files changed, 10 insertions(+) create mode 100644 monkey/common/event_serializers/register.py diff --git a/monkey/common/event_serializers/__init__.py b/monkey/common/event_serializers/__init__.py index 8701670ab..388edf6f3 100644 --- a/monkey/common/event_serializers/__init__.py +++ b/monkey/common/event_serializers/__init__.py @@ -1,3 +1,4 @@ from .i_event_serializer import IEventSerializer, JSONSerializable from .event_serializer_registry import EventSerializerRegistry from .pydantic_event_serializer import PydanticEventSerializer +from .register import register_common_agent_event_serializers diff --git a/monkey/common/event_serializers/register.py b/monkey/common/event_serializers/register.py new file mode 100644 index 000000000..6c533c9bf --- /dev/null +++ b/monkey/common/event_serializers/register.py @@ -0,0 +1,9 @@ +from common.events import CredentialsStolenEvent + +from . import EventSerializerRegistry, PydanticEventSerializer + + +def register_common_agent_event_serializers(event_serializer_registry: EventSerializerRegistry): + event_serializer_registry[CredentialsStolenEvent] = PydanticEventSerializer( + CredentialsStolenEvent + )