Add get project_overview to retrieve the value of the Overview tab

Author: paulnoirel

libs/labelbox/src/labelbox/schema/project.py

@@ -39,6 +39,7 @@
 from labelbox.schema.task import Task
 from labelbox.schema.task_queue import TaskQueue
 from labelbox.schema.ontology_kind import (EditorTaskType, OntologyKind)
+from labelbox.schema.project_overview import Project_Overview
 
 if TYPE_CHECKING:
     from labelbox import BulkImportRequest
@@ -1746,6 +1747,63 @@ def __check_data_rows_have_been_processed(
         return response["queryAllDataRowsHaveBeenProcessed"][
             "allDataRowsHaveBeenProcessed"]
 
+    def get_overview(self) -> Project_Overview:
+        """ Return the number of data rows per task queue, and issues of a project
+            Equivalent of the Overview tab of a project
+
+                Args:
+                    with_issues: (optional) boolean to include issues in the overview
+                Returns:
+                    Object Project_Overview
+                """
+
+        query = """query ProjectGetOverviewPyApi($projectId: ID!) {
+            project(where: { id: $projectId }) {      
+            workstreamStateCounts {
+                state
+                count
+            }
+            taskQueues {
+                queueType
+                name
+                dataRowCount
+            }
+            issues {
+                totalCount
+            }
+            completedDataRowCount
+            }
+        }
+        """
+
+        # Must use experimental to access "issues"
+        result = self.client.execute(query, {"projectId": self.uid},
+                                     experimental=True)["project"]
+
+        overview = {
+            utils.snake_case(st["state"]): st["count"]
+            for st in result.get("workstreamStateCounts")
+            if st["state"] != "NotInTaskQueue"
+        }
+
+        review_queues = {
+            tq["name"]: tq.get("dataRowCount")
+            for tq in result.get("taskQueues")
+            if tq.get("queueType") == "MANUAL_REVIEW_QUEUE"
+        }
+
+        # Store the total number of data rows in review
+        review_queues["all"] = overview.get("in_review")
+        overview["in_review"] = review_queues
+
+        overview["issues"] = result.get("issues", {}).get("totalCount")
+
+        # Rename keys
+        overview["to_label"] = overview.pop("unlabeled")
+        overview["all_in_data_rows"] = overview.pop("all")
+
+        return Project_Overview(**overview)
+
 
 class ProjectMember(DbObject):
     user = Relationship.ToOne("User", cache=True)

libs/labelbox/src/labelbox/schema/project_overview.py

@@ -0,0 +1,15 @@
+from typing import Dict, Optional
+from labelbox import pydantic_compat
+
+class Project_Overview(pydantic_compat.BaseModel):
+    """
+    Class that represents a project overview.
+    """
+    to_label: int
+    in_review: Dict[str, int]
+    in_rework: int
+    skipped: int
+    done: int
+    issues: int
+    labeled: int
+    all_in_data_rows: int

libs/labelbox/tests/integration/test_task_queue.py

@@ -1,5 +1,5 @@
 import time
-
+from typing import Dict
 from labelbox import Project
 from labelbox.schema.identifiables import GlobalKeys, UniqueIds
 
@@ -12,6 +12,19 @@ def test_get_task_queue(project: Project):
     assert review_queue
 
 
+def test_get_overview(project: Project):
+    po = project.get_overview()
+
+    assert isinstance(po.to_label, int)
+    assert isinstance(po.in_review, Dict[str, int])
+    assert isinstance(po.in_rework, int)
+    assert isinstance(po.skipped, int)
+    assert isinstance(po.done, int)
+    assert isinstance(po.issues, int)
+    assert isinstance(po.labeled, int)
+    assert isinstance(po.all_in_data_rows, int)
+
+
 def _validate_moved(project, queue_name, data_row_count):
     timeout_seconds = 30
     sleep_time = 2