first

handlers/SetRoleHandler.py

@@ -0,0 +1,40 @@
+from bot.Handler import Handler
+from db.services.ChatService import ChatService
+from bot import SET_EMAIL_STEP
+from bot.DaMsg import DaMsg
+from bot.msg.SetEmailMsg import SetEmailMsg
+
+
+class SetRoleHandler(Handler):
+
+    @staticmethod
+    async def set_partner(update, context):
+        chat = ChatService.get_by_chat_id(update.effective_chat.id)
+        ChatService.edit(chat, "status", 20)
+
+        msg = SetEmailMsg(update.effective_message)
+
+        await context.bot.send_message(chat_id=update.effective_chat.id, text=msg.get_msg())
+
+        return SET_EMAIL_STEP
+
+    @staticmethod
+    async def set_developer(update, context):
+        chat = ChatService.get_by_chat_id(update.effective_chat.id)
+        ChatService.edit(chat, "status", 10)
+
+        msg = SetEmailMsg(update.effective_message)
+
+        await context.bot.send_message(chat_id=update.effective_chat.id, text=msg.get_msg())
+
+        return SET_EMAIL_STEP
+
+    @staticmethod
+    async def set_email(update, context):
+        chat = ChatService.get_by_chat_id(update.effective_chat.id)
+        msg = DaMsg(update.effective_message)
+        ChatService.edit(chat, "email", msg.text)
+
+        await context.bot.send_message(chat_id=update.effective_chat.id, text="Спасибо)")
+
+        return SET_EMAIL_STEP