X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=plomtask%2Fdays.py;h=aee4bf82354574b91b9b9f0983865daf59f54db1;hb=7bd9c956b0a0efaed731683732d3e54753e5269e;hp=0936d047aa3eda7c4ce9374c90ca3f9ad3362ad5;hpb=5a1e4ec7d9671dd9aa40d05e2af0d7ab25ea8911;p=plomtask

diff --git a/plomtask/days.py b/plomtask/days.py
index 0936d04..aee4bf8 100644
--- a/plomtask/days.py
+++ b/plomtask/days.py
@@ -44,7 +44,7 @@ class Day:
 
     @classmethod
     def all(cls, db_conn: DatabaseConnection,
-            date_range: tuple[str, str] = ('', '')):
+            date_range: tuple[str, str] = ('', ''), fill_gaps: bool = False):
         """Return list of Days in database within date_range."""
         def date_from_range_str(date_str: str, default: str):
             if date_str == '':
@@ -61,6 +61,15 @@ class Day:
         for row in db_conn.exec(sql, (start_date, end_date)):
             days += [cls.from_table_row(row)]
         days.sort()
+        if fill_gaps and len(days) > 1:
+            gapless_days = []
+            for i, day in enumerate(days):
+                gapless_days += [day]
+                if i < len(days) - 1:
+                    while day.next_date != days[i+1].date:
+                        day = Day(day.next_date)
+                        gapless_days += [day]
+            days = gapless_days
         return days
 
     @classmethod