Functions to compute the wozards agenda (WIP)

Author: cmdelatorre

src/pycamp_bot/commands/wizard.py

@@ -1,7 +1,59 @@
-from telegram.ext import CommandHandler
-from pycamp_bot.models import Pycampista
 import random
+from datetime import datetime, timedelta
+
+from telegram.ext import CommandHandler
+from pycamp_bot.models import Pycampista, WizardAtPycamp
+from pycamp_bot.commands.auth import admin_needed
+from pycamp_bot.commands.manage_pycamp import get_active_pycamp
+from pycamp_bot.logger import logger
+
+
+
+def compute_wizards_slots(pycamp):
+    """
+    * Magos trabajan de 9 a 19, sacando almuerzo (13 a 14).
+    * Magos trabajan desde el mediodía del primer día, hasta el mediodía del último día.
+    Slots son [start; end)
+    """
+    wizard_start = pycamp.init
+    wizard_end = pycamp.end
+    slots = []
+    current_period = wizard_start
+    while current_period < wizard_end:
+        slot_start = current_period
+        slot_end = current_period + timedelta(minutes=pycamp.wizard_slot_duration)
+        slots.append(
+            (slot_start, slot_end)
+        )
+        current_period = slot_end
+
+    return slots
+
+
+def define_wizards_schedule(pycamp):
+    """
+    Returns a dict whose keys are times and values are wizards (Pycampistas instances).
 
+    """
+    slots_list = compute_wizards_slots(pycamp)
+    wizards_list = pycamp.get_wizards()
+    n_wizards = wizards_list.count()
+    wizard_per_slot = {}
+    idx = 0
+    for slot in slots_list:
+        wizard = wizards_list[idx%n_wizards]
+        n_iter = 0  # railguard
+        while wizard.is_busy(slot):
+            logger.info('Mago {} con conflicto en el slot {}. Pruebo otro.'.format(wizard.username, slot))
+            if n_iter == n_wizards:
+                logger.warning('Queda el mago {} con conflicto en el slot {}'.format(wizard, slot))
+                break
+            idx += 1
+            wizard = wizards_list[idx%n_wizards]
+            n_iter += 1
+        wizard_per_slot[slot] = wizard
+        idx += 1
+    return wizard_per_slot
 
 async def become_wizard(update, context):
     current_wizards = Pycampista.select().where(Pycampista.wizard is True)