karl/time.txt
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add prototype 'add' mode

Browse Source
This commit is contained in:
karl 2020-12-15 20:11:40 +01:00
parent 20a66b8f48
commit f4a2735da0
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tx
View File

@ -11,24 +11,26 @@ parser.add_argument("-f", "--file")
parser.add_argument("mode", parser.add_argument("mode",
help="display the square of a given number", help="display the square of a given number",
nargs='?', default="total", nargs='?', default="total",
choices=["total"]) choices=["total", "add"])
args = parser.parse_args() args = parser.parse_args()
if args.file: filename = args.file if args.file else 'time.txt'
time_file = open(args.file,'r')
else:
time_file = open('time.txt','r')
if args.mode == "total": if args.mode == "total":
total_time = timedelta() with open(filename, 'r') as time_file:
total_time = timedelta()
for line in time_file: for line in time_file:
date_time_description = line.split(" ", 2) date_time_description = line.split(" ", 2)
hours, minutes = date_time_description[1].split(':', 1) hours, minutes = date_time_description[1].split(':', 1)
delta = timedelta(hours=int(hours), minutes=int(minutes)) delta = timedelta(hours=int(hours), minutes=int(minutes))
total_time += delta 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")