github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-11 07:46:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

telegram updates to notifiers.py

Browse source
This commit is contained in:
devin 2015-10-28 07:31:07 -04:00
parent f75fca12c8
commit c723d33d38
1 changed files with 78 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

79
plexpy/notifiers.py
View file

@ -35,6 +35,8 @@ import json
import oauth2 as oauth
import pythontwitter as twitter
import telegram
from email.mime.text import MIMEText
import smtplib
import email.utils
@ -51,7 +53,8 @@ AGENT_IDS = {"Growl": 0,
"Boxcar2": 9,
"Email": 10,
"Twitter": 11,
"IFTTT": 12}
"IFTTT": 12,
"Telegram": 13}
def available_notification_agents():
agents = [{'name': 'Growl',
@ -209,6 +212,19 @@ def available_notification_agents():
'on_buffer': plexpy.CONFIG.IFTTT_ON_BUFFER,
'on_watched': plexpy.CONFIG.IFTTT_ON_WATCHED,
'on_created': plexpy.CONFIG.IFTTT_ON_CREATED
},
{'name': 'Telegram',
'id': AGENT_IDS['Telegram'],
'config_prefix': 'telegram',
'has_config': True,
'state': checked(plexpy.CONFIG.TELEGRAM_ENABLED),
'on_play': plexpy.CONFIG.TELEGRAM_ON_PLAY,
'on_stop': plexpy.CONFIG.TELEGRAM_ON_STOP,
'on_pause': plexpy.CONFIG.TELEGRAM_ON_PAUSE,
'on_resume': plexpy.CONFIG.TELEGRAM_ON_RESUME,
'on_buffer': plexpy.CONFIG.TELEGRAM_ON_BUFFER,
'on_watched': plexpy.CONFIG.TELEGRAM_ON_WATCHED,
'on_created': plexpy.CONFIG.TELEGRAM_ON_CREATED
}
]
@ -274,6 +290,9 @@ def get_notification_agent_config(config_id):
elif config_id == 12:
iftttClient = IFTTT()
return iftttClient.return_config_options()
elif config_id == 13:
telegramClient = TELEGRAM()
return telegramClient.return_config_options()
else:
return []
else:
@ -322,6 +341,9 @@ def send_notification(config_id, subject, body):
elif config_id == 12:
iftttClient = IFTTT()
iftttClient.notify(subject=subject, message=body)
elif config_id == 13:
telegramClient = TELEGRAM()
telegramClient.notify(message=body, event=subject)
else:
logger.debug(u"PlexPy Notifier :: Unknown agent id received.")
else:
@ -1417,3 +1439,58 @@ class IFTTT(object):
]
return config_option
class TELEGRAM(object):
def __init__(self):
self.enabled = plexpy.CONFIG.TELEGRAM_ENABLED
self.bot_token = plexpy.CONFIG.TELEGRAM_BOT_TOKEN
self.chat_id = plexpy.CONFIG.TELEGRAM_CHAT_ID
self.on_play = plexpy.CONFIG.TELEGRAM_ON_PLAY
self.on_stop = plexpy.CONFIG.TELEGRAM_ON_STOP
self.on_watched = plexpy.CONFIG.TELEGRAM_ON_WATCHED
def conf(self, options):
return cherrypy.config['config'].get('Telegram', options)
def notify(self, message, event):
if not message or not event:
return
bot = telegram.Bot(self.bot_token)
bot.sendMessage(chat_id=self.chat_id, text=message.encode("utf-8"))
logger.info(u"Telegram notifications sent.")
return True
def updateLibrary(self):
#For uniformity reasons not removed
return
def test(self, bot_token, chat_id):
self.enabled = True
self.bot_token = bot_token
self.chat_id = chat_id
self.notify('Main Screen Activate', 'Test Message')
def return_config_options(self):
config_option = [{'label': 'Telegram Bot ID',
'value': self.bot_token,
'name': 'telegram_bot_token',
'description': 'Your Bot ID here.',
'input_type': 'text'
},
{'label': 'Telegram Chat ID',
'value': self.chat_id,
'name': 'telegram_chat_id',
'description': 'Your Telegram Chat or Group ID.',
'input_type': 'text'
}
]
return config_option