From 707cf630c7c333c1cf9954a988b750831295d0b5 Mon Sep 17 00:00:00 2001 From: Ivlev Denis Date: Fri, 4 May 2018 18:34:47 +0300 Subject: [PATCH] Add check date type --- project/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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)