35 lines
969 B
Python
35 lines
969 B
Python
import os, requests
|
|
|
|
from typing import List, Tuple, AnyStr
|
|
|
|
|
|
def _get_vars(params: List[Tuple], kw: dict) -> List[AnyStr]:
|
|
result = []
|
|
|
|
for kw_name, env_name in params:
|
|
env_name = f'MY_{env_name}'
|
|
if kw_name in kw:
|
|
result.append(kw[kw_name])
|
|
elif env_name in os.environ:
|
|
result.append(os.environ[env_name])
|
|
else:
|
|
raise RuntimeError("missing parameter %s or variable %s" % (kw_name, env_name))
|
|
|
|
return result
|
|
|
|
|
|
def telegram_notify(text: str, parse_mode: str = 'html', **kwargs):
|
|
token, chat_id = _get_vars([
|
|
('chat_id', 'TELEGRAM_NOTIFY_CHAT_ID'),
|
|
('token', 'TELEGRAM_NOTIFY_TOKEN')
|
|
], kwargs)
|
|
|
|
r = requests.post('https://api.telegram.org/bot%s/sendMessage' % token, data={
|
|
'chat_id': chat_id,
|
|
'text': text,
|
|
'parse_mode': parse_mode
|
|
})
|
|
|
|
if r.status_code != 200:
|
|
raise RuntimeError("telegram returned %d" % r.status_code)
|