time-dr-jira-sync/sync.py at master · ekalinak/time-dr-jira-sync · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/python3.6
import argparse
from colorama import init
from termcolor import colored
from libs.time_doctor.timedoctor import TimeDoctor
from libs.time_doctor.menu import Menu as TimeDoctorMenu
from libs.jira.jira import Jira

from libs.time_doctor.connector import Connector


def print_message(message, debug=False):
    """
    Message is written to stdOut in case debug is allowed

    :param message:
    :param debug:
    :return:
    """
    if debug:
        print(message)


def prepare_data_from_file(file, debug):
    """
    Returns data exported from time doctor to Jira.

    :param file:
    :return:
    """
    print_message('[ info ] Parsing data from file ... ', debug)
    td = TimeDoctor(file)
    print_message('[ info ] Preparing data from file ... ', debug)
    return td.get_parsed_data()


def prepare_data_from_input(debug):
    print_message('[ info ]: Parsing data via API', debug)
    td_connector = Connector()
    print_message('[ info ]: TD Connector initialized ... ', debug)
    td_menu = TimeDoctorMenu()
    print_message('[ info ]: TD menu initialized ...', debug)
    range_menu_option = td_menu.show_range_menu()
    data_to_sync = None
    if td_menu.is_range_menu_day(range_menu_option):
        print_message('[ info ]: Preparing day data ...', debug)
        data_to_sync = td_connector.get_user_day_worklogs()
        print_message('[ info ]: Day data prepared ...', debug)
    if td_menu.is_range_menu_range(range_menu_option):
        print_message('[ info ]: Preparing range data ...', debug)
        data_to_sync = td_connector.get_user_range_worklogs()
        print_message('[ info ]: Range data prepared', debug)

    print_message('[ info ]: Sync data prepared ...', debug)
    if data_to_sync:
        td = TimeDoctor(data_to_sync, False)
    else:
        return False
    return td.get_parsed_data()


def prepare_data_automatically():
    td_connector = Connector()
    data_to_sync = td_connector.get_user_day_worklogs(True)
    if data_to_sync:
        td = TimeDoctor(data_to_sync, False)
    else:
        return False
    return td.get_parsed_data()


def main():
    _EXAMPLE_FILE = 'data/example_data.csv'
    parser = argparse.ArgumentParser(description='Sync data from TimeDoctor to Jira')
    parser.add_argument('-f', '--file',
                        help='Input source file. Exported CSV from TimeDoctor application.',
                        default=_EXAMPLE_FILE)
    parser.add_argument('-d', '--debug',
                        help='Debugging flag',
                        action='store_const',
                        const=True,
                        default=False)
    parser.add_argument('-dr', '--dry-run',
                        help='If used, data won\'t be sent to Jira',
                        action='store_const',
                        const=True,
                        default=False)
    parser.add_argument('-a', '--auto',
                        help='Automatically tries to sync current day',
                        action='store_const',
                        const=True,
                        default=False)
    args = parser.parse_args()

    if args.auto:
        prepared_data = prepare_data_automatically()
    elif (args.file and args.file != _EXAMPLE_FILE) or (args.file == _EXAMPLE_FILE and args.debug):
        prepared_data = prepare_data_from_file(args.file, args.debug)
    else:
        prepared_data = prepare_data_from_input(args.debug)

    print_message('[ info ]: Data preparation complete  ...', args.debug)

    if prepared_data:
        jira = Jira(prepared_data, args.dry_run)
        print_message('[ info ]: Data for syncing prepared in Jira classes ....', args.debug)
        jira.sync_data(args.auto)
        print_message('[ info ]: Data synced to Jira', args.debug)
    else:
        print_message('[ info ]: Data were not prepared. No action taken ...', args.debug)


if __name__ == '__main__':
    try:
        init()  # For colorama and windows platform
        main()
    except Exception as e:
        print(colored('\nError:\n{}\n\n'.format(str(e)), 'grey', 'on_red'))