From b84537af0c547fa058ed1c5af25475dce4821f0d Mon Sep 17 00:00:00 2001
From: Evgeny Zinoviev <me@ch1p.io>
Date: Sun, 25 Apr 2021 23:17:45 +0300
Subject: fixes

---
 ch1p/functions.py | 15 +++++++++------
 ch1p/state.py     |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

(limited to 'ch1p')

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:
-- 
cgit v1.2.3