Task #3716: [/api/v1/stats/discover] Show stats for discover secction

taiga/stats/api.py

@@ -17,25 +17,40 @@ from collections import OrderedDict
 from django.conf import settings
 from django.views.decorators.cache import cache_page
 
-from taiga.base.api import viewsets
 from taiga.base import response
+from taiga.base.api import viewsets
 
 from . import permissions
 from . import services
 
 
-class SystemStatsViewSet(viewsets.ViewSet):
+CACHE_TIMEOUT = getattr(settings, "STATS_CACHE_TIMEOUT", 0)
+
+
+class BaseStatsViewSet(viewsets.ViewSet):
+    @property
+    def _cache_timeout(self):
+        return CACHE_TIMEOUT
+
+    def dispatch(self, *args, **kwargs):
+        return cache_page(self._cache_timeout)(super().dispatch)(*args, **kwargs)
+
+
+class SystemStatsViewSet(BaseStatsViewSet):
     permission_classes = (permissions.SystemStatsPermission,)
 
     def list(self, request, **kwargs):
         stats = OrderedDict()
-        stats["users"] = services.get_users_stats()
-        stats["projects"] = services.get_projects_stats()
-        stats["userstories"] = services.get_user_stories_stats()
+        stats["users"] = services.get_users_public_stats()
+        stats["projects"] = services.get_projects_public_stats()
+        stats["userstories"] = services.get_user_stories_public_stats()
         return response.Ok(stats)
 
-    def _get_cache_timeout(self):
-        return getattr(settings, "STATS_CACHE_TIMEOUT", 0)
 
-    def dispatch(self, *args, **kwargs):
-        return cache_page(self._get_cache_timeout())(super().dispatch)(*args, **kwargs)
+class DiscoverStatsViewSet(BaseStatsViewSet):
+    permission_classes = (permissions.DiscoverStatsPermission,)
+
+    def list(self, request, **kwargs):
+        stats = OrderedDict()
+        stats["projects"] = services.get_projects_discover_stats(user=request.user)
+        return response.Ok(stats)

taiga/stats/apps.py

@@ -14,7 +14,6 @@
 
 from django.apps import AppConfig
 from django.apps import apps
-from django.conf import settings
 from django.conf.urls import include, url
 
 from .routers import router
@@ -25,6 +24,5 @@ class StatsAppConfig(AppConfig):
     verbose_name = "Stats"
 
     def ready(self):
-        if settings.STATS_ENABLED:
         from taiga.urls import urlpatterns
         urlpatterns.append(url(r'^api/v1/', include(router.urls)))

taiga/stats/permissions.py

@@ -18,3 +18,7 @@ from taiga.base.api import permissions
 
 class SystemStatsPermission(permissions.TaigaResourcePermission):
     global_perms = permissions.AllowAny()
+
+
+class DiscoverStatsPermission(permissions.TaigaResourcePermission):
+    global_perms = permissions.AllowAny()

taiga/stats/routers.py

@@ -12,9 +12,16 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+from django.conf import settings
+
 from taiga.base import routers
+
 from . import api
 
 
 router = routers.DefaultRouter(trailing_slash=False)
+
+if settings.STATS_ENABLED:
     router.register(r"stats/system", api.SystemStatsViewSet, base_name="system-stats")
+
+router.register(r"stats/discover", api.DiscoverStatsViewSet, base_name="discover-stats")

taiga/stats/services.py

@@ -22,7 +22,11 @@ from datetime import timedelta
 from collections import OrderedDict
 
 
-def get_users_stats():
+###########################################################################
+# Public Stats
+###########################################################################
+
+def get_users_public_stats():
     model = apps.get_model("users", "User")
     queryset = model.objects.filter(is_active=True, is_system=False)
     stats = OrderedDict()
@@ -71,7 +75,7 @@ def get_users_stats():
     return stats
 
 
-def get_projects_stats():
+def get_projects_public_stats():
     model = apps.get_model("projects", "Project")
     queryset = model.objects.all()
     stats = OrderedDict()
@@ -109,7 +113,7 @@ def get_projects_stats():
     return stats
 
 
-def get_user_stories_stats():
+def get_user_stories_public_stats():
     model = apps.get_model("userstories", "UserStory")
     queryset = model.objects.all()
     stats = OrderedDict()
@@ -130,3 +134,20 @@ def get_user_stories_stats():
                                                        .count()) / 5
 
     return stats
+
+###########################################################################
+# Discover Stats
+###########################################################################
+
+def get_projects_discover_stats(user=None):
+    model = apps.get_model("projects", "Project")
+    queryset = model.objects.all()
+    stats = OrderedDict()
+
+    # Get Public (visible) projects
+    queryset = queryset.filter(Q(is_private=False) |
+                               Q(is_private=True, anon_permissions__contains=["view_project"]))
+
+    stats["total"] = queryset.count()
+
+    return stats