feat: Add new method to update project workflow (#1836)

Author: florianlega

docs/sdk/project_workflow.md

@@ -0,0 +1,3 @@
+# Project Workflow module
+
+::: kili.presentation.client.project_workflow.ProjectWorkflowClientMethods

mkdocs.yml

@@ -28,6 +28,7 @@ nav:
           - Project: sdk/project.md
           - Project User: sdk/project_user.md
           - Project Version: sdk/project_version.md
+          - Project Workflow: sdk/project_workflow.md
           - Tag: sdk/tag.md
           - User: sdk/user.md
   - Command Line Interface:

src/kili/adapters/kili_api_gateway/kili_api_gateway.py

@@ -17,6 +17,9 @@
     OrganizationOperationMixin,
 )
 from kili.adapters.kili_api_gateway.project.operations_mixin import ProjectOperationMixin
+from kili.adapters.kili_api_gateway.project_workflow.operations_mixin import (
+    ProjectWorkflowOperationMixin,
+)
 from kili.adapters.kili_api_gateway.tag import TagOperationMixin
 from kili.adapters.kili_api_gateway.user.operation_mixin import UserOperationMixin
 from kili.core.graphql.graphql_client import GraphQLClient
@@ -32,6 +35,7 @@ class KiliAPIGateway(
     NotificationOperationMixin,
     OrganizationOperationMixin,
     ProjectOperationMixin,
+    ProjectWorkflowOperationMixin,
     TagOperationMixin,
     UserOperationMixin,
     EventOperationMixin,

src/kili/adapters/kili_api_gateway/project_workflow/__init__.py

@@ -0,0 +1,12 @@
+"""GraphQL payload data mappers for project operations."""
+
+from typing import Dict
+
+from .types import ProjectWorkflowDataKiliAPIGatewayInput
+
+
+def project_input_mapper(data: ProjectWorkflowDataKiliAPIGatewayInput) -> Dict:
+    """Build the GraphQL ProjectWorfklowData variable to be sent in an operation."""
+    return {
+        "enforceStepSeparation": data.enforce_step_separation,
+    }

src/kili/adapters/kili_api_gateway/project_workflow/mappers.py

@@ -0,0 +1,12 @@
+"""GraphQL Project Workflow operations."""
+
+
+def get_update_project_workflow_mutation(fragment: str) -> str:
+    """Return the GraphQL editProjectWorkflowSettings mutation."""
+    return f"""
+        mutation editProjectWorkflowSettings($input: EditProjectWorkflowSettingsInput!) {{
+            data: editProjectWorkflowSettings(input: $input) {{
+                {fragment}
+            }}
+        }}
+        """

src/kili/adapters/kili_api_gateway/project_workflow/operations.py

@@ -0,0 +1,37 @@
+"""Mixin extending Kili API Gateway class with Projects related operations."""
+
+from typing import Dict
+
+from kili.adapters.kili_api_gateway.base import BaseOperationMixin
+from kili.adapters.kili_api_gateway.helpers.queries import (
+    fragment_builder,
+)
+from kili.domain.project import ProjectId
+
+from .mappers import project_input_mapper
+from .operations import (
+    get_update_project_workflow_mutation,
+)
+from .types import ProjectWorkflowDataKiliAPIGatewayInput
+
+
+class ProjectWorkflowOperationMixin(BaseOperationMixin):
+    """Mixin extending Kili API Gateway class with Projects workflow related operations."""
+
+    def update_project_workflow(
+        self,
+        project_id: ProjectId,
+        project_workflow_data: ProjectWorkflowDataKiliAPIGatewayInput,
+    ) -> Dict:
+        """Update properties in a project workflow."""
+        project_workflow_input = project_input_mapper(data=project_workflow_data)
+
+        fields = tuple(name for name, val in project_workflow_input.items() if val is not None)
+        fragment = fragment_builder(fields)
+        mutation = get_update_project_workflow_mutation(fragment)
+
+        project_workflow_input["projectId"] = project_id
+
+        variables = {"input": project_workflow_input}
+        result = self.graphql_client.execute(mutation, variables)
+        return result["data"]

src/kili/adapters/kili_api_gateway/project_workflow/operations_mixin.py

@@ -0,0 +1,11 @@
+"""Types for the ProjectWorkflow-related Kili API gateway functions."""
+
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class ProjectWorkflowDataKiliAPIGatewayInput:
+    """ProjectWorkflow input data for Kili API Gateway."""
+
+    enforce_step_separation: Optional[bool]

src/kili/adapters/kili_api_gateway/project_workflow/types.py

@@ -32,6 +32,7 @@
 from kili.presentation.client.notification import NotificationClientMethods
 from kili.presentation.client.organization import OrganizationClientMethods
 from kili.presentation.client.project import ProjectClientMethods
+from kili.presentation.client.project_workflow import ProjectWorkflowClientMethods
 from kili.presentation.client.tag import TagClientMethods
 from kili.presentation.client.user import UserClientMethods
 from kili.use_cases.api_key import ApiKeyUseCases
@@ -68,6 +69,7 @@ class Kili(  # pylint: disable=too-many-ancestors,too-many-instance-attributes
     NotificationClientMethods,
     OrganizationClientMethods,
     ProjectClientMethods,
+    ProjectWorkflowClientMethods,
     TagClientMethods,
     UserClientMethods,
 ):

src/kili/client.py

@@ -212,7 +212,6 @@ def assign_assets_to_labelers(
         external_ids: Optional[List[str]] = None,
         project_id: Optional[str] = None,
     ) -> List[Dict[str, Any]]:
-        # pylint: disable=line-too-long
         """Assign a list of assets to a list of labelers.
 
         Args: