From ed19fd0254b1301b5ba73619530f6cb792870c98 Mon Sep 17 00:00:00 2001
From: Dean Gardiner <gardiner91@gmail.com>
Date: Wed, 20 Nov 2013 22:04:11 +1300
Subject: [PATCH] Added Pushbullet notifications

---
 .../core/notifications/pushbullet/__init__.py      | 39 ++++++++++
 couchpotato/core/notifications/pushbullet/main.py  | 86 ++++++++++++++++++++++
 2 files changed, 125 insertions(+)
 create mode 100644 couchpotato/core/notifications/pushbullet/__init__.py
 create mode 100644 couchpotato/core/notifications/pushbullet/main.py

diff --git a/couchpotato/core/notifications/pushbullet/__init__.py b/couchpotato/core/notifications/pushbullet/__init__.py
new file mode 100644
index 0000000..e61a44e
--- /dev/null
+++ b/couchpotato/core/notifications/pushbullet/__init__.py
@@ -0,0 +1,39 @@
+from .main import Pushbullet
+
+def start():
+    return Pushbullet()
+
+config = [{
+    'name': 'pushbullet',
+    'groups': [
+        {
+            'tab': 'notifications',
+            'list': 'notification_providers',
+            'name': 'pushbullet',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'default': 0,
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'api_key',
+                    'label': 'User API Key'
+                },
+                {
+                    'name': 'devices',
+                    'default': '',
+                    'advanced': True,
+                    'description': 'IDs of devices to send notifications to, empty = all devices'
+                },
+                {
+                    'name': 'on_snatch',
+                    'default': 0,
+                    'type': 'bool',
+                    'advanced': True,
+                    'description': 'Also send message when movie is snatched.',
+                },
+            ],
+        }
+    ],
+}]
diff --git a/couchpotato/core/notifications/pushbullet/main.py b/couchpotato/core/notifications/pushbullet/main.py
new file mode 100644
index 0000000..2e6db29
--- /dev/null
+++ b/couchpotato/core/notifications/pushbullet/main.py
@@ -0,0 +1,86 @@
+from couchpotato.core.helpers.encoding import toUnicode
+from couchpotato.core.helpers.variable import tryInt
+from couchpotato.core.logger import CPLog
+from couchpotato.core.notifications.base import Notification
+import base64
+import json
+
+log = CPLog(__name__)
+
+
+class Pushbullet(Notification):
+
+    url = 'https://api.pushbullet.com/api/%s'
+
+    def notify(self, message = '', data = None, listener = None):
+        if not data: data = {}
+
+        devices = self.getDevices()
+        if devices is None:
+            return False
+
+        # Get all the device IDs linked to this user
+        if not len(devices):
+            response = self.request('devices')
+            if not response:
+                return False
+
+            devices += [device.get('id') for device in response['devices']]
+
+        successful = 0
+        for device in devices:
+            response = self.request(
+                'pushes',
+                cache = False,
+                device_id = device,
+                type = 'note',
+                title = self.default_title,
+                body = toUnicode(message)
+            )
+
+            if response:
+                successful += 1
+            else:
+                log.error('Unable to push notification to Pushbullet device with ID %s' % device)
+
+        return successful == len(devices)
+
+    def getDevices(self):
+        devices = [d.strip() for d in self.conf('devices').split(',')]
+
+        # Remove empty items
+        devices = [d for d in devices if len(d)]
+
+        # Break on any ids that aren't integers
+        valid_devices = []
+
+        for device_id in devices:
+            d = tryInt(device_id, None)
+
+            if not d:
+                log.error('Device ID "%s" is not valid', device_id)
+                return None
+
+            valid_devices.append(d)
+
+        return valid_devices
+
+    def request(self, method, cache = True, **kwargs):
+        try:
+            base64string = base64.encodestring('%s:' % self.conf('api_key'))[:-1]
+
+            headers = {
+                "Authorization": "Basic %s" % base64string
+            }
+
+            if cache:
+                return self.getJsonData(self.url % method, headers = headers, params = kwargs)
+            else:
+                data = self.urlopen(self.url % method, headers = headers, params = kwargs)
+                return json.loads(data)
+
+        except Exception, ex:
+            log.error('Pushbullet request failed')
+            log.debug(ex)
+
+        return None