Source code for core.signals.save_Daemon

# SPDX-License-Identifier: AGPL-3.0-or-later
#
# Eonvelope - a open-source self-hostable email archiving server
# Copyright (C) 2024 David Aderbauer & The Eonvelope Contributors
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.

"""Save signal receivers for the :class:`core.models.Daemon` model."""

from __future__ import annotations

import logging
from typing import Any

from django.db.models.signals import post_save
from django.dispatch import receiver

from core.models import Daemon

logger = logging.getLogger(__name__)




@receiver(post_save, sender=Daemon)
def post_save_daemon_is_healthy(
    sender: Daemon,
    instance: Daemon,
    created: bool,  # noqa: FBT001  # required for receiver decorator
    **kwargs: Any,
) -> None:
    """Receiver function flagging the mailbox of a daemon as healthy once that daemon becomes healthy again.

    Args:
        sender: The class type that sent the post_save signal.
        instance: The instance that has been saved.
        created: Whether the instance was newly created.
        **kwargs: Other keyword arguments.
    """
    if created or instance.is_healthy is None:
        return

    if instance.is_healthy and "is_healthy" in instance.get_dirty_fields():
        logger.debug(
            "%s has become healthy, flagging its mailbox as healthy ...", instance
        )
        instance.mailbox.set_healthy()
        logger.debug("Successfully flagged mailbox as healthy.")