jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 05:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update telegram.py add send_document (#2937)

Browse Source 
🐬
This commit is contained in:
icovada 2016-08-27 02:52:44 +02:00 committed by Teagan Glenn
parent 4aad83d60b
commit a4b8c3cab0
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/components/notify/telegram.py
View File

@ -21,6 +21,7 @@ _LOGGER = logging.getLogger(__name__)
REQUIREMENTS = ['python-telegram-bot==5.0.0'] REQUIREMENTS = ['python-telegram-bot==5.0.0']
ATTR_PHOTO = "photo" ATTR_PHOTO = "photo"
ATTR_DOCUMENT = "document"
ATTR_CAPTION = "caption" ATTR_CAPTION = "caption"
CONF_CHAT_ID = 'chat_id' CONF_CHAT_ID = 'chat_id'
@ -102,6 +103,8 @@ class TelegramNotificationService(BaseNotificationService):
return return
elif data is not None and ATTR_LOCATION in data: elif data is not None and ATTR_LOCATION in data:
return self.send_location(data.get(ATTR_LOCATION)) return self.send_location(data.get(ATTR_LOCATION))
elif data is not None and ATTR_DOCUMENT in data:
return self.send_document(data.get(ATTR_DOCUMENT))
# send message # send message
try: try:
@ -125,6 +128,20 @@ class TelegramNotificationService(BaseNotificationService):
_LOGGER.exception("Error sending photo.") _LOGGER.exception("Error sending photo.")
return return
def send_document(self, data):
"""Send a document."""
import telegram
caption = data.pop(ATTR_CAPTION, None)
# send photo
try:
document = load_data(**data)
self.bot.sendDocument(chat_id=self._chat_id,
document=document, caption=caption)
except telegram.error.TelegramError:
_LOGGER.exception("Error sending document.")
return
def send_location(self, gps): def send_location(self, gps):
"""Send a location.""" """Send a location."""
import telegram import telegram