From 5cef8525109a8e3c607af3545a2171b4973f5df7 Mon Sep 17 00:00:00 2001
From: Klaus-Uwe Mitterer <git@kumi.email>
Date: Fri, 26 Feb 2021 07:38:17 +0100
Subject: [PATCH] Fix moodstats signature Remove unused variables Fix time
 selection for CSV

---
 mood/statistics.py |  8 +-------
 mood/views.py      | 15 ++++++++++++---
 2 files changed, 13 insertions(+), 10 deletions(-)

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"