PYAPI-42 Jira: Add statuses, issues, projects, transitions

Author: astromatt

atlassian/__init__.py

@@ -32,22 +32,32 @@ def request(self, method='GET', path='/', data=None,
             data=json.dumps(data),
             auth=(self.username, self.password),
             timeout=60)
-        if response.status_code != 200:
-            self.log_curl_debug(method=method, path=path, headers=headers, data=data, level=logging.WARNING)
-            log.warning(response.json())
-            response.raise_for_status()
-        else:
+        if response.status_code == 200:
             log.debug('Received: {0}'.format(response.json()))
+        elif response.status_code == 204:
+            log.debug('Received "204 No Content" response')
+        else:
+            self.log_curl_debug(method=method, path=path, headers=headers, data=data, level=logging.DEBUG)
+            log.info(response.json())
+            response.raise_for_status()
         return response
 
     def get(self, path, data=None, headers={'Content-Type': 'application/json', 'Accept': 'application/json'}):
         return self.request('GET', path=path, data=data, headers=headers).json()
 
     def post(self, path, data=None, headers={'Content-Type': 'application/json', 'Accept': 'application/json'}):
-        return self.request('POST', path=path, data=data, headers=headers).json()
+        try:
+            return self.request('POST', path=path, data=data, headers=headers).json()
+        except ValueError:
+            log.debug('Received response with no content')
+            return None
 
     def put(self, path, data=None, headers={'Content-Type': 'application/json', 'Accept': 'application/json'}):
-        return self.request('PUT', path=path, data=data, headers=headers).json()
+        try:
+            return self.request('PUT', path=path, data=data, headers=headers).json()
+        except ValueError:
+            log.debug('Received response with no content')
+            return None
 
     def delete(self, path, data=None, headers={'Content-Type': 'application/json', 'Accept': 'application/json'}):
         return self.request('DELETE', path=path, data=data, headers=headers).json()

atlassian/jira.py

@@ -1,4 +1,5 @@
 import logging
+from requests.exceptions import HTTPError
 from atlassian import AtlassianRestAPI
 
 
@@ -28,14 +29,14 @@ def user(self, username):
     def project(self, key):
         return self.get('/rest/api/2/project/{0}'.format(key))
 
-    def issue(self, key):
-        return self.get('/rest/api/2/issue/{0}'.format(key))
+    def issue(self, key, fields='*all'):
+        return self.get('/rest/api/2/issue/{0}?fields={1}'.format(key, fields))
 
     def issue_field_value(self, key, field):
         issue = self.get('/rest/api/2/issue/{0}?fields={1}'.format(key, field))
         return issue['fields'][field]
 
-    def update_issue_field(self, key, fields):
+    def update_issue_field(self, key, fields='*all'):
         return self.put('/rest/api/2/issue/{0}'.format(key), data={'fields': fields})
 
     def project_leaders(self):
@@ -55,3 +56,92 @@ def rename_sprint(self, sprint_id, name, start_date, end_date):
             'name': name,
             'startDate': start_date,
             'endDate': end_date})
+
+    def get_project_issuekey_last(self, project):
+        jql = 'project = {project} ORDER BY issuekey DESC'.format(project=project)
+        return self.jql(jql)['issues'][0]['key']
+
+    def get_project_issuekey_all(self, project):
+        jql = 'project = {project} ORDER BY issuekey ASC'.format(project=project)
+        return [issue['key'] for issue in self.jql(jql)['issues']]
+
+    def get_project_issues_count(self, project):
+        jql = 'project = {project}'.format(project=project)
+        return self.jql(jql, fields='*none')['total']
+
+    def get_all_project_issues(self, project, fields='*all'):
+        jql = 'project = {project} ORDER BY key'.format(project=project)
+        return self.jql(jql, fields=fields)['issues']
+
+    def issue_exists(self, issuekey):
+        try:
+            self.issue(issuekey, fields='*none')
+            log.info('Issue "{issuekey}" exists'.format(issuekey=issuekey))
+            return True
+        except HTTPError as e:
+            if e.response.status_code == 404:
+                log.info('Issue "{issuekey}" does not exists'.format(issuekey=issuekey))
+                return False
+            else:
+                log.info('Issue "{issuekey}" existsted, but now it\'s deleted'.format(issuekey=issuekey))
+                return True
+
+    def issue_deleted(self, issuekey):
+        try:
+            self.issue(issuekey, fields='*none')
+            log.info('Issue "{issuekey}" is not deleted'.format(issuekey=issuekey))
+            return False
+        except HTTPError:
+            log.info('Issue "{issuekey}" is deleted'.format(issuekey=issuekey))
+            return True
+
+    def issue_update(self, issuekey, fields):
+        log.warning('Updating issue "{issuekey}" with "{summary}"'.format(issuekey=issuekey, summary=fields['summary']))
+        url = '/rest/api/2/issue/{0}'.format(issuekey)
+        return self.put(url, data={'fields': fields})
+
+    def issue_create(self, fields):
+        log.warning('Creating issue "{summary}"'.format(summary=fields['summary']))
+        url = '/rest/api/2/issue/'
+        return self.post(url, data={'fields': fields})
+
+    def issue_create_or_update(self, fields):
+        issuekey = fields.get('issuekey', None)
+
+        if not issuekey or not self.issue_exists(issuekey):
+            log.info('Issuekey is not provided or does not exists in destination. Will attempt to create an issue')
+            del fields['issuekey']
+            return self.issue_create(fields)
+
+        if self.issue_deleted(issuekey):
+            log.warning('Issue "{issuekey}" deleted, skipping'.format(issuekey=issuekey))
+            return None
+
+        log.info('Issue "{issuekey}" exists, will update'.format(issuekey=issuekey))
+        del fields['issuekey']
+        return self.issue_update(issuekey, fields)
+
+    def get_issue_transitions(self, issuekey):
+        url = '/rest/api/2/issue/{issuekey}?expand=transitions.fields&fields=status'.format(issuekey=issuekey)
+        return [{'name': transition['name'], 'id': int(transition['id']), 'to': transition['to']['name']} for transition in self.get(url)['transitions']]
+
+    def get_status_id_from_name(self, status_name):
+        url = '/rest/api/2/status/{name}'.format(name=status_name)
+        return int(self.get(url)['id'])
+
+    def get_transition_id_to_status_name(self, issuekey, status_name):
+        for transition in self.get_issue_transitions(issuekey):
+            if status_name == transition['to']:
+                return int(transition['id'])
+
+    def issue_transition(self, issuekey, status):
+        return self.set_issue_status(issuekey, status)
+
+    def set_issue_status(self, issuekey, status_name):
+        url = '/rest/api/2/issue/{issuekey}/transitions'.format(issuekey=issuekey)
+        transition_id = self.get_transition_id_to_status_name(issuekey, status_name)
+        return self.post(url, data={'transition': {'id': transition_id}})
+
+    def get_issue_status(self, issuekey):
+        url = '/rest/api/2/issue/{issuekey}?fields=status'.format(issuekey=issuekey)
+        return self.get(url)['fields']['status']['name']

setup.cfg

@@ -1,4 +1,4 @@
 [pep8]
 max-line-length = 939
-ignore = E402
+ignore = E402,W391
 

setup.py

@@ -10,7 +10,7 @@
     description='Python Atlassian REST API Wrapper',
     long_description='Python Atlassian REST API Wrapper',
     license='Apache License 2.0',
-    version='0.12.5',
+    version='0.13.1',
     download_url='https://github.com/MattAgile/atlassian-python-api',
 
     author='Matt Harasymczuk',