diff --git a/mood/statistics.py b/mood/statistics.py index 518d583..d4e7c64 100644 --- a/mood/statistics.py +++ b/mood/statistics.py @@ -9,12 +9,9 @@ from dateutil.relativedelta import relativedelta from .models import Status, Mood -def moodstats(user, mindate=None, maxdate=None, days=7): +def moodstats(user): hv.extension('bokeh') - maxdate = maxdate or timezone.now() - mindate = mindate or (maxdate - relativedelta(days=days)) - tooltips = [ ('Date', '@date{%F %H:%M}'), ('Value', '@value') @@ -60,9 +57,6 @@ def moodstats(user, mindate=None, maxdate=None, days=7): return output def activitystats(user): - maxdate = maxdate or timezone.now() - mindate = mindate or (maxdate - relativedelta(days=days)) - output = {} for status in Status.objects.filter(user=user): diff --git a/mood/views.py b/mood/views.py index 50be667..1119740 100644 --- a/mood/views.py +++ b/mood/views.py @@ -356,14 +356,23 @@ class MoodCSVView(LoginRequiredMixin, View): startdate = request.GET.get("start") enddate = request.GET.get("end") + maxdate = None + mindate = None + if enddate: maxdate = datetime.strptime(enddate, "%Y-%m-%d") - else: - maxdate = timezone.now() + + if not startdate: + mindate = maxdate - relativedelta.relativedelta(weeks=1) if startdate: mindate = datetime.strptime(startdate, "%Y-%m-%d") - else: + + if not enddate: + maxdate = mindate + relativedelta.relativedelta(weeks=1) + + if not maxdate: + maxdate = timezone.now() mindate = maxdate - relativedelta.relativedelta(weeks=1) output = "date,value"