Add mode argument, default 'total'
This commit is contained in:
parent
59722e19a2
commit
20a66b8f48
23
tx
23
tx
@ -8,6 +8,10 @@ parser = argparse.ArgumentParser(description="Description")
|
|||||||
|
|
||||||
# add arguments to the parser
|
# add arguments to the parser
|
||||||
parser.add_argument("-f", "--file")
|
parser.add_argument("-f", "--file")
|
||||||
|
parser.add_argument("mode",
|
||||||
|
help="display the square of a given number",
|
||||||
|
nargs='?', default="total",
|
||||||
|
choices=["total"])
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
@ -16,14 +20,15 @@ if args.file:
|
|||||||
else:
|
else:
|
||||||
time_file = open('time.txt','r')
|
time_file = open('time.txt','r')
|
||||||
|
|
||||||
total_time = timedelta()
|
if args.mode == "total":
|
||||||
|
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)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user