myjunk/tl.py at master · karibou/myjunk · 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
#!/usr/bin/python3
# Add a new task to the gtimelog log file.
# Propose a list of task if only the category is supplied
#
# Copyright (c) 2006 Canonical Ltd.
# Author: Louis Bouchard <louis.bouchard@ubuntu.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2 of the License, or (at your
# option) any later version.  See http://www.gnu.org/copyleft/gpl.html for
# the full text of the license.

import re, sys, time

LogFile='~/.local/share/gtimelog/timelog.txt'

Categories = {
    'ua': 'L3 / L3 support',
    'up': 'Upstream',
    'ubu': 'Upstream Ubuntu',
    'meet': 'Internal meetings',
    'pers': 'Personal management',
    'skill': 'Skills building',
    'know': 'Knowledge transfer',
    'team': 'Team support'
    }


def show_help():
    return("Categories : {}".format(sorted(list(Categories.keys()))))


def select_tasks(category):
    cases = []
    mytask = 0
    regex = re.compile(r'{}'.format(Categories[category]))
    with open(LogFile, 'r') as timelog:
        for line in timelog:
            if regex.findall(line):
                case = regex.split(line)[-1].strip()
                if case.lstrip(": ") not in cases:
                    cases.append(case.lstrip(": "))
    for I in cases:
        print("{}) {}".format(cases.index(I) + 1, I))
    try:
        mytask = input("Select task (0 to exit): ")
        if mytask == '' or mytask == '0' or not mytask.isdecimal():
            return(None, None)
        else:
            mytask = int(mytask)

        if mytask > 0 and mytask <= len(cases):
            return(category, cases[mytask - 1])
        else:
            print("Invalid task number")
            return(None, None)

    except KeyboardInterrupt:
        print("Terminated\n")
        sys.exit(1)


def log_activity(category, task=None):

    now = time.localtime()
    today = '{:04d}-{:02d}-{:02d} {:02d}:{:02d}'.format(
            now.tm_year, now.tm_mon, now.tm_mday,
            now.tm_hour, now.tm_min)

    with open(LogFile, 'a') as timelog:
        if task is None:
            timelog.write('{}: {}\n'.format(today, category))
        else:
            if category in Categories.keys():
                timelog.write('{}: {} : {}\n'.format(
                    today, Categories[category], task))
            else:
                timelog.write('{}: {} : {}\n'.format(
                    today, category, task))
    return(0)

if __name__ == '__main__':
    try:
        if len(sys.argv) == 1:
            raise ValueError

        if len(sys.argv) == 2:
            if sys.argv[1] == '?':
                print("{}".format(show_help()))
                sys.exit(0)
            elif sys.argv[1] == 'new':
                log_activity('new', 'Arrived')
            elif sys.argv[1] in Categories.keys():
                (category, task) = select_tasks(sys.argv[1])
                if category is not None:
                    log_activity(category, task)
            else:
                log_activity(sys.argv[1])
            sys.exit(0)
        else:
            log_activity(sys.argv[1], " ".join(sys.argv[3:]))
        sys.exit(0)

    except ValueError:
        print("Missing Argument")
        sys.exit(1)