CouchPotatoServer/couchpotato/core/notifications/base.py

from couchpotato.api import addApiView
from couchpotato.core.event import addEvent
from couchpotato.core.logger import CPLog
from couchpotato.core.providers.base import Provider
from couchpotato.environment import Env

log = CPLog(__name__)


class Notification(Provider):

    type = 'notification'

    default_title = Env.get('appname')
    test_message = 'ZOMG Lazors Pewpewpew!'

    listen_to = [
        'renamer.after', 'movie.snatched',
        'updater.available', 'updater.updated',
        'core.message',
    ]
    dont_listen_to = []

    def __init__(self):
        addEvent('notify.%s' % self.getName().lower(), self._notify)

        addApiView(self.testNotifyName(), self.test)

        # Attach listeners
        for listener in self.listen_to:
            if not listener in self.dont_listen_to:
                addEvent(listener, self.createNotifyHandler(listener))

    def createNotifyHandler(self, listener):
        def notify(message = None, group = {}, data = None):
            if not self.conf('on_snatch', default = True) and listener == 'movie.snatched':
                return
            return self._notify(message = message, data = data if data else group, listener = listener)

        return notify

    def getNotificationImage(self, size = 'small'):
        return 'https://raw.github.com/RuudBurger/CouchPotatoServer/master/couchpotato/static/images/notify.couch.%s.png' % size

    def _notify(self, *args, **kwargs):
        if self.isEnabled():
            return self.notify(*args, **kwargs)

    def notify(self, message = '', data = {}, listener = None):
        pass

    def test(self):

        test_type = self.testNotifyName()

        log.info('Sending test to %s', test_type)

        success = self._notify(
            message = self.test_message,
            data = {},
            listener = 'test'
        )

        return {
            'success': success
        }

    def testNotifyName(self):
        return 'notify.%s.test' % self.getName().lower()