Jira: Add workflow association for projects

Author: gonchik

atlassian/jira/jira_cloud.py

@@ -94,6 +94,36 @@ def enhanced_jql_get_list_of_tickets(
                 break
         return results
 
+    def get_project_workflow_scheme_associations(self, project_ids):
+        """Return workflow-scheme associations for one or more project IDs.
+
+        Team-managed and non-existent projects are omitted by Jira Cloud. The
+        caller requires the Administer Jira global permission.
+
+        :param project_ids: Project ID or iterable of project IDs.
+        :return: Jira's workflow-scheme association container.
+        """
+        if isinstance(project_ids, (str, int)):
+            project_ids = [project_ids]
+        else:
+            project_ids = list(project_ids)
+        if not project_ids:
+            raise ValueError("project_ids must contain at least one project ID")
+        return self.get(self.endpoint("workflowscheme/project", api_version=3), params={"projectId": project_ids})
+
+    def assign_project_workflow_scheme(self, project_id, workflow_scheme_id):
+        """Assign a workflow scheme to a classic Jira Cloud project.
+
+        Jira only permits this operation when the project has no issues. The
+        caller requires the Administer Jira global permission.
+
+        :param project_id: Numeric Jira project ID.
+        :param workflow_scheme_id: Workflow scheme ID.
+        :return: Jira response (normally ``None`` for HTTP 204).
+        """
+        data = {"projectId": str(project_id), "workflowSchemeId": str(workflow_scheme_id)}
+        return self.put(self.endpoint("workflowscheme/project", api_version=3), data=data)
+
 
 class JiraSoftware(JiraSoftwareMethods, AtlassianRestAPI):
     """Jira Software Cloud REST APIs.

docs/jira_cloud.rst

@@ -105,3 +105,18 @@ Core generated operations use the selected ``JiraCloud(api_version=2|3)``
 version in their route.  Existing snake_case ``Jira`` and ``ServiceDesk``
 methods retain their endpoint selection and behavior; they are neither renamed
 nor overwritten by the generated Cloud surface.
+Workflow scheme project associations
+------------------------------------
+
+Use :class:`atlassian.jira.JiraCloud` to inspect or assign classic-project
+workflow schemes. Team-managed projects are not returned by Jira for this
+endpoint. Both operations require the Administer Jira global permission.
+
+.. code-block:: python
+
+    from atlassian.jira import JiraCloud
+
+    jira = JiraCloud("https://your-domain.atlassian.net", username=email, password=api_token)
+
+    associations = jira.get_project_workflow_scheme_associations(["10001", "10002"])
+    jira.assign_project_workflow_scheme(project_id="10001", workflow_scheme_id="10032")

tests/test_jira_cloud_clients.py

@@ -16,6 +16,22 @@ def test_legacy_clients_keep_their_defaults(self):
         self.assertIs(Jira, JiraServer)
         self.assertIs(PackageJira, JiraServer)
 
+    def test_jira_cloud_workflow_scheme_association_helpers_use_v3(self):
+        jira = JiraCloud("https://example.atlassian.net")
+
+        with patch.object(jira, "get", return_value={"values": []}) as get:
+            result = jira.get_project_workflow_scheme_associations(["10001", 10002])
+
+        self.assertEqual(result, {"values": []})
+        get.assert_called_once_with("rest/api/3/workflowscheme/project", params={"projectId": ["10001", 10002]})
+
+        with patch.object(jira, "put", return_value=None) as put:
+            jira.assign_project_workflow_scheme("10001", 10032)
+
+        put.assert_called_once_with(
+            "rest/api/3/workflowscheme/project", data={"projectId": "10001", "workflowSchemeId": "10032"}
+        )
+
     def test_core_client_defaults_to_v3_and_forces_cloud_mode(self):
         jira = JiraCloud("https://example.atlassian.net", cloud=False)