Fix front page stats

This merge request fixes the way the numbers are gathered for the statistics on the front page.

It now shows an accurate picture to the user on the page whether anonymous or not.

Fixes #514 (closed)

beat/web/reports/models.py

@@ -33,6 +33,7 @@ from django.conf import settings
 from django.contrib.auth.models import User
 from django.core import exceptions as django_exceptions
 from django.db import models
+from django.db.models import Q
 from django.urls import reverse
 from django.utils.encoding import force_bytes
 from django.utils.encoding import force_text
@@ -93,6 +94,17 @@ class ReportManager(models.Manager):
     def published(self):
         return self.filter(status=Report.PUBLISHED)
 
+    def for_user(self, user, add_public=False):
+        if user.is_anonymous:
+            return self.published()
+
+        query = Q(author=user)
+
+        if add_public:
+            query |= Q(status=Report.PUBLISHED)
+
+        return self.filter(query).distinct()
+
 
 # ----------------------------------------------------------