From f1ebd8d4c6e8f1a55705d90626ac220f40678db9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Ml=C3=A1dek?= Date: Wed, 23 Oct 2019 10:25:52 +0200 Subject: [PATCH] don't show entries that are only declines --- dudlebot.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) 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)]