X-Git-Url: https://plomlompom.com/repos/foo.html?a=blobdiff_plain;ds=sidebyside;f=plomtask%2Fdays.py;h=e3d56d7fb1049c161b146af555b05ccf2ed2f904;hb=edafc35b8b46e86b446df9b90a2754fc876da884;hp=fe1ba44e80e9a1c4182130b870d6d531dc9c9c06;hpb=e150bee233a648950061b716dc1780581105ede6;p=plomtask
diff --git a/plomtask/days.py b/plomtask/days.py
index fe1ba44..e3d56d7 100644
--- a/plomtask/days.py
+++ b/plomtask/days.py
@@ -63,15 +63,20 @@ class Day(BaseModel[str]):
for row in db_conn.exec(sql, (start_date, end_date)):
days += [cls.by_id(db_conn, row[0])]
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
+ if fill_gaps:
+ if start_date not in [d.date for d in days]:
+ days = [Day(start_date)] + days
+ if end_date not in [d.date for d in days]:
+ days += [Day(end_date)]
+ if 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
@property