Improved Pushbullet plugin

This commit is contained in:
Fabio Manganiello 2019-12-09 11:00:31 +00:00
parent d7266b21e5
commit f8a709fe98

View file

@ -18,45 +18,116 @@ class PushbulletPlugin(Plugin):
* The :class:`platypush.backend.pushbullet.Pushbullet` backend enabled
"""
def __init__(self, *args, **kwargs):
def __init__(self, token: str = None, *args, **kwargs):
"""
:param token: Pushbullet API token. If not set the plugin will try to retrieve it from
the Pushbullet backend configuration, if available
"""
super().__init__(*args, **kwargs)
if not token:
backend = get_backend('pushbullet')
if not backend or not backend.token:
raise AttributeError('No Pushbullet token specified')
self.token = backend.token
else:
self.token = token
self._devices = []
self._devices_by_id = {}
self._devices_by_name = {}
@action
def send_note(self, body=None, title=None, **kwargs):
def get_devices(self):
"""
Get the list of available devices
"""
resp = requests.get('https://api.pushbullet.com/v2/devices',
headers={'Authorization': 'Bearer ' + self.token,
'Content-Type': 'application/json'})
self._devices = resp.json().get('devices', [])
self._devices_by_id = {
dev['iden']: dev
for dev in self._devices
}
self._devices_by_name = {
dev['nickname']: dev
for dev in self._devices if 'nickname' in dev
}
@action
def get_device(self, device):
"""
:param device: Device ID or name
"""
output = None
refreshed = False
while not output:
if device in self._devices_by_id:
return self._devices_by_id[device]
if device in self._devices_by_name:
return self._devices_by_name[device]
if refreshed:
return None
self.get_devices()
refreshed = True
@action
def send_note(self, device: str = None, body: str = None, title: str = None, url: str = None, **kwargs):
"""
Send a note push.
:param device: Device ID or name (default: None, all devices)
:param body: Note body
:param title: Note title
:param url: URL attached to the note
:param kwargs: Push arguments, see https://docs.pushbullet.com/#create-push
:type kwargs: dict
"""
pushbullet = get_backend('pushbullet')
if device:
device = self.get_device(device).output
if not device:
raise RuntimeError('No such device')
kwargs['body'] = body
kwargs['title'] = title
kwargs['type'] = 'note'
kwargs['type'] = 'link' if url else 'note'
if device:
kwargs['device_iden'] = device['iden']
resp = requests.post('https://api.pushbullet.com/v2/pushes',
data=json.dumps(kwargs),
headers={'Authorization': 'Bearer ' + pushbullet.token,
headers={'Authorization': 'Bearer ' + self.token,
'Content-Type': 'application/json'})
if resp.status_code >= 400:
raise Exception('Pushbullet push failed with status {}: {}'.
format(resp.status_code, resp.json()))
@action
def send_file(self, filename):
def send_file(self, filename: str, device: str = None):
"""
Send a file.
:param device: Device ID or name (default: None, all devices)
:param filename: Path to the local file
:type filename: str
"""
if device:
device = self.get_device(device).output
if not device:
raise RuntimeError('No such device')
pushbullet = get_backend('pushbullet')
resp = requests.post('https://api.pushbullet.com/v2/upload-request',
data=json.dumps({'file_name': os.path.basename(filename)}),
headers={'Authorization': 'Bearer ' + pushbullet.token,
headers={'Authorization': 'Bearer ' + self.token,
'Content-Type': 'application/json'})
if resp.status_code != 200:
@ -72,11 +143,12 @@ class PushbulletPlugin(Plugin):
format(resp.status_code))
resp = requests.post('https://api.pushbullet.com/v2/pushes',
headers={'Authorization': 'Bearer ' + pushbullet.token,
headers={'Authorization': 'Bearer ' + self.token,
'Content-Type': 'application/json'},
data=json.dumps({
'type': 'file',
'device_iden': device['iden'] if device else None,
'file_name': r['file_name'],
'file_type': r['file_type'],
'file_url': r['file_url'] }))