forked from platypush/platypush
162 lines
4.9 KiB
Python
162 lines
4.9 KiB
Python
import logging
|
|
import json
|
|
import requests
|
|
import time
|
|
import websocket
|
|
|
|
from platypush.config import Config
|
|
from platypush.message import Message
|
|
from platypush.message.event.pushbullet import PushbulletEvent
|
|
|
|
from .. import Backend
|
|
|
|
class PushbulletBackend(Backend):
|
|
def __init__(self, token, device, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
self.token = token
|
|
self.device_name = device
|
|
self.pb_device_id = self.get_device_id()
|
|
|
|
self._last_received_msg = {
|
|
'request' : { 'body': None, 'time': None },
|
|
'response' : { 'body': None, 'time': None },
|
|
'event' : { 'body': None, 'time': None },
|
|
}
|
|
|
|
def _get_latest_push(self):
|
|
t = int(time.time()) - 5
|
|
try:
|
|
response = requests.get(
|
|
u'https://api.pushbullet.com/v2/pushes',
|
|
headers = { 'Access-Token': self.token },
|
|
params = {
|
|
'modified_after': str(t),
|
|
'active' : 'true',
|
|
'limit' : 1,
|
|
}
|
|
)
|
|
|
|
response = response.json()
|
|
except Exception as e:
|
|
logging.exception(e)
|
|
raise e
|
|
|
|
if 'pushes' in response and response['pushes']:
|
|
return response['pushes'][0]
|
|
else:
|
|
return {}
|
|
|
|
def _should_skip_last_received_msg(self, msg):
|
|
if not isinstance(msg, dict): return True # We received something weird
|
|
|
|
is_duplicate=False
|
|
last_msg = self._last_received_msg[msg['type']]
|
|
|
|
if last_msg:
|
|
msg = Message.parse(msg)
|
|
if str(msg) == str(last_msg['body']) \
|
|
and time.time() - last_msg['time'] <= 2:
|
|
# Duplicate message sent on the Pushbullet socket within
|
|
# two seconds, ignore it
|
|
logging.debug('Ignoring duplicate message received on the socket')
|
|
is_duplicate = True
|
|
|
|
self._last_received_msg[msg['type']] = {
|
|
'body': msg, 'time': time.time()
|
|
}
|
|
|
|
return is_duplicate
|
|
|
|
def on_push(self):
|
|
def _f(ws, data):
|
|
try:
|
|
# Parse the push
|
|
try:
|
|
data = json.loads(data) if isinstance(data, str) else push
|
|
except Exception as e:
|
|
logging.exception(e)
|
|
return
|
|
|
|
# If it's a push, get it
|
|
if data['type'] == 'tickle' and data['subtype'] == 'push':
|
|
push = self._get_latest_push()
|
|
elif data['type'] == 'push':
|
|
push = data['push']
|
|
else: return # Not a push notification
|
|
|
|
# Post an event, useful to react on mobile notifications if
|
|
# you enabled notification mirroring on your PushBullet app
|
|
event = PushbulletEvent(**push)
|
|
self.on_message(event)
|
|
|
|
if 'body' not in push: return
|
|
logging.debug('Received push: {}'.format(push))
|
|
|
|
body = push['body']
|
|
try: body = json.loads(body)
|
|
except ValueError as e: return # Some other non-JSON push
|
|
|
|
if not self._should_skip_last_received_msg(body):
|
|
self.on_message(body)
|
|
except Exception as e:
|
|
logging.exception(e)
|
|
return
|
|
|
|
return _f
|
|
|
|
def on_error(self):
|
|
def _f(ws, e):
|
|
logging.exception(e)
|
|
logging.info('Restarting PushBullet backend')
|
|
ws.close()
|
|
self._init_socket()
|
|
|
|
return _f
|
|
|
|
def _init_socket(self):
|
|
self.ws = websocket.WebSocketApp(
|
|
'wss://stream.pushbullet.com/websocket/' + self.token,
|
|
on_message = self.on_push(),
|
|
on_error = self.on_error())
|
|
|
|
def get_device_id(self):
|
|
response = requests.get(
|
|
u'https://api.pushbullet.com/v2/devices',
|
|
headers = { 'Access-Token': self.token },
|
|
).json()
|
|
|
|
devices = [dev for dev in response['devices'] if 'nickname' in dev
|
|
and dev['nickname'] == self.device_name]
|
|
|
|
if not devices:
|
|
raise RuntimeError('No such Pushbullet device: {}'
|
|
.format(self.device_name))
|
|
|
|
return devices[0]['iden']
|
|
|
|
def send_message(self, msg):
|
|
requests.post(
|
|
u'https://api.pushbullet.com/v2/pushes',
|
|
headers = { 'Access-Token': self.token },
|
|
json = {
|
|
'type': 'note',
|
|
'device_iden': self.pb_device_id,
|
|
'body': str(msg)
|
|
}
|
|
).json()
|
|
|
|
def on_stop(self):
|
|
self.ws.close()
|
|
|
|
def run(self):
|
|
super().run()
|
|
|
|
self._init_socket()
|
|
logging.info('Initialized Pushbullet backend - device_id: {}'
|
|
.format(self.device_name))
|
|
|
|
self.ws.run_forever()
|
|
|
|
# vim:sw=4:ts=4:et:
|
|
|