From f4a2735da07f88e0db1b6a0fe14912a26f8e50f4 Mon Sep 17 00:00:00 2001 From: karl Date: Tue, 15 Dec 2020 20:11:40 +0100 Subject: [PATCH] Add prototype 'add' mode --- tx | 30 ++++++++++++++++-------------- 1 file changed, 16 insertions(+), 14 deletions(-) diff --git a/tx b/tx index 48ac767..6ea2bcc 100755 --- a/tx +++ b/tx @@ -11,24 +11,26 @@ parser.add_argument("-f", "--file") parser.add_argument("mode", help="display the square of a given number", nargs='?', default="total", - choices=["total"]) + choices=["total", "add"]) args = parser.parse_args() -if args.file: - time_file = open(args.file,'r') -else: - time_file = open('time.txt','r') +filename = args.file if args.file else 'time.txt' if args.mode == "total": - total_time = timedelta() + with open(filename, 'r') as time_file: + total_time = timedelta() - for line in time_file: - date_time_description = line.split(" ", 2) - - hours, minutes = date_time_description[1].split(':', 1) - delta = timedelta(hours=int(hours), minutes=int(minutes)) - - total_time += delta + for line in time_file: + date_time_description = line.split(" ", 2) + + hours, minutes = date_time_description[1].split(':', 1) + delta = timedelta(hours=int(hours), minutes=int(minutes)) + + total_time += delta - print('Total time: ', total_time) + print('Total time: ', total_time) + +elif args.mode == "add": + with open(filename, 'a') as time_file: + time_file.write("A new line")