Added taskcount filters

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -2435,7 +2435,7 @@ def get_labeling_service_dashboards(
             >>> seven_days_ago = (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d")
             >>> workforce_requested_filter_before = WorkforceRequestedDateFilter(
             >>>     operation=OperationType.WorforceRequestedDate,
-            >>>     value=DateValue(operator=DateOperator.GreaterThanOrEqual,
+            >>>     value=DateValue(operator=RangeOperatorWithSingleValue.GreaterThanOrEqual,
             >>>                     value=seven_days_ago))
             >>> labeling_service_dashboard = [ld for ld in project.client.get_labeling_service_dashboards(search_query=[workforce_requested_filter_before])]
 

libs/labelbox/src/labelbox/schema/search_filters.py

@@ -36,6 +36,8 @@ class OperationType(Enum):
     Stage = 'stage'
     WorforceRequestedDate = 'workforce_requested_at'
     WorkforceStageUpdatedDate = 'workforce_stage_updated_at'
+    TaskCompletedCount = 'task_completed_count'
+    TaskRemainingCount = 'task_remaining_count'
 
 
 class IdOperator(Enum):
@@ -45,7 +47,7 @@ class IdOperator(Enum):
     Is = 'is'
 
 
-class DateOperator(Enum):
+class RangeOperatorWithSingleValue(Enum):
     """
     Supported operators for dates
     """
@@ -54,7 +56,7 @@ class DateOperator(Enum):
     LessThanOrEqual = 'LESS_THAN_OR_EQUAL'
 
 
-class DateRangeOperator(Enum):
+class RangeOperatorWithValue(Enum):
     """
     Supported operators for date ranges
     """
@@ -109,10 +111,15 @@ class DateValue(BaseSearchFilter):
             so for a string '2024-01-01' that is run on a computer in PST, we would convert it to '2024-01-01T08:00:00Z'
             while the same string in EST will get converted to '2024-01-01T05:00:00Z'
     """
-    operator: DateOperator
+    operator: RangeOperatorWithSingleValue
     value: datetime.datetime
 
 
+class IntegerValue(BaseSearchFilter):
+    operator: RangeOperatorWithSingleValue
+    value: int
+
+
 class WorkforceStageUpdatedFilter(BaseSearchFilter):
     """
     Filter for workforce stage updated date
@@ -141,7 +148,7 @@ class DateRangeValue(BaseSearchFilter):
     """
     Date range value for a search filter 
     """
-    operator: DateRangeOperator
+    operator: RangeOperatorWithValue
     value: DateRange
 
 
@@ -161,11 +168,28 @@ class WorkforceStageUpdatedRangeFilter(BaseSearchFilter):
     value: DateRangeValue
 
 
+class TaskCompletedCountFilter(BaseSearchFilter):
+    """
+    Filter for workforce stage updated date
+    """
+    operation: Literal[OperationType.TaskCompletedCount]
+    value: IntegerValue
+
+
+class TaskRemainingCountFilter(BaseSearchFilter):
+    """
+    Filter for workforce stage updated date
+    """
+    operation: Literal[OperationType.TaskRemainingCount]
+    value: IntegerValue
+
+
 SearchFilter = Union[OrganizationFilter, WorkspaceFilter, TagFilter,
                      ProjectStageFilter, WorkforceRequestedDateFilter,
                      WorkforceStageUpdatedFilter,
                      WorkforceRequestedDateRangeFilter,
-                     WorkforceStageUpdatedRangeFilter]
+                     WorkforceStageUpdatedRangeFilter, TaskCompletedCountFilter,
+                     TaskRemainingCountFilter]
 
 
 def _dict_to_graphql_string(d: Union[dict, list, str, int]) -> str:

libs/labelbox/tests/integration/test_labeling_dashboard.py

@@ -1,8 +1,8 @@
 from datetime import datetime, timedelta
 from labelbox.schema.labeling_service import LabelingServiceStatus
-from labelbox.schema.search_filters import DateOperator, DateRange, DateRangeOperator, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter
 from labelbox.schema.ontology_kind import EditorTaskType
 from labelbox.schema.media_type import MediaType
+from labelbox.schema.search_filters import RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter
 
 
 def test_request_labeling_service_dashboard(rand_gen,
@@ -48,12 +48,13 @@ def test_request_labeling_service_dashboard_filters(requested_labeling_service):
 
     workforce_requested_filter_before = WorkforceRequestedDateFilter(
         operation=OperationType.WorforceRequestedDate,
-        value=DateValue(operator=DateOperator.GreaterThanOrEqual,
-                        value=datetime.strptime("2024-01-01", "%Y-%m-%d")))
+        value=DateValue(
+            operator=RangeOperatorWithSingleValue.GreaterThanOrEqual,
+            value=datetime.strptime("2024-01-01", "%Y-%m-%d")))
     year_from_now = (datetime.now() + timedelta(days=365))
     workforce_requested_filter_after = WorkforceRequestedDateFilter(
         operation=OperationType.WorforceRequestedDate,
-        value=DateValue(operator=DateOperator.LessThanOrEqual,
+        value=DateValue(operator=RangeOperatorWithSingleValue.LessThanOrEqual,
                         value=year_from_now))
 
     labeling_service_dashboard = [
@@ -66,7 +67,7 @@ def test_request_labeling_service_dashboard_filters(requested_labeling_service):
 
     workforce_date_range_filter = WorkforceRequestedDateRangeFilter(
         operation=OperationType.WorforceRequestedDate,
-        value=DateRangeValue(operator=DateRangeOperator.Between,
+        value=DateRangeValue(operator=RangeOperatorWithValue.Between,
                              value=DateRange(min="2024-01-01T00:00:00-0800",
                                              max=year_from_now)))
 

libs/labelbox/tests/unit/test_unit_search_filters.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from labelbox.schema.search_filters import DateOperator, DateRange, DateRangeOperator, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, ProjectStageFilter, TagFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkforceStageUpdatedFilter, WorkforceStageUpdatedRangeFilter, WorkspaceFilter, build_search_filter
+from labelbox.schema.search_filters import IntegerValue, RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, ProjectStageFilter, TagFilter, TaskCompletedCountFilter, TaskRemainingCountFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkforceStageUpdatedFilter, WorkforceStageUpdatedRangeFilter, WorkspaceFilter, build_search_filter
 from labelbox.utils import format_iso_datetime
 
 
@@ -31,12 +31,14 @@ def test_date_filters():
     filters = [
         WorkforceRequestedDateFilter(
             operation=OperationType.WorforceRequestedDate,
-            value=DateValue(operator=DateOperator.GreaterThanOrEqual,
-                            value=local_time_start)),
+            value=DateValue(
+                operator=RangeOperatorWithSingleValue.GreaterThanOrEqual,
+                value=local_time_start)),
         WorkforceStageUpdatedFilter(
             operation=OperationType.WorkforceStageUpdatedDate,
-            value=DateValue(operator=DateOperator.LessThanOrEqual,
-                            value=local_time_end)),
+            value=DateValue(
+                operator=RangeOperatorWithSingleValue.LessThanOrEqual,
+                value=local_time_end)),
     ]
     expected_start = format_iso_datetime(local_time_start)
     expected_end = format_iso_datetime(local_time_end)
@@ -49,7 +51,7 @@ def test_date_range_filters():
     filters = [
         WorkforceRequestedDateRangeFilter(
             operation=OperationType.WorforceRequestedDate,
-            value=DateRangeValue(operator=DateRangeOperator.Between,
+            value=DateRangeValue(operator=RangeOperatorWithValue.Between,
                                  value=DateRange(min=datetime.strptime(
                                      "2024-01-01T00:00:00-0800",
                                      "%Y-%m-%dT%H:%M:%S%z"),
@@ -58,7 +60,7 @@ def test_date_range_filters():
                                                      "%Y-%m-%dT%H:%M:%S%z")))),
         WorkforceStageUpdatedRangeFilter(
             operation=OperationType.WorkforceStageUpdatedDate,
-            value=DateRangeValue(operator=DateRangeOperator.Between,
+            value=DateRangeValue(operator=RangeOperatorWithValue.Between,
                                  value=DateRange(min=datetime.strptime(
                                      "2024-01-01T00:00:00-0800",
                                      "%Y-%m-%dT%H:%M:%S%z"),
@@ -69,3 +71,21 @@ def test_date_range_filters():
     assert build_search_filter(
         filters
     ) == '[{value: {operator: "BETWEEN", value: {min: "2024-01-01T08:00:00Z", max: "2025-01-01T08:00:00Z"}}, type: "workforce_requested_at"}, {value: {operator: "BETWEEN", value: {min: "2024-01-01T08:00:00Z", max: "2025-01-01T08:00:00Z"}}, type: "workforce_stage_updated_at"}]'
+
+
+def test_task_count_filters():
+    filters = [
+        TaskCompletedCountFilter(
+            operation=OperationType.TaskCompletedCount,
+            value=IntegerValue(
+                operator=RangeOperatorWithSingleValue.GreaterThanOrEqual,
+                value=1)),
+        TaskRemainingCountFilter(
+            operation=OperationType.TaskRemainingCount,
+            value=IntegerValue(
+                operator=RangeOperatorWithSingleValue.LessThanOrEqual,
+                value=10)),
+    ]
+
+    expected = '[{value: {operator: "GREATER_THAN_OR_EQUAL", value: 1}, type: "task_completed_count"}, {value: {operator: "LESS_THAN_OR_EQUAL", value: 10}, type: "task_remaining_count"}]'
+    assert build_search_filter(filters) == expected