p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Agent: Add default argument and type hint in send_all_events_to_island.py

This commit is contained in:
Shreya Malviya 2022-09-14 15:43:38 +05:30
parent fe63b28ee1
commit 37c4362b60
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
monkey/infection_monkey/send_all_events_to_island.py
View File

@ -6,6 +6,7 @@ from time import sleep
import requests
from common.common_consts.timeouts import MEDIUM_REQUEST_TIMEOUT
from common.event_serializers.i_event_serializer import JSONSerializable
from common.events import AbstractAgentEvent
logger = logging.getLogger(__name__)
@ -23,9 +24,7 @@ class SendAllAgentEventsToIsland:
def __init__(self, server_address: str):
self._server_address = server_address
self._agent_events_to_island_sender = _AgentEventsToIslandSender(
self._server_address, DEFAULT_TIME_PERIOD_SECONDS
)
self._agent_events_to_island_sender = AgentEventsToIslandSender(self._server_address)
self._agent_events_to_island_sender.start()
def __del__(self):
@ -43,12 +42,12 @@ class SendAllAgentEventsToIsland:
pass
class _AgentEventsToIslandSender:
class AgentEventsToIslandSender:
"""
Handles the batching and sending of the Agent's events to the Island
"""
def __init__(self, server_address: str, time_period: int):
def __init__(self, server_address: str, time_period: int = DEFAULT_TIME_PERIOD_SECONDS):
self._server_address = server_address
self._time_period = time_period
@ -61,7 +60,7 @@ class _AgentEventsToIslandSender:
)
self._batch_and_send_thread.start()
def add_event_to_queue(self, serialized_event: AbstractAgentEvent):
def add_event_to_queue(self, serialized_event: JSONSerializable):
self._queue.put(serialized_event)
def _manage_event_batches(self):