Support fields query param

gundotio · Oct 25, 2021 · f37172e · f37172e
1 parent 0bf8ae4
commit f37172e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/worf/views/list.py b/worf/views/list.py
@@ -181,6 +181,12 @@ def paginated_results(self):
         except EmptyPage:
             return []
 
+    def specific_fields(self, result):
+        fields = self.bundle.get("fields", [])
+        if fields:
+            return {key: value for key, value in result.items() if key in fields}
+        return result
+
     def parse_sort(self, fields):
         return [self.transform_sort(field) for field in fields]
 
@@ -193,7 +199,8 @@ def serialize(self):
 
         payload = {
             str(self.name): [
-                serializer.read(instance) for instance in self.paginated_results()
+                self.specific_fields(serializer.read(instance))
+                for instance in self.paginated_results()
             ]
         }