Current File : //opt/imunify360/venv/lib64/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,
                )