Skip to content

Commit d64e5cd

Browse files
author
rightlag
committed
Added condition to check if start_time is greater than or equal to end_time, added use_verbose method
1 parent 8c81fdc commit d64e5cd

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

timely/timely.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ def __init__(self, region='us-east-1', iso=False, verbose=False):
2828
self.weekdays = weekdays
2929
self.verbose = verbose
3030

31+
def use_verbose(self):
32+
self.verbose = True
33+
3134
def all(self, instance_ids=None):
3235
"""Read weekday run times for all or specific EC2 instances.
3336
@@ -71,9 +74,12 @@ def add(self, instance_ids=None, weekdays=None, start_time=None,
7174
"""
7275
ftime = '%H:%M'
7376
if start_time and end_time:
74-
start_time = datetime.strptime(start_time,
75-
'%I:%M %p').strftime(ftime)
76-
end_time = datetime.strptime(end_time, '%I:%M %p').strftime(ftime)
77+
start_time = datetime.strptime(start_time, '%I:%M %p')
78+
end_time = datetime.strptime(end_time, '%I:%M %p')
79+
if start_time >= end_time:
80+
raise ValueError('start time can\'t be greater than end time')
81+
start_time = start_time.strftime(ftime)
82+
end_time = end_time.strftime(ftime)
7783
updated = '{0}-{1}'.format(start_time, end_time)
7884
else:
7985
updated = None

0 commit comments

Comments
 (0)