fix: add item_counts to ProjectSerializer for new sidebar design

OpenGeoscience · Sep 30, 2024 · d5db4f1 · d5db4f1
1 parent 4413524
commit d5db4f1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/uvdat/core/rest/serializers.py b/uvdat/core/rest/serializers.py
@@ -33,6 +33,7 @@ class ProjectSerializer(serializers.ModelSerializer):
     owner = serializers.SerializerMethodField('get_owner')
     collaborators = serializers.SerializerMethodField('get_collaborators')
     followers = serializers.SerializerMethodField('get_followers')
+    item_counts = serializers.SerializerMethodField('get_item_counts')
 
     def get_center(self, obj):
         # Web client expects Lon, Lat
@@ -53,6 +54,14 @@ def get_followers(self, obj):
         users = get_users_with_perms(obj, only_with_perms_in=['follower'])
         return [UserSerializer(user).data for user in users.all()]
 
+    def get_item_counts(self, obj):
+        return {
+            'datasets': obj.datasets.count(),
+            'regions': obj.derived_regions.count(),
+            'charts': obj.charts.count(),
+            'simulations': obj.simulation_results.count(),
+        }
+
     def to_internal_value(self, data):
         center = data.get('default_map_center')
         data = super().to_internal_value(data)