tyo/tyo.py at master · ercco/tyo · 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
import datetime
import sys

def get_formatted_time(datetime_object):
    return datetime_object.strftime('%Y-%m-%dT%H:%M:%S')

def get_time_object(formatted_time):
    return datetime.datetime.strptime(formatted_time,'%Y-%m-%dT%H:%M:%S')

def write_to_file(datetime_object,event_type,tag='',additional_info=''):
    formatted_time = get_formatted_time(datetime_object)
    with open(formatted_time[0:7]+'.txt','a') as f:
        if event_type == 'start':
            f.write('\t'.join(filter(None,['A',formatted_time,tag,additional_info]))+'\n')
        elif event_type == 'stop':
            f.write('\t'.join(filter(None,['O',formatted_time,tag,additional_info]))+'\n')

def start(tag='',additional_info=''):
    start_time = datetime.datetime.now()
    write_to_file(start_time,'start',tag,additional_info)

def stop(tag='',additional_info=''):
    stop_time = datetime.datetime.now()
    write_to_file(stop_time,'stop',tag,additional_info)

def summarize(time_period):
    if len(time_period) == 4 or len(time_period) == 15:
        summarize_period(time_period)
    elif len(time_period) == 7:
        summarize_month(time_period)
    else:
        print('Summarize argument invalid')

def summarize_period(time_period):
    if len(time_period) == 4:
        start_year = int(time_period)
        start_month = 1
        end_year = int(time_period)
        end_month = 12
    elif len(time_period) == 15:
        start_year = int(time_period[0:4])
        start_month = int(time_period[5:7])
        end_year = int(time_period[8:12])
        end_month = int(time_period[13:15])
    if start_year == end_year:
        months = [m for m in range(start_month,end_month+1)]
    else:
        months = [m for m in range(start_month,13)] + list(range(1,13))*(end_year-start_year-1) + [m for m in range(1,end_month+1)]
    return months

def summarize_month(year_month):
    pass

if __name__ == '__main__':
    data = [sys.argv[1]]
    if len(sys.argv) == 2:
        data.append('')
        data.append('')
    elif len(sys.argv) == 3:
        data.append(sys.argv[2])
        data.append('')
    elif len(sys.argv) == 4:
        data.append(sys.argv[2])
        data.append(sys.argv[3])
    if sys.argv[1] == 'start':
        start(data[1],data[2])
    elif sys.argv[1] == 'stop':
        stop(data[1],data[2])