first

handlers/AuthHandler.py

@@ -0,0 +1,35 @@
+from bot.Handler import Handler
+from bot import SET_DATE_STEP, SET_TABLE_STEP, REQUEST_CODE_STEP, FIRST, AUTH_FINISH, config
+from bot.keyboards.DateToBookingKeyboard import DateToBookingKeyboard
+from bot.keyboards.FirstStepKeyboard import FirstStepKeyboard
+from bot.keyboards.GetTableListKeyboard import GetTableListKeyboard
+from itguild_api import client
+from bot.DaMsg import DaMsg
+
+
+class AuthHandler(Handler):
+
+    @staticmethod
+    async def request_code(update, context):
+        await context.bot.send_message(chat_id=update.effective_chat.id, text="Введите код:")
+
+        return REQUEST_CODE_STEP
+
+    @staticmethod
+    async def add_user_at_dialog(update, context):
+        msg = DaMsg(update.effective_message)
+
+        user = client.tgBot.get_user({'token': msg.text})
+
+        if "error" in user:
+            reply_markup = FirstStepKeyboard()
+            await context.bot.send_message(chat_id=update.effective_chat.id,
+                                           text="Пользователь не найден. Проверте верность введенного кода.",
+                                           reply_markup=reply_markup.create_keyboard())
+        else:
+            dialog = client.tgBot.set_user_at_dialog({'dialog_id': update.effective_chat.id, 'user_id': user['id']})
+            await context.bot.send_message(chat_id=update.effective_chat.id,
+                                           text="Авторизация прошла успешно.")
+
+        print(user)
+        return FIRST