staffeln/conductor/result.py

# Email notification package
# This should be upgraded by integrating with mail server to send batch
import staffeln.conf
from oslo_log import log
from staffeln.common import constants, email
from staffeln.common import time as xtime
from staffeln.conductor import backup
from staffeln.i18n import _

CONF = staffeln.conf.CONF
LOG = log.getLogger(__name__)


class BackupResult(object):
    def __init__(self):
        pass

    def initialize(self):
        self.content = ""
        self.project_list = set()

    def add_project(self, project_id, project_name):
        self.project_list.add((project_id, project_name))

    def send_result_email(self):
        subject = "Staffeln Backup result"
        try:
            if len(CONF.notification.receiver) == 0:
                LOG.info(
                    "Directly record report in log as no receiver "
                    "email provided. Report: %s" % self.content
                )
                return
            smtp_profile = {
                "src_email": CONF.notification.sender_email,
                "src_name": "Staffeln",
                "src_pwd": CONF.notification.sender_pwd,
                "dest_email": CONF.notification.receiver,
                "subject": subject,
                "content": self.content,
                "smtp_server_domain": CONF.notification.smtp_server_domain,
                "smtp_server_port": CONF.notification.smtp_server_port,
            }
            email.send(smtp_profile)
            LOG.info(_("Backup result email sent"))
        except Exception as e:
            LOG.error(
                _(
                    "Backup result email send failed. Please check email configuration. %s"
                    % (str(e))
                )
            )
            raise

    def publish(self):
        # 1. get quota
        self.content = "<h3>${TIME}</h3><br>"
        self.content = self.content.replace("${TIME}", xtime.get_current_strtime())
        backup_mgt = backup.Backup()
        project_success = {}
        project_failed = {}
        success_tasks = backup_mgt.get_queues(
            filters={"backup_status": constants.BACKUP_COMPLETED}
        )
        for task in success_tasks:
            if task.project_id in project_success:
                project_success[task.project_id].append(task)
            else:
                project_success[task.project_id] = [task]
        failed_tasks = backup_mgt.get_queues(
            filters={"backup_status": constants.BACKUP_FAILED}
        )
        for task in failed_tasks:
            if task.project_id in project_failed:
                project_failed[task.project_id].append(task)
            else:
                project_failed[task.project_id] = [task]

        html = ""
        for project_id, project_name in self.project_list:
            quota = backup_mgt.get_backup_quota(project_id)

            html += (
                "<h3>Project: ${PROJECT}</h3><h3>Quota Usage</h3>"
                "<FONT COLOR=${QUOTA_COLLOR}><h4>Limit: ${QUOTA_LIMIT}, In Use: "
                "${QUOTA_IN_USE}, Reserved: ${QUOTA_RESERVED}, Total "
                "rate: ${QUOTA_USAGE}</h4></FONT>"
                "<h3>Success List</h3>"
                "<FONT COLOR=GREEN><h4>${SUCCESS_VOLUME_LIST}</h4></FONT><br>"
                "<h3>Failed List</h3>"
                "<FONT COLOR=RED><h4>${FAILED_VOLUME_LIST}</h4></FONT><br>"
            )

            if project_id in project_success:
                success_volumes = "<br>".join(
                    [
                        (
                            f"Volume ID: {str(e.volume_id)}, Backup ID: {str(e.backup_id)}, "
                            f"Backup mode: {'Incremental' if e.incremental else 'Full'}, "
                            f"Created at: {str(e.created_at)}, Last updated at: "
                            f"{str(e.updated_at)}"
                        )
                        for e in project_success[project_id]
                    ]
                )
            else:
                success_volumes = "<br>"
            if project_id in project_failed:
                failed_volumes = "<br>".join(
                    [
                        (
                            f"Volume ID: {str(e.volume_id)}, Reason: {str(e.reason)}, "
                            f"Created at: {str(e.created_at)}, Last updated at: "
                            f"{str(e.updated_at)}"
                        )
                        for e in project_failed[project_id]
                    ]
                )
            else:
                failed_volumes = "<br>"
            quota_usage = (quota["in_use"] + quota["reserved"]) / quota["limit"]
            if quota_usage > 0.8:
                quota_color = "RED"
            elif quota_usage > 0.5:
                quota_color = "YALLOW"
            else:
                quota_color = "GREEN"
            html = html.replace("${QUOTA_USAGE}", str(quota_usage))
            html = html.replace("${QUOTA_COLLOR}", quota_color)
            html = html.replace("${QUOTA_LIMIT}", str(quota["limit"]))
            html = html.replace("${QUOTA_IN_USE}", str(quota["in_use"]))
            html = html.replace("${QUOTA_RESERVED}", str(quota["reserved"]))
            html = html.replace("${SUCCESS_VOLUME_LIST}", success_volumes)
            html = html.replace("${FAILED_VOLUME_LIST}", failed_volumes)
            html = html.replace("${PROJECT}", project_name)
        if html == "":
            return
        self.content += html
        self.send_result_email()