Skip to content
Snippets Groups Projects

Dataset sets management API endpoints

Merged ml bonhomme requested to merge sets-mgmt-api into master
All threads resolved!
3 files
+ 7
4
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -284,9 +284,10 @@ class DatasetSetsCountField(serializers.DictField):
def get_attribute(self, instance):
if not self.context.get("sets_count", True):
return None
elts_count = {k.name: 0 for k in instance.sets.all()}
dataset_sets = instance.sets.all()
elts_count = {k.name: 0 for k in dataset_sets}
elts_count.update(
DatasetElement.objects.filter(set_id__in=instance.sets.values_list("id"))
DatasetElement.objects.filter(set_id__in=[ds.id for ds in dataset_sets])
.values("set__name")
.annotate(count=Count("id"))
.values_list("set__name", "count")
Loading