CTX-5789: Updated dataset sample fetching

coretex/entities/dataset/network_dataset.py

@@ -132,25 +132,41 @@ def _keyDescriptors(cls) -> Dict[str, KeyDescriptor]:
 
         return descriptors
 
-    # NetworkObject overrides
-
     @classmethod
-    def _endpoint(cls) -> str:
-        return "dataset"
+    def decode(cls, encodedObject: Dict[str, Any]) -> Self:
+        obj = super().decode(encodedObject)
 
-    @classmethod
-    def fetchById(cls, objectId: int, **kwargs: Any) -> Self:
-        if "include_sessions" not in kwargs:
-            kwargs["include_sessions"] = 1
+        page = 1
+        hasMorePages = True
+        samples: List[SampleType] = []
 
-        return super().fetchById(objectId, **kwargs)
+        while hasMorePages:
+            params = {
+                "dataset_id": obj.id,
+                "page": page
+            }
+            response = networkManager.get(f"{cls._endpoint()}/sessions", params)
 
-    @classmethod
-    def fetchAll(cls, **kwargs: Any) -> List[Self]:
-        if "include_sessions" not in kwargs:
-            kwargs["include_sessions"] = 1
+            if response.hasFailed():
+                raise NetworkRequestError(response, "Failed to fetch Dataset Samples")
+
+            responseJson = response.getJson(dict)
+            samples.extend([obj._sampleType.decode(sample) for sample in responseJson["data"]])
+
+            metadata = responseJson.get("metadata", {})
+            if page == metadata["total_pages"]:
+                hasMorePages = False
 
-        return super().fetchAll(**kwargs)
+            page += 1
+
+        obj.samples = samples
+        return obj
+
+    # NetworkObject overrides
+
+    @classmethod
+    def _endpoint(cls) -> str:
+        return "dataset"
 
     @classmethod
     def fetchCachedDataset(cls, dependencies: List[str]) -> Self: