diff --git a/project/utils.py b/project/utils.py index 902683e5..a1529b02 100644 --- a/project/utils.py +++ b/project/utils.py @@ -1,8 +1,12 @@ -from datetime import timedelta +from datetime import datetime, timedelta from collections import Counter def date_range(start, end): + if isinstance(start, datetime): + start = start.date() + if isinstance(end, datetime): + end = end.date() delta = end - start for d in range(delta.days + 1): yield start + timedelta(days=d)