diff --git a/dudlebot.py b/dudlebot.py index 7a9a669..52e5273 100644 --- a/dudlebot.py +++ b/dudlebot.py @@ -127,12 +127,15 @@ class DudleBot: formatted_plan += f"{'|'.join([day.strftime('%A')[:2] for day in days])}" formatted_plan += "\n" - if len(plan.entries) == 0: - entries = {-1: PlanEntry(name="???", responses=[])} - else: - entries = plan.entries + entries_with_content = [entry for entry in plan.entries.values() if + any(response != PlanResponse.NO for response in entry.responses)] - for entry in entries.values(): + if len(entries_with_content) == 0: + entries = [PlanEntry(name="???", responses=[])] + else: + entries = plan.entries.values() + + for entry in entries: responses = [entry.responses[i] if i < len(entry.responses) else PlanResponse.UNKNOWN for i in range(plan.duration.days)]