Ruud
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from couchpotato.api import addApiView
|
|
from couchpotato.core.event import addEvent
|
|
from couchpotato.core.logger import CPLog
|
|
from couchpotato.core.notifications.base import Notification
|
|
from xml.dom import minidom
|
|
import urllib
|
|
|
|
log = CPLog(__name__)
|
|
|
|
|
|
class Plex(Notification):
|
|
|
|
listen_to = ['movie.downloaded', 'movie.snatched']
|
|
|
|
def notify(self, message = '', data = {}, type = None):
|
|
if self.dontNotify(type): return
|
|
|
|
log.info('Sending notification to Plex')
|
|
hosts = [x.strip() for x in self.conf('host').split(",")]
|
|
|
|
for host in hosts:
|
|
|
|
source_type = ['movie']
|
|
base_url = 'http://%s/library/sections' % host
|
|
refresh_url = '%s/%%s/refresh' % base_url
|
|
|
|
try:
|
|
xml_sections = minidom.parse(urllib.urlopen(base_url))
|
|
sections = xml_sections.getElementsByTagName('Directory')
|
|
for s in sections:
|
|
if s.getAttribute('type') in source_type:
|
|
url = refresh_url % s.getAttribute('key')
|
|
x = urllib.urlopen(url)
|
|
except:
|
|
log.error('Plex library update failed for %s.' % host)
|
|
|
|
return True
|