Add prototype 'add' mode

This commit is contained in:
karl 2020-12-15 20:11:40 +01:00
parent 20a66b8f48
commit f4a2735da0

26
tx
View File

@ -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)
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))
hours, minutes = date_time_description[1].split(':', 1)
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")