| Linux server27.hostingraja.org 2.6.32-954.3.5.lve1.4.93.el6.x86_64 #1 SMP Wed Oct 4 17:04:29 UTC 2023 x86_64 Path : /opt/imunify360/venv/lib64/python3.11/site-packages/defence360agent/plugins/ |
| Current File : //opt/imunify360/venv/lib64/python3.11/site-packages/defence360agent/plugins/event_hook_executor.py |
from defence360agent.contracts.hook_events import HookEvent
from defence360agent.contracts.plugins import (
MessageSink,
MessageSource,
expect,
)
from defence360agent.hooks.execute import execute_hooks
EVENTS = (
HookEvent.AgentStarted,
HookEvent.AgentMisconfig,
HookEvent.LicenseExpired,
HookEvent.LicenseExpiring,
HookEvent.LicenseRenewed,
)
class EventHookExecutor(MessageSink, MessageSource):
PROCESSING_ORDER = MessageSink.ProcessingOrder.EVENT_HOOK
async def create_sink(self, loop):
self._loop = loop
async def create_source(self, loop, sink):
self._loop = loop
self._sink = sink
@expect(*EVENTS)
async def receive_event(self, event):
self._loop.create_task(execute_hooks(event))