sensors_bot: experimental fix

1 files changed, 23 insertions, 4 deletions

diff --git a/src/home/telegram/bot.py b/src/home/telegram/bot.py
index 9044b07..86ed734 100644
--- a/src/home/telegram/bot.py
+++ b/src/home/telegram/bot.py
@@ -76,8 +76,6 @@ def _handler_of_handler(*args, **kwargs):
                   store=db)
     try:
         _args.insert(0, ctx)
-        if self:
-            _args.insert(0, self)
 
         f = kwargs['f']
         del kwargs['f']
@@ -88,6 +86,17 @@ def _handler_of_handler(*args, **kwargs):
         else:
             return_with_context = False
 
+        if 'argument' in kwargs and kwargs['argument'] == 'message_key':
+            mkey = None
+            for k, v in lang.get_langpack(ctx.lang).items():
+                if ctx.text == v:
+                    mkey = k
+                    break
+            _args.insert(0, mkey)
+
+        if self:
+            _args.insert(0, self)
+
         result = f(*_args, **kwargs)
         return result if not return_with_context else (result, ctx)
 
@@ -104,16 +113,22 @@ def _handler_of_handler(*args, **kwargs):
 def handler(**kwargs):
     def inner(f):
         @wraps(f)
-        def _handler(*args, **kwargs):
-            return _handler_of_handler(f=f, *args, **kwargs)
+        def _handler(*args, **inner_kwargs):
+            if 'argument' in kwargs and kwargs['argument'] == 'message_key':
+                inner_kwargs['argument'] = 'message_key'
+            return _handler_of_handler(f=f, *args, **inner_kwargs)
 
         if 'message' in kwargs:
             _updater.dispatcher.add_handler(MessageHandler(text_filter(lang.all(kwargs['message'])), _handler), group=0)
+
         if 'command' in kwargs:
             _updater.dispatcher.add_handler(CommandHandler(kwargs['command'], _handler), group=0)
+
         if 'callback' in kwargs:
             _updater.dispatcher.add_handler(CallbackQueryHandler(_handler, pattern=kwargs['callback']), group=0)
+
         return _handler
+
     return inner
 
 
@@ -447,6 +462,10 @@ def add_conversation(conv: conversation) -> None:
     _updater.dispatcher.add_handler(conv.get_handler(), group=0)
 
 
+def add_handler(h):
+    _updater.dispatcher.add_handler(h, group=0)
+
+
 def start(ctx: Context):
     return _start_handler_ref(ctx)