jirasync.py

import click
import os
from passlib.context import CryptContext
from wrappers.jirawrapper import MyJiraWrapper
from plugins.jira_workflow import (
    start_issue_workflow,
    start_create_pull_requests_workflow,
    start_review_pull_requests_workflow
)

from utils.utils import (
    echo_success,
    echo_error,
    update_yaml,
    configure_interval,
    get_yaml_data
)
from plugins.github import (
    GitHubPlugin,
)


class Config(object):

    def __init__(self):
        self.config_file = "config.yaml"
        self.pwd_context = CryptContext(
            schemes=["pbkdf2_sha256"],
            default="pbkdf2_sha256",
            pbkdf2_sha256__default_rounds=30000
        )


pass_config = click.make_pass_decorator(Config, ensure=True)


def encrypt_password(pwd_context, password):
    return pwd_context.encrypt(password)


def check_encrypted_password(pwd_context, password, hashed):
    return pwd_context.verify(password, hashed)


def set_config_properties(git_urls, config_dict):
    content = {}
    git_urls = {'git_urls': str(git_urls).encode().split(',')}
    content.update(git_urls)
    if config_dict is None:
        config_dict = dict()
    for key, value in config_dict.items():
        content.update({key: value})
    return content


def formatting(header, github_list):
    echo_error("=================================")
    echo_error(header)
    echo_error("=================================")
    if len(github_list) == 0:
        echo_error("No new item found !")
    else:
        for item in github_list:
            echo_error(item.data['url'])
    print("\n")


@click.group()
@click.option('--debug/--no-debug', default=False)
def cli(debug):
    """This CLI tool to sync and track GitHub issues, pr's, pr_reviews
     with Jira. Currently supported to only for create, update and
     close status.
    """
    pass


@cli.command()
@click.option('--git-urls', default=None, prompt=True,
              help="Pass the git project names for Jira sync e.g. airgun, robottelo")
@click.option('--jira-url', prompt=True, confirmation_prompt=False,
              help="Pass the Jira url")
@click.option('--github-username', default=None, prompt=True,
              help="Pass GitHub username")
@click.option('--jira-username', default=None, prompt=True,
              help="Pass Jira username")
@click.option('--jira-email-id', default=None, prompt=True,
              help="Pass Jira Email Id")
@click.option('--jira-board', prompt=True, confirmation_prompt=False,
              help='Pass the Jira board name')
@click.option('--jira-project', prompt=True, confirmation_prompt=False,
              help='Pass the Jira project name')
@pass_config
def set_config(config, git_urls, jira_url, github_username, jira_username,
               jira_email_id, jira_board, jira_project):
    """Set git_urls and github username"""
    if os.path.exists(config.config_file):
        os.unlink(config.config_file)
    try:
        if None not in (git_urls, github_username, jira_username, jira_email_id,
                        jira_board, jira_project, jira_url):
            auth_dict = {
                'kerberos': True,
                'basic_auth': False,
                'username': str(jira_username).encode()
            }
            content_dict = {
                'url': str(jira_url).encode(),
                'github_username': str(github_username).encode(),
                'board': str(jira_board).encode(),
                'project': str(jira_project).encode(),
                'email_id': str(jira_email_id).encode(),
                'auth': auth_dict,
                'ca_cert_path': "/etc/ssl/certs/ca-bundle.crt",
                'verify_ssl': True,
                'label_check': False,
                'check_for_updates': 'false'
            }
            content = set_config_properties(git_urls,
                                            content_dict)
            update_yaml(config.config_file, content)
            echo_success("Configs are set !")
            jira = MyJiraWrapper('config.yaml', 'labels.yaml')

        else:
            echo_error("Please Pass missing options!")
    except Exception as err:
        click.echo(err)


@cli.command()
@click.option('--interval', default='week',
              help="please pass interval e.g. week, day")
@pass_config
def check_github_history(config, interval):
    """See GitHub History result based on intervals
     e.g. jirasync check-github-history --interval week """
    if interval not in ['week', 'day']:
        echo_error("Please pass correct interval. e.g. 'week', 'day'")
        exit(1)
    else:
        interval = configure_interval(interval)
    yaml_data = get_yaml_data(config.config_file)
    git_hub_plugin = GitHubPlugin(yaml_data['github_username'],
                                  yaml_data['email_id'],
                                  interval,
                                  yaml_data['git_urls']
                                  )
    issue_list = git_hub_plugin.get_github_issues_created_list()
    assigned_issue_list = git_hub_plugin.get_github_issues_assigned_list()
    pr_list = git_hub_plugin.get_opened_pull_requests()
    pr_closed_list = git_hub_plugin.get_pull_requests_merged_closed()
    pr_review_list_closed = git_hub_plugin.get_pull_requests_reviewed()
    pr_review_list_open = git_hub_plugin.get_pull_requests_review_in_progress()
    formatting('Issues Created', issue_list)
    formatting('Issues Assigned', assigned_issue_list)
    formatting('PR Raised', pr_list)
    formatting('PR Merged', pr_closed_list)
    formatting('PR Reviewed Closed Status', pr_review_list_closed)
    formatting('PR Reviewed Open Status', pr_review_list_open)


@cli.command()
@click.option('--interval', default='week',
              help="please pass interval e.g. week, day")
@pass_config
def start_syncing(config, interval):
    """Sync github issues, pr, pr_reviews as Jira tasks.
    Currently supported day and week interval
    e.g. jirasync start-syncing --interval week
    """
    if interval not in ['week', 'day']:
        echo_error("Please pass correct interval. e.g. 'week', 'day'")
        exit(1)
    else:
        interval = configure_interval(interval)

    yaml_data = get_yaml_data(config.config_file)
    git_hub_plugin = GitHubPlugin(yaml_data['github_username'],
                                  yaml_data['email_id'],
                                  interval, yaml_data['git_urls']
                                  )
    created_issue_list = git_hub_plugin.get_github_issues_created_list()
    assigned_issue_list = git_hub_plugin.get_github_issues_assigned_list()
    pr_list = git_hub_plugin.get_opened_pull_requests()
    pr_closed_list = git_hub_plugin.get_pull_requests_merged_closed()
    pr_review_list_closed = git_hub_plugin.get_pull_requests_reviewed()
    pr_review_list_open = git_hub_plugin.get_pull_requests_review_in_progress()
    jira = MyJiraWrapper('config.yaml', 'labels.yaml')
    # formatting('Issues Created', created_issue_list)
    # formatting('Issues Assigned', assigned_issue_list)
    # formatting('PR Raised', pr_list)
    # formatting('PR Merged', pr_closed_list)
    # formatting('PR Reviewed Closed Status', pr_review_list_closed)
    # formatting('PR Reviewed Open Status', pr_review_list_open)
    start_issue_workflow(github_issues=created_issue_list, jira=jira)
    start_issue_workflow(github_issues=assigned_issue_list, jira=jira)
    start_create_pull_requests_workflow(github_issues=pr_list, jira=jira)
    start_create_pull_requests_workflow(github_issues=pr_closed_list, jira=jira)
    start_review_pull_requests_workflow(github_issues=pr_review_list_closed, jira=jira)
    start_review_pull_requests_workflow(github_issues=pr_review_list_open, jira=jira)