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

Common: Register serializers for {Exploitation,Propagation}Event

This commit is contained in:
Mike Salvatore 2022-10-03 13:15:12 -04:00
parent eb16969a56
commit 368ddde20f
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/common/agent_event_serializers/register.py
View File

@ -1,4 +1,10 @@
from common.agent_events import CredentialsStolenEvent, PingScanEvent, TCPScanEvent from common.agent_events import (
CredentialsStolenEvent,
ExploitationEvent,
PingScanEvent,
PropagationEvent,
TCPScanEvent,
)
from . import AgentEventSerializerRegistry, PydanticAgentEventSerializer from . import AgentEventSerializerRegistry, PydanticAgentEventSerializer
@ -11,3 +17,5 @@ def register_common_agent_event_serializers(
) )
event_serializer_registry[PingScanEvent] = PydanticAgentEventSerializer(PingScanEvent) event_serializer_registry[PingScanEvent] = PydanticAgentEventSerializer(PingScanEvent)
event_serializer_registry[TCPScanEvent] = PydanticAgentEventSerializer(TCPScanEvent) event_serializer_registry[TCPScanEvent] = PydanticAgentEventSerializer(TCPScanEvent)
event_serializer_registry[PropagationEvent] = PydanticAgentEventSerializer(PropagationEvent)
event_serializer_registry[ExploitationEvent] = PydanticAgentEventSerializer(ExploitationEvent)