JIRA：自動創建issue

from jira import JIRA
from jira.resources import Issue
import datetime

"""
pip install jira
"""


class JIRATool(object):
    def __init__(self, server: str, username: str, password: str, project: str = 'NET', root_issue: str = 'NET-15',
                 fix_version: str = None):
        """
        :param server:
        :param username:
        :param password:
        :param project:
        :param root_issue:
        :param fix_version:
        Usage:
            # >>> #server = 'https://jira.xpaas.lenovo.com'
            # >>> #root_issue = 'NET-834'
            # >>> fix_version = 'FY1819'
            # >>> fix_version = 'CWN1920'
            # >>> description = 'Sub-task of %s test JIRA api' % root_issue
            # >>> assignee = 'luoyj2'
            # >>> summary = 'GSD331'
            # >>> username = 'username'
            # >>> password = 'password'
            # >>> j = JIRATool(server, username, password, project='project', root_issue=root_issue, fix_version='')
            # >>> issue = j.create_issue(description, assignee, summary, self_watch=False, attachment='screenshots.py')
            # >>> for i in ['881', '882', '883', '884']:
            # >>>     iss = j.find_issue_by_id('NET-'+i)
            # >>>     print(j.delete_issue(iss))
            # >>> for k, v in j.find_issue_by_id('NET-30').raw.get('fields').items():
            # >>>     if k.startswith('customfield_'):
            # >>>         continue
            # >>>     print(k, v)
        """
        self.server = server
        self.username = username
        self.basic_auth = (username, password)
        self.project = project
        self.root_issue = root_issue
        self.fix_version = fix_version
        self.due_date = str(datetime.date.today() + datetime.timedelta(days=30))
        self.client = JIRA(server=self.server, basic_auth=self.basic_auth, options=dict(verify=False))

    def find_issue_by_id(self, issue_id: str):
        """
        :param issue_id: 'NET-15'
        :return: issue
        """
        if issue_id:
            return self.client.issue(issue_id)
        else:
            return None

    def create_issue(
            self, description: str, assignee: str, summary: str, issuetype: str = 'Sub-task', component: str = 'Devops',
            self_watch: bool = True, attachment: str = '', labels: str = None
    ):

        """
        :param description:
        :param assignee:
        :param summary:
        :param issuetype
        :param component:
        :param self_watch: True or False
        :param attachment: filename
        :param labels
        :return: issue object
        """

        issue_dict = {
            'project': {'key': self.project},
            'issuetype': {'name': issuetype},
            'parent': {'key': self.root_issue},
            'summary': summary,
            'description': description,
            'assignee': {'name': assignee},
            'priority': {'name': 'Medium'},
            'duedate': self.due_date
        }
        if self.fix_version:
            issue_dict.update(fixVersions=[{'name': self.fix_version}])
        if component:
            issue_dict.update(components=[{'name': component}])

        issue = self.client.create_issue(DataConverter(issue_dict).convert())
        if labels:
            try:
                if isinstance(labels, (list, tuple)):
                    for label in labels:
                        issue.add_field_value('labels', label)
                elif isinstance(labels, (str, bytes)):
                    issue.add_field_value('labels', labels)
            except Exception as e:
                pass

        self.client.assign_issue(issue, assignee)
        if isinstance(attachment, (str, bytes)) and os.path.exists(attachment):
            self.client.add_attachment(issue, attachment)
        if not self_watch:
            self.client.remove_watcher(issue, self.username)
            self.remove_watcher(issue, self.username)
        return issue

    def delete_issue(self, issue: Issue):
        """
        :param issue: 'NET-15' or jira.resources.Issue object
        :return:
        """
        if not isinstance(issue, Issue):
            issue = self.find_issue_by_id(issue)
        if issue:
            issue.delete()
            return True
        return False

    def add_watcher(self, issue: Issue, watchers: list):
        """
        :param issue: 'NET-15' or jira.resources.Issue object
        :param watchers: ['xxx', 'xxx'] or 'xxx'
        :return:
        """
        if issue and watchers:
            if isinstance(watchers, (str, bytes)):
                watchers = [watchers]
            for watcher in watchers:
                if self.client.search_users(user=watcher):
                    self.client.add_watcher(issue=issue, watcher=watcher)

    def remove_watcher(self, issue: Issue, watchers: list):
        """
        :param issue: 'NET-15' or jira.resources.Issue object
        :param watchers: ['xxx', 'xxx'] or 'xxx'
        :return:
        """
        if isinstance(watchers, (str, bytes)):
            watchers = [watchers]
        issue_watchers = self.client.watchers(issue)
        for watcher in watchers:
            try:
                issue_watchers.delete(watcher)
            except:
                pass