Electoral data vis upgrades

hub/graphql/permissions.py

@@ -0,0 +1,14 @@
+from django.contrib.auth.models import AbstractBaseUser
+
+from hub import models
+
+
+def check_user_can_view_source(user: AbstractBaseUser | str, source_id: str):
+    external_data_source = models.ExternalDataSource.objects.get(pk=source_id)
+    permissions = models.ExternalDataSource.user_permissions(user, external_data_source)
+    if not permissions.get("can_display_points") or not permissions.get(
+        "can_display_details"
+    ):
+        raise ValueError(
+            f"User {user} does not have permission to external data source: {source_id}"
+        )

hub/graphql/schema.py

@@ -11,6 +11,7 @@
 from gqlauth.user.queries import UserQueries
 from graphql import GraphQLError
 from strawberry.extensions import QueryDepthLimiter
+from strawberry.scalars import JSON
 from strawberry.types import ExecutionContext
 from strawberry_django import mutations as django_mutations
 from strawberry_django.optimizer import DjangoOptimizerExtension
@@ -126,6 +127,16 @@ class Query(UserQueries):
     choropleth_data_for_source: List[model_types.GroupedDataCount] = (
         model_types.choropleth_data_for_source
     )
+    statistics: Optional[List[JSON]] = strawberry_django.field(
+        resolver=model_types.statistics,
+        extensions=[IsAuthenticated()],
+    )
+    statistics_for_choropleth: Optional[List[model_types.GroupedDataCount]] = (
+        strawberry_django.field(
+            resolver=model_types.statistics_for_choropleth,
+            extensions=[IsAuthenticated()],
+        )
+    )
 
     @strawberry.field
     def test_data_source(