jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-10 06:47:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for file attachments in pushbullet (#8763)

Browse Source 
* Add support for file attachments in pishbullet

* Check filepath is allowed
This commit is contained in:
Tsvi Mostovicz 2017-08-01 15:55:46 +03:00 committed by Pascal Vizeli
parent 342ec8ec99
commit 075422e7ad
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
homeassistant/components/notify/pushbullet.py
View File

@ -19,6 +19,7 @@ REQUIREMENTS = ['pushbullet.py==0.11.0']
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
ATTR_URL = 'url' ATTR_URL = 'url'
ATTR_FILE = 'file'
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_API_KEY): cv.string, vol.Required(CONF_API_KEY): cv.string,
@ -80,14 +81,21 @@ class PushBulletNotificationService(BaseNotificationService):
title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT) title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)
data = kwargs.get(ATTR_DATA) data = kwargs.get(ATTR_DATA)
url = None url = None
filepath = None
if data: if data:
url = data.get(ATTR_URL, None) url = data.get(ATTR_URL, None)
filepath = data.get(ATTR_FILE, None)
refreshed = False refreshed = False
if not targets: if not targets:
# Backward compatibility, notify all devices in own account # Backward compatibility, notify all devices in own account
if url: if url:
self.pushbullet.push_link(title, url, body=message) self.pushbullet.push_link(title, url, body=message)
if self.hass.config.is_allowed_path(filepath):
with open(filepath, "rb") as fileh:
filedata = self.pushbullet.upload_file(fileh, filepath)
self.pushbullet.push_file(title=title, body=message,
**filedata)
else: else:
self.pushbullet.push_note(title, message) self.pushbullet.push_note(title, message)
_LOGGER.info("Sent notification to self") _LOGGER.info("Sent notification to self")
@ -107,6 +115,11 @@ class PushBulletNotificationService(BaseNotificationService):
if url: if url:
self.pushbullet.push_link( self.pushbullet.push_link(
title, url, body=message, email=tname) title, url, body=message, email=tname)
if self.hass.config.is_allowed_path(filepath):
with open(filepath, "rb") as fileh:
filedata = self.pushbullet.upload_file(fileh, filepath)
self.pushbullet.push_file(title=title, body=message,
**filedata)
else: else:
self.pushbullet.push_note(title, message, email=tname) self.pushbullet.push_note(title, message, email=tname)
_LOGGER.info("Sent notification to email %s", tname) _LOGGER.info("Sent notification to email %s", tname)