p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Common: Rename EventSerializerRegistry -> AgentEventSerializerRegistry

This commit is contained in:
Shreya Malviya 2022-09-16 12:24:44 +05:30 committed by Mike Salvatore
parent c4ab5d99a9
commit 6ced4af5dc
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/common/event_serializers/__init__.py
View File

@ -1,5 +1,5 @@
from .consts import EVENT_TYPE_FIELD
from .i_agent_event_serializer import IAgentEventSerializer, JSONSerializable
from .event_serializer_registry import EventSerializerRegistry
from .event_serializer_registry import AgentEventSerializerRegistry
from .pydantic_event_serializer import PydanticEventSerializer
from .register import register_common_agent_event_serializers

4
monkey/common/event_serializers/event_serializer_registry.py
View File

@ -4,12 +4,12 @@ from common.event_serializers import IAgentEventSerializer
from common.events import AbstractAgentEvent
class EventSerializerRegistry:
class AgentEventSerializerRegistry:
"""
Registry for event serializers using event class.
Example:
event_serializer_registry = EventSerializerRegistry()
event_serializer_registry = AgentEventSerializerRegistry()
event_serializer_registry[MyEvent] = MyEventSerializer()
my_event_dict = {"type": "MyEvent", "data": "123"}

6
monkey/common/event_serializers/register.py
View File

@ -1,9 +1,11 @@
from common.events import CredentialsStolenEvent
from . import EventSerializerRegistry, PydanticEventSerializer
from . import AgentEventSerializerRegistry, PydanticEventSerializer
def register_common_agent_event_serializers(event_serializer_registry: EventSerializerRegistry):
def register_common_agent_event_serializers(
event_serializer_registry: AgentEventSerializerRegistry,
):
event_serializer_registry[CredentialsStolenEvent] = PydanticEventSerializer(
CredentialsStolenEvent
)