diff options
Diffstat (limited to 'ch1p')
| -rw-r--r-- | ch1p/functions.py | 15 | ||||
| -rw-r--r-- | ch1p/state.py | 2 |
2 files changed, 10 insertions, 7 deletions
diff --git a/ch1p/functions.py b/ch1p/functions.py index c4f9c99..cf75221 100644 --- a/ch1p/functions.py +++ b/ch1p/functions.py @@ -18,17 +18,20 @@ def _get_vars(params: List[Tuple], kw: dict) -> List[AnyStr]: return result -def telegram_notify(text: str, parse_mode: str = 'html', **kwargs): - token, chat_id = _get_vars([ +def telegram_notify(text: str, parse_mode: str = None, **kwargs): + chat_id, token = _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={ + data = { 'chat_id': chat_id, - 'text': text, - 'parse_mode': parse_mode - }) + 'text': text + } + if parse_mode is not None: + data['parse_mode'] = parse_mode + + r = requests.post('https://api.telegram.org/bot%s/sendMessage' % token, data=data) if r.status_code != 200: raise RuntimeError("telegram returned %d" % r.status_code) diff --git a/ch1p/state.py b/ch1p/state.py index f677f59..e7fe050 100644 --- a/ch1p/state.py +++ b/ch1p/state.py @@ -1,6 +1,6 @@ import os, json -from functions import _get_vars +from .functions import _get_vars class State: |