evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Change PR to suggestion on gitter (#3243)

Browse source
This commit is contained in:
Pascal Vizeli 2016-09-07 15:57:59 +02:00 • committed by Paulus Schoutsen
parent 3668afe306
commit 91028cbc13
1 changed files with 18 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
homeassistant/components/notify/telegram.py
View file

@ -24,6 +24,10 @@ REQUIREMENTS = ['python-telegram-bot==5.0.0']
ATTR_PHOTO = "photo"
ATTR_DOCUMENT = "document"
ATTR_CAPTION = "caption"
ATTR_URL = 'url'
ATTR_FILE = 'file'
ATTR_USERNAME = 'username'
ATTR_PASSWORD = 'password'
CONF_CHAT_ID = 'chat_id'
@ -105,8 +109,6 @@ class TelegramNotificationService(BaseNotificationService):
elif data is not None and ATTR_DOCUMENT in data:
return self.send_document(data.get(ATTR_DOCUMENT))
text = ''
if title:
text = '{} {}'.format(title, message)
else:
@ -126,11 +128,16 @@ class TelegramNotificationService(BaseNotificationService):
def send_photo(self, data):
"""Send a photo."""
import telegram
caption = data.pop(ATTR_CAPTION, None)
caption = data.get(ATTR_CAPTION)
# send photo
try:
photo = load_data(**data)
photo = load_data(
url=data.get(ATTR_URL),
file=data.get(ATTR_FILE),
username=data.get(ATTR_USERNAME),
password=data.get(ATTR_PASSWORD),
)
self.bot.sendPhoto(chat_id=self._chat_id,
photo=photo, caption=caption)
except telegram.error.TelegramError:
@ -140,11 +147,16 @@ class TelegramNotificationService(BaseNotificationService):
def send_document(self, data):
"""Send a document."""
import telegram
caption = data.pop(ATTR_CAPTION, None)
caption = data.get(ATTR_CAPTION)
# send photo
try:
document = load_data(**data)
document = load_data(
url=data.get(ATTR_URL),
file=data.get(ATTR_FILE),
username=data.get(ATTR_USERNAME),
password=data.get(ATTR_PASSWORD),
)
self.bot.sendDocument(chat_id=self._chat_id,
document=document, caption=caption)
except telegram.error.TelegramError: