refactor: Make a class for the bot

Cria uma classe para conter a instancia de bot.
Externaliza os handlers para um módulo próprio.
Muda o endpoint para rodar o bot para um script run.py no root.

* Trata de #4

bpb/bpb.py

@@ -1,152 +1,37 @@
-import locale
 import logging
-import os
-from datetime import timedelta
 
-from telegram.ext import CommandHandler, Filters, MessageHandler, Updater
+from telegram.ext import Handler, Updater
 
-from messages import PLEASE_SCHEDULE, SEM_REUNIAO, START, WELCOME
-from utils import parse_date
-
-# Configuration
 logging.basicConfig(
     format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
     level=logging.INFO,
 )
 logger = logging.getLogger(__name__)
 
-locale.setlocale(locale.LC_TIME, "pt_BR.UTF-8")
-
-# Handlers
-def start(update, context):
-
-    context.bot.send_message(
-        chat_id=update.effective_chat.id,
-        text=START,
-    )
-
-
-def get_meeting(update, context):
-
-    try:
-        datetime_obj = getattr(context.bot, "next_meeting")
-
-        parsed_date, _ = parse_date(datetime_obj)
-        message = f"A próxima reunião está marcada para:\n {parsed_date}.\n".lower().capitalize()
-
-        try:
-
-            next_meetings = getattr(context.bot, "next_meetings")
-            message += "Esse horário irá se repetir pelas próximas 4 semanas:\n"
-            message += " \n".join(i[0] for i in next_meetings)
-
-        except AttributeError:
-
-            pass
-
-    except Exception as _:
-
-        message = SEM_REUNIAO + PLEASE_SCHEDULE
-
-    context.bot.send_message(chat_id=update.effective_chat.id, text=message)
-
-
-def set_meeting(update, context):
-
-    try:
-
-        # Get message meeting and following ones
-        parsed_date, datetime_obj = parse_date(context.args)
-        next_meetings, interval = [
-            (parsed_date, datetime_obj),
-        ], 7
-        for _ in range(3):
-            next_meetings.append(parse_date(datetime_obj + timedelta(days=interval)))
-            interval += interval
-
-        # Add attributes to bot
-        context.bot.next_meeting = datetime_obj
-        context.bot.next_meetings = next_meetings
-
-        message = f"Reunião marcada para:\n {parsed_date}.\n".lower().capitalize()
-        message += "Esse horário irá se repetir pelas próximas 4 semanas:\n"
-        message += " \n".join(i[0] for i in context.bot.next_meetings)
-
-    except Exception as e:
-
-        message = (
-            "Não consegui marcar a reunião. Por favor verifique a mensagem submetida:\n"
-        )
-        message += (
-            f"'{' '.join(context.args)}'"
-            if len(context.args) > 1
-            else "Ué, você não enviou nada!"
-        )
-
-        logger.error(e)
-
-    context.bot.send_message(chat_id=update.effective_chat.id, text=message)
-
-
-def clear_meetings(update, context):
-
-    try:
-
-        delattr(context.bot, "next_meeting")
-        delattr(context.bot, "next_meetings")
-        message = "Limpei a agenda de reuniões." + PLEASE_SCHEDULE
-
-    except AttributeError:
-
-        message = SEM_REUNIAO
-
-    context.bot.send_message(chat_id=update.effective_chat.id, text=message)
-
-
-def links(update, context):
-
-    with open("bpb/links_importantes.html", "r") as link_file:
-        links_importantes = link_file.read()
-
-    context.bot.send_message(
-        chat_id=update.effective_chat.id, text=links_importantes, parse_mode="HTML"
-    )
-
-
-def welcome(update, context):
-
-    new_member = update.message.new_chat_members[0]
-
-    user_text = f"[{new_member.full_name}](tg://user?id={new_member.id})"
-
-    welcome_message = WELCOME.format(user_text=user_text)
-
-    update.message.reply_text(welcome_message, parse_mode="Markdown")
 
+class CodBonitoBot:
+    def __init__(self, token):
 
-handlers = [
-    MessageHandler(Filters.status_update.new_chat_members, welcome),
-    CommandHandler("start", start),
-    CommandHandler("setmeeting", set_meeting),
-    CommandHandler("getmeeting", get_meeting),
-    CommandHandler("clear", clear_meetings),
-    CommandHandler("links", links),
-]
+        logging.info("Inicializando o bot...")
 
+        self.token = token
+        self._updater = Updater(token=self.token)
 
-def main():
+    def _add_handler(self, handler):
+        if not isinstance(handler, Handler):
+            raise ValueError("Handler deve ser do tipo Handler!")
+        self._updater.dispatcher.add_handler(handler)
 
-    # Defining bot
-    updater = Updater(token=os.environ.get("TEL_TOKEN"))
-    dispatcher = updater.dispatcher
+    def config_handlers(self, handlers):
 
-    for handler in handlers:
-        dispatcher.add_handler(handler)
+        logging.info("Configurando os handlers...")
 
-    updater.start_polling()
+        for handler in handlers:
+            self._add_handler(handler)
 
-    updater.idle()
+    def run(self):
 
+        self._updater.start_polling()
+        logging.info("Bot está rodando!")
 
-if __name__ == "__main__":
-    main()
+        self._updater.idle()

bpb/handlers.py

@@ -0,0 +1,137 @@
+import locale
+import logging
+from datetime import timedelta
+
+from telegram.ext import CommandHandler, Filters, MessageHandler
+
+from .messages import PLEASE_SCHEDULE, SEM_REUNIAO, START, WELCOME
+from .utils import parse_date
+
+logging.basicConfig(
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    level=logging.INFO,
+)
+logger = logging.getLogger(__name__)
+
+locale.setlocale(locale.LC_TIME, "pt_BR.UTF-8")
+
+
+def start(update, context):
+
+    context.bot.send_message(
+        chat_id=update.effective_chat.id,
+        text=START,
+    )
+
+
+def get_meeting(update, context):
+
+    try:
+        datetime_obj = getattr(context.bot, "next_meeting")
+
+        parsed_date, _ = parse_date(datetime_obj)
+        message = f"A próxima reunião está marcada para:\n {parsed_date}.\n".lower().capitalize()
+
+        try:
+
+            next_meetings = getattr(context.bot, "next_meetings")
+            message += "Esse horário irá se repetir pelas próximas 4 semanas:\n"
+            message += " \n".join(i[0] for i in next_meetings)
+
+        except AttributeError as e:
+
+            logger.error(e)
+            pass
+
+    except Exception as _:
+
+        message = SEM_REUNIAO + PLEASE_SCHEDULE
+
+    context.bot.send_message(chat_id=update.effective_chat.id, text=message)
+
+
+def set_meeting(update, context):
+
+    try:
+
+        # Get message meeting and following ones
+        parsed_date, datetime_obj = parse_date(context.args)
+        next_meetings, interval = [
+            (parsed_date, datetime_obj),
+        ], 7
+        for _ in range(3):
+            next_meetings.append(parse_date(datetime_obj + timedelta(days=interval)))
+            interval += interval
+
+        # Add attributes to bot
+        context.bot.next_meeting = datetime_obj
+        context.bot.next_meetings = next_meetings
+
+        message = f"Reunião marcada para:\n {parsed_date}.\n".lower().capitalize()
+        message += "Esse horário irá se repetir pelas próximas 4 semanas:\n"
+        message += " \n".join(i[0] for i in context.bot.next_meetings)
+
+    except Exception as e:
+
+        message = (
+            "Não consegui marcar a reunião. Por favor verifique a mensagem submetida:\n"
+        )
+        message += (
+            f"'{' '.join(context.args)}'"
+            if len(context.args) > 1
+            else "Ué, você não enviou nada!"
+        )
+
+        logger.error(e)
+
+    context.bot.send_message(chat_id=update.effective_chat.id, text=message)
+
+
+def clear_meetings(update, context):
+
+    try:
+
+        delattr(context.bot, "next_meeting")
+        delattr(context.bot, "next_meetings")
+        message = "Limpei a agenda de reuniões." + PLEASE_SCHEDULE
+
+    except AttributeError:
+
+        message = SEM_REUNIAO
+
+    context.bot.send_message(chat_id=update.effective_chat.id, text=message)
+
+
+def links(update, context):
+
+    with open("bpb/links_importantes.html", "r") as link_file:
+        links_importantes = link_file.read()
+
+    context.bot.send_message(
+        chat_id=update.effective_chat.id, text=links_importantes, parse_mode="HTML"
+    )
+
+
+def welcome(update, context):
+
+    new_member = update.message.new_chat_members[0]
+
+    user_text = f"[{new_member.full_name}](tg://user?id={new_member.id})"
+
+    welcome_message = WELCOME.format(user_text=user_text)
+
+    update.message.reply_text(welcome_message, parse_mode="Markdown")
+
+
+def get_handlers():
+
+    handlers = [
+        MessageHandler(Filters.status_update.new_chat_members, welcome),
+        CommandHandler("start", start),
+        CommandHandler("setmeeting", set_meeting),
+        CommandHandler("getmeeting", get_meeting),
+        CommandHandler("clear", clear_meetings),
+        CommandHandler("links", links),
+    ]
+
+    return handlers

bpb/messages.py

@@ -9,4 +9,4 @@
     "Se apresente e conte como encontrou o grupo!"
 )
 
-SEM_REUNIAO = "Nenhuma reunião agendada."
+SEM_REUNIAO = "Nenhuma reunião agendada."

run.py

@@ -0,0 +1,15 @@
+import os
+from bpb.bpb import CodBonitoBot
+from bpb.handlers import get_handlers
+
+if __name__ == "__main__":
+
+    token = os.environ.get("TEL_TOKEN")
+
+    bot = CodBonitoBot(token)
+
+    handlers = get_handlers()
+
+    bot.config_handlers(handlers)
+
+    bot.run()