78 lines
2.2 KiB
Python
Executable File
78 lines
2.2 KiB
Python
Executable File
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
|
|
|
|
for client_name in client_names:
|
|
|
|
client_success = False
|
|
client = self.server.clients.get(client_name)
|
|
|
|
if client and client['found']:
|
|
client_success = fireEvent('notify.plex.notifyClient', client, message, single = True)
|
|
|
|
if not client_success:
|
|
if self.server.staleClients() or not client:
|
|
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)
|
|
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}
|