daily-todo.py

@@ -93,10 +93,7 @@ with DAVClient(
         task.expand_rrule(yesterday, next_week)
         due = coerce_date(due)
 
-        try:
-            start = coerce_date(task.icalendar_component.get("dtstart").dt or yesterday)
-        except:
-            start = yesterday
+        start = coerce_date(task.icalendar_component.get("dtstart").dt or yesterday)
 
         if today < start:
             continue
@@ -121,15 +118,28 @@ with DAVClient(
         else:
             scheduled.append(event)
 
-    should_want = shuffle_with_priority(should_do) + shuffle_with_priority(want_to_do)
+    should_do = shuffle_with_priority(should_do)
+    want_to_do = shuffle_with_priority(want_to_do)
+
+    while len(must_do) < 2:
+        if should_do:
+            must_do.append(should_do.pop(0))
+        elif want_to_do:
+            must_do.append(want_to_do.pop(0))
+        else:
+            break  # No remaining tasks
+
+    while len(should_do) < 3:
+        if want_to_do:
+            should_do.append(want_to_do.pop(0))
+        else:
+            break  # No remaining tasks
 
     categories = {
         "Must do": must_do,
-        "Should do": should_want[:3],
-        "Want to do": should_want[3:6],
+        "Should do": should_do[:3],
+        "Want to do": want_to_do[:3],
     }
-    if len(should_want) > 6:
-        categories["Leftover"] = sould_want[7:]
 
 with p:
     log.debug("Printing output")