Add prototype 'add' mode
This commit is contained in:
parent
20a66b8f48
commit
f4a2735da0
26
tx
26
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)
|
||||
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")
|
||||
|
Loading…
x
Reference in New Issue
Block a user