NotifierManager.py

import sys
import os
import logging
import config
from time import time
from sqlalchemy import or_
from PluginManager import PluginManager
from db import session, Node

logger = logging.getLogger(__name__)


class NotifierManager(PluginManager):
    modulename = 'Notifier'

    def find_matching(self, contact):
        for notifier in self.plugins:
            try:
                if notifier.suitable(contact):
                    return notifier
            except:
                logger.exception(
                    "Exception while determining if %s is suitable for %s",
                    notifier.__class__.__name__,
                    contact
                )

    @staticmethod
    def split_contacts(contacts):
        try:
            return contacts.split(', ')
        except AttributeError:
            return [contacts]

    def notify_node(self, node):
        contacts = self.split_contacts(node.contact)
        copy_contacts = self.split_contacts(config.copy_contact)

        notified = False
        for contact in contacts:
            notifier = self.find_matching(contact)
            if notifier:
                try:
                    notified |= bool(notifier.notify(contact, node))
                except:
                    logger.exception(
                        "Exception during notifiction via %s",
                        notifier.__class__.__name__
                    )

        if notified:
            for contact in set(copy_contacts) - set(contacts):
                notifier = self.find_matching(contact)
                if notifier:
                    try:
                        notifier.notify(contact, node)
                    except:
                        logger.exception(
                            "Exception during notifiction via %s",
                            notifier.__class__.__name__
                        )

        return notified

    def notify_down(self):
        to_notify = session.query(Node).filter(
            Node.lastseen <= time() - config.notify_timeout,
            or_(Node.lastcontact < Node.lastseen, Node.lastcontact == None),
            Node.contact != None,
            Node.ignore == 0 if config.whitelisting
            else or_(Node.ignore == None, Node.ignore == 0),
        )
        for node in to_notify:
            logger.info("Notifying node %s (ID %s)", node.name, node.id)
            if self.notify_node(node):
                node.lastcontact = time()
        session.commit()

if __name__ == '__main__':
    m = NotifierManager()
    print("Available notifiers:")
    for notifier in m.plugins:
        print("%s\t%s" % (notifier.__class__.__name__, notifier.regex.pattern))
    m.quit()