diff --git a/.config/i3pystatus/ice-status.py b/.config/i3pystatus/ice-status.py index 1f41417..195dbff 100755 --- a/.config/i3pystatus/ice-status.py +++ b/.config/i3pystatus/ice-status.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 -from datetime import datetime +from datetime import datetime, timedelta from requests import get from subprocess import check_output from sys import exit @@ -52,8 +52,12 @@ try: else: delay = '' - arrival = datetime.fromtimestamp(stop['timetable']['actualArrivalTime']/1000) - arrival_in = arrival - now + if stop['timetable'].get('actualArrivalTime', 0): + arrival = datetime.fromtimestamp(stop['timetable']['actualArrivalTime']/1000) + arrival_in = arrival - now + else: + arrival = datetime.now() + arrival_in = timedelta() next_stop = '{} [{}] {} ({}{})'.format( stop['station']['name'],