Brett Pemberton
8 years ago
committed by
JackDandy
JackDandy
7 changed files with 254 additions and 3 deletions
|
After Width: | Height: | Size: 15 KiB |
@ -0,0 +1,90 @@ |
|||
# coding=utf-8 |
|||
# |
|||
# This file is part of SickGear. |
|||
# |
|||
# Thanks to: mallen86 |
|||
# |
|||
# SickGear is free software: you can redistribute it and/or modify |
|||
# it under the terms of the GNU General Public License as published by |
|||
# the Free Software Foundation, either version 3 of the License, or |
|||
# (at your option) any later version. |
|||
# |
|||
# SickGear is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU General Public License |
|||
# along with SickGear. If not, see <http://www.gnu.org/licenses/>. |
|||
|
|||
import json |
|||
import sickbeard |
|||
|
|||
from sickbeard import logger, common |
|||
|
|||
class SlackNotifier: |
|||
|
|||
def _send_to_slack(self, message, accessToken, channel, as_user, bot_name, icon_url): |
|||
SLACK_ENDPOINT = "https://slack.com/api/chat.postMessage" |
|||
|
|||
data = {} |
|||
data["token"] = accessToken |
|||
data["channel"] = channel |
|||
data["username"] = bot_name |
|||
data["text"] = message |
|||
data["icon_url"] = icon_url |
|||
data["as_user"] = as_user |
|||
|
|||
urlResp = sickbeard.helpers.getURL(url=SLACK_ENDPOINT, post_data=data) |
|||
if urlResp: |
|||
resp = json.loads(urlResp) |
|||
else: |
|||
return False |
|||
|
|||
# if ("error" in resp): |
|||
# raise Exception(resp["error"]) |
|||
|
|||
if (resp["ok"] == True): |
|||
logger.log(u"Slack: Succeeded sending message.", logger.MESSAGE) |
|||
return True |
|||
|
|||
logger.log(u"Slack: Failed sending message: " + resp["error"], logger.ERROR) |
|||
return False |
|||
|
|||
def _notify(self, message, accessToken='', channel='', as_user='', bot_name='', icon_url='', force=False): |
|||
# suppress notifications if the notifier is disabled but the notify options are checked |
|||
if not sickbeard.USE_SLACK and not force: |
|||
return False |
|||
|
|||
if not accessToken: |
|||
accessToken = sickbeard.SLACK_ACCESS_TOKEN |
|||
if not channel: |
|||
channel = sickbeard.SLACK_CHANNEL |
|||
if not as_user: |
|||
as_user = sickbeard.SLACK_AS_USER |
|||
if not bot_name: |
|||
bot_name = sickbeard.SLACK_BOT_NAME |
|||
if not icon_url: |
|||
icon_url = sickbeard.SLACK_ICON_URL |
|||
|
|||
return self._send_to_slack(message, accessToken, channel, as_user, bot_name, icon_url) |
|||
|
|||
############################################################################## |
|||
# Public functions |
|||
############################################################################## |
|||
|
|||
def notify_snatch(self, ep_name): |
|||
if sickbeard.SLACK_NOTIFY_ONSNATCH: |
|||
self._notify(common.notifyStrings[common.NOTIFY_SNATCH] + ': ' + ep_name) |
|||
|
|||
def notify_download(self, ep_name): |
|||
if sickbeard.SLACK_NOTIFY_ONDOWNLOAD: |
|||
self._notify(common.notifyStrings[common.NOTIFY_DOWNLOAD] + ': ' + ep_name) |
|||
|
|||
def test_notify(self, accessToken, channel, as_user, bot_name, icon_url): |
|||
return self._notify("This is a test notification from SickGear", accessToken, channel, as_user, bot_name, icon_url, force=True) |
|||
|
|||
def update_library(self, ep_obj): |
|||
pass |
|||
|
|||
notifier = SlackNotifier |
|||
Loading…
Reference in new issue