Improved Pushbullet plugin
This commit is contained in:
parent
d7266b21e5
commit
f8a709fe98
1 changed files with 83 additions and 11 deletions
|
@ -18,45 +18,116 @@ class PushbulletPlugin(Plugin):
|
||||||
* The :class:`platypush.backend.pushbullet.Pushbullet` backend enabled
|
* 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)
|
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
|
@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.
|
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
|
: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['body'] = body
|
||||||
kwargs['title'] = title
|
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',
|
resp = requests.post('https://api.pushbullet.com/v2/pushes',
|
||||||
data=json.dumps(kwargs),
|
data=json.dumps(kwargs),
|
||||||
headers={'Authorization': 'Bearer ' + pushbullet.token,
|
headers={'Authorization': 'Bearer ' + self.token,
|
||||||
'Content-Type': 'application/json'})
|
'Content-Type': 'application/json'})
|
||||||
|
|
||||||
if resp.status_code >= 400:
|
if resp.status_code >= 400:
|
||||||
raise Exception('Pushbullet push failed with status {}: {}'.
|
raise Exception('Pushbullet push failed with status {}: {}'.
|
||||||
format(resp.status_code, resp.json()))
|
format(resp.status_code, resp.json()))
|
||||||
|
|
||||||
|
|
||||||
@action
|
@action
|
||||||
def send_file(self, filename):
|
def send_file(self, filename: str, device: str = None):
|
||||||
"""
|
"""
|
||||||
Send a file.
|
Send a file.
|
||||||
|
|
||||||
|
:param device: Device ID or name (default: None, all devices)
|
||||||
:param filename: Path to the local file
|
: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')
|
pushbullet = get_backend('pushbullet')
|
||||||
resp = requests.post('https://api.pushbullet.com/v2/upload-request',
|
resp = requests.post('https://api.pushbullet.com/v2/upload-request',
|
||||||
data=json.dumps({'file_name': os.path.basename(filename)}),
|
data=json.dumps({'file_name': os.path.basename(filename)}),
|
||||||
headers={'Authorization': 'Bearer ' + pushbullet.token,
|
headers={'Authorization': 'Bearer ' + self.token,
|
||||||
'Content-Type': 'application/json'})
|
'Content-Type': 'application/json'})
|
||||||
|
|
||||||
if resp.status_code != 200:
|
if resp.status_code != 200:
|
||||||
|
@ -72,11 +143,12 @@ class PushbulletPlugin(Plugin):
|
||||||
format(resp.status_code))
|
format(resp.status_code))
|
||||||
|
|
||||||
resp = requests.post('https://api.pushbullet.com/v2/pushes',
|
resp = requests.post('https://api.pushbullet.com/v2/pushes',
|
||||||
headers={'Authorization': 'Bearer ' + pushbullet.token,
|
headers={'Authorization': 'Bearer ' + self.token,
|
||||||
'Content-Type': 'application/json'},
|
'Content-Type': 'application/json'},
|
||||||
|
|
||||||
data=json.dumps({
|
data=json.dumps({
|
||||||
'type': 'file',
|
'type': 'file',
|
||||||
|
'device_iden': device['iden'] if device else None,
|
||||||
'file_name': r['file_name'],
|
'file_name': r['file_name'],
|
||||||
'file_type': r['file_type'],
|
'file_type': r['file_type'],
|
||||||
'file_url': r['file_url'] }))
|
'file_url': r['file_url'] }))
|
||||||
|
|
Loading…
Reference in a new issue