show plot only for logged in user

Author: ConstantinoSchillebeeckx

oolong_app/models.py

@@ -762,6 +762,9 @@ class PlotResponse(models.Model):
     avg = models.FloatField()
     std = models.FloatField()
     median = models.FloatField()
+    user = models.ForeignKey(
+        settings.AUTH_USER_MODEL,
+    )
 
     class Meta:
             managed = False

oolong_app/templates/plot.html

@@ -11,7 +11,13 @@
     <div id="bodywrap" class="plot">
         {% container type="fluid" %}
             {% row %}
-                <div class="col-xs-12" id="chart"></div>
+                {% column %}
+                    {% if plotdat %}
+                        <div id="chart"></div>
+                    {% else %}
+                        <p class="lead">No data found for {{ user.get_username }}; please fill out the <a href='/submit_questionnaire/'>questionnaire</a> first.</p>
+                    {% endif %}
+                {% endcolumn %}
             {% endrow %}
         {% endcontainer %}
     </div>

oolong_app/views.py

@@ -38,9 +38,10 @@ def index(request):
 @login_required 
 def plot(request):
 
-    dat = PlotResponse.objects.all()
+    dat = PlotResponse.objects.filter(user=request.user)
+    plotdat = None if not dat.count() else json.dumps(list(dat.values()))
     context = {
-        'plotdat': json.dumps(list(dat.values()))
+        'plotdat': plotdat
     }
 
     response = render(request, 'plot.html', context)