Cleanup

Author: cmdelatorre

src/pycamp_bot/commands/wizard.py

@@ -15,6 +15,8 @@
 WIZARD_TIME_START_HOUR = 9
 WIZARD_TIME_END_HOUR = 20
 
+MSG_MAX_LEN = 4096
+
 
 def is_wizard_time_slot(slot):
     return slot[0].hour in range(WIZARD_TIME_START_HOUR, WIZARD_TIME_END_HOUR)
@@ -231,12 +233,17 @@ async def schedule_wizards(update, context):
     agenda = WizardAtPycamp.select().where(WizardAtPycamp.pycamp == pycamp)
 
     msg = format_wizards_schedule(agenda)
-    
-    await context.bot.send_message(
-        chat_id=update.message.chat_id,
-        text=msg,
-        parse_mode="MarkdownV2"
-    )
+    try:
+        await context.bot.send_message(
+            chat_id=update.message.chat_id,
+            text=msg,
+            parse_mode="MarkdownV2"
+        )  
+    except BadRequest as e:
+        m = "Coulnd't return the Wizards list to the admin. ".format(update.message.from_user.username)
+        if len(msg) >= MSG_MAX_LEN:
+            m += "The message is too long. Check the data in the DB ;-)"
+        logger.exception(m)
 
 
 def format_wizards_schedule(agenda):

src/pycamp_bot/models.py

@@ -60,6 +60,8 @@ class Pycamp(BaseModel):
     end: time of end
     vote_authorized: the vote is auth in this pycamp
     project_load_authorized: the project load is auth in this pycamp
+    active: boolean telling wheter this PyCamp instance is active (or an old one)
+    wizard_slot_duration: config to compute the schedule of mages
     '''
     headquarters = pw.CharField(unique=True)
     init = pw.DateTimeField(null=True)

test/conftest.py

@@ -3,24 +3,19 @@
 from datetime import datetime
 from functools import wraps
 from peewee import SqliteDatabase
-import pytest
 from telegram import Bot
 
 from pycamp_bot.models import Pycampista, Slot, Pycamp, WizardAtPycamp
 
 
-
-bot = Bot(token=os.environ['TOKEN'])
-
-
 # use an in-memory SQLite for tests.
 test_db = SqliteDatabase(':memory:')
 
 MODELS = [Pycampista, Slot, Pycamp, WizardAtPycamp]
 
 
-# Bind the given models to the db for the duration of wrapped block.
 def use_test_database(fn):
+    """Bind the given models to the db for the duration of wrapped block."""
     @wraps(fn)
     def inner(self):
         with test_db.bind_ctx(MODELS):
@@ -30,7 +25,3 @@ def inner(self):
             finally:
                 test_db.drop_tables(MODELS)
     return inner
-
-
-TEST_INIT_DATE = datetime(2024, 6, 20, 0, 0, 0)
-TEST_END_DATE = datetime(2024, 6, 24, 23, 59, 59, 99999)

test/test_wizard.py

@@ -1,12 +1,9 @@
 from datetime import datetime
 from pycamp_bot.models import Pycamp, Pycampista, Slot
 from pycamp_bot.commands import wizard
-from test.conftest import use_test_database, test_db, MODELS, TEST_END_DATE, TEST_INIT_DATE
+from test.conftest import use_test_database, test_db, MODELS
 
 
-# ---------------------------
-# Module Level Setup/TearDown
-# ---------------------------
 def setup_module(module):
     """setup any state specific to the execution of the given module."""
     test_db.bind(MODELS, bind_refs=False, bind_backrefs=False)