CouchPotatoServer/couchpotato/core/notifications/plex/main.py

from couchpotato.core.event import addEvent, fireEvent
from couchpotato.core.logger import CPLog
from couchpotato.core.notifications.base import Notification
from .client import PlexClientHTTP, PlexClientJSON
from .server import PlexServer

log = CPLog(__name__)


class Plex(Notification):

    http_time_between_calls = 0

    def __init__(self):
        super(Plex, self).__init__()

        self.server = PlexServer(self)

        self.client_protocols = {
            'http': PlexClientHTTP(self),
            'json': PlexClientJSON(self)
        }

        addEvent('renamer.after', self.addToLibrary)


    def addToLibrary(self, message = None, group = {}):
        if self.isDisabled(): return

        return self.server.refresh()

    def getClientNames(self):
        return [
            x.strip().lower()
            for x in self.conf('clients').split(',')
        ]

    def notifyClients(self, message, client_names):
        success = True

        while len(client_names):
            client_name = client_names[0]
            client_success = False
            client = self.server.clients.get(client_name)

            if client:
                client_success = fireEvent('notify.plex.notifyClient', client, message, single=True)

                if client_success:
                    client_names.pop(0)

            if not client_success:
                if self.server.staleClients():
                    log.info('Failed to send notification to client "%s". '
                             'Client list is stale, updating the client list and retrying.', client_name)
                    self.server.updateClients(self.getClientNames())
                else:
                    log.warning('Failed to send notification to client %s, skipping this time', client_name)
                    client_names.pop(0)
                    success = False

        return success

    def notify(self, message = '', data = {}, listener = None):
        return self.notifyClients(message, self.getClientNames())

    def test(self, **kwargs):

        test_type = self.testNotifyName()

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

        notify_success = self.notify(
            message = self.test_message,
            data = {},
            listener = 'test'
        )

        refresh_success = self.addToLibrary()

        return {'success': notify_success or refresh_success}