Skip to content

Commit 585da03

Browse files
committed
add dataset count to collaborative document
1 parent a60f427 commit 585da03

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

search/documents/collaborative_document.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ class CollaborativeDocument(Document):
152152
platform_url = fields.TextField(analyzer=ngram_analyser)
153153
started_on = fields.DateField()
154154
completed_on = fields.DateField()
155+
dataset_count = fields.IntegerField()
155156

156157
def prepare_metadata(self, instance: Collaborative) -> List[Dict[str, Any]]:
157158
processed_metadata: List[Dict[str, Any]] = []
@@ -255,6 +256,9 @@ def prepare_cover_image(self, instance: Collaborative) -> str:
255256
return str(instance.cover_image.path.replace("/code/files/", ""))
256257
return ""
257258

259+
def prepare_dataset_count(self, instance: Collaborative) -> int:
260+
return instance.datasets.count()
261+
258262
def should_index_object(self, obj: Collaborative) -> bool:
259263
return obj.status == CollaborativeStatus.PUBLISHED
260264

0 commit comments

Comments
 (0)