40 lines
1.3 KiB
Python
40 lines
1.3 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 couchpotato.core.notifications.growl.growl import GROWL_UDP_PORT, \
|
|
GrowlRegistrationPacket, GrowlNotificationPacket
|
|
from couchpotato.environment import Env
|
|
from socket import AF_INET, SOCK_DGRAM, socket
|
|
|
|
log = CPLog(__name__)
|
|
|
|
|
|
class Growl(Notification):
|
|
|
|
listen_to = ['movie.downloaded', 'movie.snatched']
|
|
|
|
def conf(self, attr):
|
|
return Env.setting(attr, 'growl')
|
|
|
|
def notify(self, message = '', data = {}, type = None):
|
|
if self.dontNotify(type): return
|
|
|
|
hosts = [x.strip() for x in self.conf('host').split(",")]
|
|
password = self.conf('password')
|
|
|
|
for curHost in hosts:
|
|
addr = (curHost, GROWL_UDP_PORT)
|
|
|
|
s = socket(AF_INET, SOCK_DGRAM)
|
|
p = GrowlRegistrationPacket(password = password)
|
|
p.addNotification()
|
|
s.sendto(p.payload(), addr)
|
|
|
|
# send notification
|
|
p = GrowlNotificationPacket(title = self.default_title, description = message, priority = 0, sticky = False, password = password)
|
|
s.sendto(p.payload(), addr)
|
|
s.close()
|
|
|
|
log.info('Growl notifications sent.')
|