Current File : /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/subsys//systemd_notifier.py
"""Notify systemd about process state"""
import logging
import os
from defence360agent.contracts.config import ANTIVIRUS_MODE
logger = logging.getLogger(__name__)
class AgentState(object):
"""Allowed agent state for notifying systemd."""
READY = "READY=1"
STARTING = "STATUS=Starting main process"
MIGRATING = "STATUS=Applying database migrations"
DAEMONIZED = "STATUS=Demonized"
def notify(state):
"""
Send notification to systemd, allowed formats described here
https://www.freedesktop.org/software/systemd/man/sd_notify.html
For example:
notify("STATUS=Almost ready")
"""
if ANTIVIRUS_MODE:
return
try:
import sdnotify
# We've made this import inside function deliberately because some
# systems don't have systemd and stdnotify thus
except ImportError as e:
logger.info("Can't import sdnotify, reason: %s", e)
else:
if "NOTIFY_SOCKET" not in os.environ:
logger.warning(
"$NOTIFY_SOCKET environment variable is not set. It is set by"
" service manager for supervised processes for status and"
" start-up completion notification. Nowhere to send the"
" status."
)
else:
try:
n = sdnotify.SystemdNotifier(debug=True)
n.notify(state)
except Exception as e: # NOSONAR
logger.exception(
"some problem has occurred during notifying of"
" systemd: %s",
e,
)
