Jira: Add error handler for method get issue security field scheme method

Author: Gonchik Tsymzhitov

atlassian/jira.py

@@ -1,7 +1,17 @@
 # coding=utf-8
 import logging
 import re
+
+from requests import HTTPError
+
 from .rest_client import AtlassianRestAPI
+from .errors import (
+    ApiError,
+    ApiNotFoundError,
+    ApiPermissionError,
+    ApiValueError,
+    ApiConflictError
+)
 
 log = logging.getLogger(__name__)
 
@@ -1730,11 +1740,26 @@ def get_project_issue_security_scheme(self, project_id_or_key, only_levels=False
         :return: list
         """
         url = 'rest/api/2/project/{}/issuesecuritylevelscheme'.format(project_id_or_key)
-
-        if only_levels is True:
-            return self.get(url).get('levels')
-        else:
-            return self.get(url)
+        response = None
+        try:
+            response = self.get(url)
+        except HTTPError as e:
+            if e.response.status_code == 401:
+                raise ApiPermissionError(
+                    "Returned if the user is not logged in.",
+                    reason=e)
+            elif e.response.status_code == 403:
+                raise ApiPermissionError(
+                    "Returned if the project is visible for calling user, but the user doesn't have administrative permissions",
+                    reason=e)
+            elif e.response.status_code == 404:
+                raise ApiNotFoundError(
+                    "Returned if the project does not exist, or is not visible to the calling user",
+                    reason=e)
+            raise
+        if only_levels is True and response:
+            return response.get('levels') or None
+        return response
 
     # Priority Schemes
     def get_all_priority_schemes(self, start=0, limit=100, expand=None):

examples/jira/jira-add-issues-to-sprint.py

@@ -9,7 +9,5 @@
     username='admin',
     password='admin')
 
-resp = jira.add_issues_to_sprint(
-                                    sprint_id=sprint_id,
-                                    issues=issues_lst
-                                )
+resp = jira.add_issues_to_sprint(sprint_id=sprint_id,
+                                 issues=issues_lst)

examples/jira/jira-project-security-level-checker.py

@@ -0,0 +1,24 @@
+# coding=utf-8
+import logging
+from atlassian import Jira
+from pprint import pprint
+
+""" How to get server info with health check"""
+
+jira = Jira(
+    url="https://jira.example.com/",
+    username='gonchik.tsymzhitov',
+    password='********')
+
+log = logging.getLogger("com.gonchik.python.scripts.example")
+logging.basicConfig(level=logging.ERROR)
+
+projects = jira.get_all_projects()
+for project in projects:
+    project_key = project.get("key")
+    try:
+        value = jira.get_project_issue_security_scheme(project_key).get("name") or "None"
+    except Exception as e:
+        log.error(e)
+        value = "None"
+    print(project_key + " has issue security scheme " + value)